battler 0.8.0

Pokémon battle engine for Rust.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180

use anyhow::Result;
use battler::{
    BattleType,
    CoreBattleEngineSpeedSortTieResolution,
    PublicCoreBattle,
    TeamData,
    WrapResultError,
};
use battler_test_utils::{
    LogMatch,
    TestBattleBuilder,
    assert_logs_since_turn_eq,
    get_controlled_rng_for_battle,
    static_local_data_store,
};

fn pikachu() -> Result<TeamData> {
    serde_json::from_str(
        r#"{
            "members": [
                {
                    "name": "Pikachu",
                    "species": "Pikachu",
                    "ability": "No Ability",
                    "moves": [
                        "Tackle"
                    ],
                    "nature": "Hardy",
                    "gender": "M",
                    "level": 50,
                    "friendship": 255
                }
            ]
        }"#,
    )
    .wrap_error()
}

fn eevee() -> Result<TeamData> {
    serde_json::from_str(
        r#"{
            "members": [
                {
                    "name": "Eevee",
                    "species": "Eevee",
                    "ability": "No Ability",
                    "moves": [
                        "Earthquake",
                        "Toxic",
                        "Pin Missile"
                    ],
                    "nature": "Hardy",
                    "gender": "M",
                    "level": 50,
                    "friendship": 255
                }
            ]
        }"#,
    )
    .wrap_error()
}

fn make_trainer_singles_battle(
    seed: u64,
    team_1: TeamData,
    team_2: TeamData,
) -> Result<PublicCoreBattle<'static>> {
    TestBattleBuilder::new()
        .with_battle_type(BattleType::Singles)
        .with_seed(seed)
        .with_team_validation(false)
        .with_pass_allowed(true)
        .with_controlled_rng(true)
        .with_speed_sort_tie_resolution(CoreBattleEngineSpeedSortTieResolution::Keep)
        .add_protagonist_to_side_1("protagonist", "Protagonist")
        .add_player_to_side_2("trainer", "Trainer")
        .with_team("protagonist", team_1)
        .with_team("trainer", team_2)
        .build(static_local_data_store())
}

#[test]
fn affection_cures_status() {
    let mut battle = make_trainer_singles_battle(0, pikachu().unwrap(), eevee().unwrap()).unwrap();
    assert_matches::assert_matches!(battle.start(), Ok(()));

    let rng = get_controlled_rng_for_battle(&mut battle).unwrap();
    rng.insert_fake_values_relative_to_sequence_count([(2, 0)]);

    assert_matches::assert_matches!(battle.set_player_choice("protagonist", "pass"), Ok(()));
    assert_matches::assert_matches!(battle.set_player_choice("trainer", "move 1"), Ok(()));

    let expected_logs = serde_json::from_str::<Vec<LogMatch>>(
        r#"[
            "move|mon:Eevee,trainer,1|name:Toxic|target:Pikachu,protagonist,1",
            "status|mon:Pikachu,protagonist,1|status:Bad Poison",
            "curestatus|mon:Pikachu,protagonist,1|status:Bad Poison|from:Affection",
            "residual",
            "turn|turn:2"
        ]"#,
    )
    .unwrap();
    assert_logs_since_turn_eq(&battle, 1, &expected_logs);
}

#[test]
fn affection_survives_hit() {
    let mut battle = make_trainer_singles_battle(0, pikachu().unwrap(), eevee().unwrap()).unwrap();
    assert_matches::assert_matches!(battle.start(), Ok(()));

    let rng = get_controlled_rng_for_battle(&mut battle).unwrap();
    rng.insert_fake_values_relative_to_sequence_count([(4, 0)]);

    assert_matches::assert_matches!(battle.set_player_choice("protagonist", "pass"), Ok(()));
    assert_matches::assert_matches!(battle.set_player_choice("trainer", "move 0"), Ok(()));

    let expected_logs = serde_json::from_str::<Vec<LogMatch>>(
        r#"[
            "move|mon:Eevee,trainer,1|name:Earthquake",
            "supereffective|mon:Pikachu,protagonist,1",
            "activate|mon:Pikachu,protagonist,1|condition:Affection|tough",
            "split|side:0",
            "damage|mon:Pikachu,protagonist,1|health:1/95",
            "damage|mon:Pikachu,protagonist,1|health:2/100",
            "residual",
            "turn|turn:2"
        ]"#,
    )
    .unwrap();
    assert_logs_since_turn_eq(&battle, 1, &expected_logs);
}

#[test]
fn affection_survives_multiple_hits() {
    let mut battle = make_trainer_singles_battle(0, pikachu().unwrap(), eevee().unwrap()).unwrap();
    assert_matches::assert_matches!(battle.start(), Ok(()));

    let rng = get_controlled_rng_for_battle(&mut battle).unwrap();
    rng.insert_fake_values_relative_to_sequence_count([
        (4, 0),
        (5, 18),
        (6, 0),
        (7, 0),
        (9, 0),
        (11, 0),
        (15, 0),
        (18, 0),
        (19, 0),
        (21, 99),
    ]);

    assert_matches::assert_matches!(battle.set_player_choice("protagonist", "pass"), Ok(()));
    assert_matches::assert_matches!(battle.set_player_choice("trainer", "move 0"), Ok(()));
    assert_matches::assert_matches!(battle.set_player_choice("protagonist", "pass"), Ok(()));
    assert_matches::assert_matches!(battle.set_player_choice("trainer", "move 2"), Ok(()));

    let expected_logs = serde_json::from_str::<Vec<LogMatch>>(
        r#"[
            "move|mon:Eevee,trainer,1|name:Pin Missile|target:Pikachu,protagonist,1",
            "crit|mon:Pikachu,protagonist,1",
            "activate|mon:Pikachu,protagonist,1|condition:Affection|tough",
            "animatemove|mon:Eevee,trainer,1|name:Pin Missile|target:Pikachu,protagonist,1",
            "activate|mon:Pikachu,protagonist,1|condition:Affection|tough",
            "animatemove|mon:Eevee,trainer,1|name:Pin Missile|target:Pikachu,protagonist,1",
            "activate|mon:Pikachu,protagonist,1|condition:Affection|tough",
            "animatemove|mon:Eevee,trainer,1|name:Pin Missile|target:Pikachu,protagonist,1",
            "activate|mon:Pikachu,protagonist,1|condition:Affection|tough",
            "animatemove|mon:Eevee,trainer,1|name:Pin Missile|target:Pikachu,protagonist,1",
            "crit|mon:Pikachu,protagonist,1",
            "split|side:0",
            "damage|mon:Pikachu,protagonist,1|health:0",
            "damage|mon:Pikachu,protagonist,1|health:0",
            "faint|mon:Pikachu,protagonist,1",
            "hitcount|hits:5",
            "win|side:1"
        ]"#,
    )
    .unwrap();
    assert_logs_since_turn_eq(&battle, 2, &expected_logs);
}