PkmnLib_rs/tests/common/data_getter.rs

23 lines
566 B
Rust
Raw Normal View History

use pkmn_lib::dynamic_data::models::battle::Battle;
use serde::Deserialize;
#[derive(Deserialize)]
#[serde(rename_all = "snake_case")]
pub enum TestDataGetter {
PokemonHealth { index: [u8; 2] },
}
impl TestDataGetter {
pub fn get(&self, battle: &Battle) -> String {
match self {
TestDataGetter::PokemonHealth { index } => battle
.get_pokemon(index[0], index[1])
.as_ref()
.unwrap()
.read()
.current_health()
.to_string(),
}
}
}