95 lines
2.4 KiB
Rust
Executable File
95 lines
2.4 KiB
Rust
Executable File
use crate::script;
|
|
use alloc::boxed::Box;
|
|
use core::any::Any;
|
|
use pkmn_lib_interface::app_interface::{ExecutingMove, Pokemon};
|
|
use pkmn_lib_interface::handling::{Script, ScriptCapabilities};
|
|
|
|
script!(Acrobatics, "acrobatics");
|
|
|
|
impl Script for Acrobatics {
|
|
fn new() -> Self {
|
|
Self {}
|
|
}
|
|
|
|
fn get_name(&self) -> &'static str {
|
|
Self::get_const_name()
|
|
}
|
|
|
|
fn get_capabilities(&self) -> &[ScriptCapabilities] {
|
|
&[ScriptCapabilities::ChangeBasePower]
|
|
}
|
|
|
|
fn change_base_power(
|
|
&self,
|
|
mv: ExecutingMove,
|
|
_target: Pokemon,
|
|
_hit: u8,
|
|
base_power: &mut u8,
|
|
) {
|
|
if mv.user().held_item().is_none() {
|
|
if *base_power >= 128_u8 {
|
|
*base_power = 255
|
|
} else {
|
|
*base_power *= 2;
|
|
}
|
|
}
|
|
}
|
|
|
|
fn as_any(&self) -> &dyn Any {
|
|
self
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use alloc::rc::Rc;
|
|
use pkmn_lib_interface::app_interface::{MockExecutingMove, MockItem, MockPokemon};
|
|
|
|
fn mock_executing_move(has_held_item: bool) -> ExecutingMove {
|
|
let mut mv = MockExecutingMove::new();
|
|
mv.expect_user().returning(move || {
|
|
let mut user = MockPokemon::new();
|
|
user.expect_held_item().returning(move || {
|
|
if has_held_item {
|
|
Some(Rc::new(MockItem::new()))
|
|
} else {
|
|
None
|
|
}
|
|
});
|
|
Rc::new(user)
|
|
});
|
|
Rc::new(mv)
|
|
}
|
|
|
|
#[test]
|
|
fn without_held_item_doubles_power() {
|
|
let mv = mock_executing_move(false);
|
|
|
|
let script = Acrobatics::new();
|
|
let mut base_power = 50u8;
|
|
script.change_base_power(mv, Rc::new(MockPokemon::new()), 0, &mut base_power);
|
|
assert_eq!(100, base_power);
|
|
}
|
|
|
|
#[test]
|
|
fn without_held_item_does_not_overflow() {
|
|
let mv = mock_executing_move(false);
|
|
|
|
let script = Acrobatics::new();
|
|
let mut base_power = 200u8;
|
|
script.change_base_power(mv, Rc::new(MockPokemon::new()), 0, &mut base_power);
|
|
assert_eq!(255, base_power);
|
|
}
|
|
|
|
#[test]
|
|
fn with_held_item_does_nothing() {
|
|
let mv = mock_executing_move(true);
|
|
|
|
let script = Acrobatics::new();
|
|
let mut base_power = 50u8;
|
|
script.change_base_power(mv, Rc::new(MockPokemon::new()), 0, &mut base_power);
|
|
assert_eq!(50, base_power);
|
|
}
|
|
}
|