104 lines
2.7 KiB
Rust
Executable File
104 lines
2.7 KiB
Rust
Executable File
use crate::common_usings::*;
|
|
|
|
pub struct AfterYou {}
|
|
|
|
impl AfterYou {
|
|
pub const fn get_const_name() -> &'static str {
|
|
"after_you"
|
|
}
|
|
}
|
|
|
|
impl Script for AfterYou {
|
|
fn new() -> Self {
|
|
Self {}
|
|
}
|
|
|
|
fn get_name(&self) -> &'static str {
|
|
Self::get_const_name()
|
|
}
|
|
|
|
fn get_capabilities(&self) -> &[ScriptCapabilities] {
|
|
&[ScriptCapabilities::OnSecondaryEffect]
|
|
}
|
|
|
|
fn on_secondary_effect(&self, mv: ExecutingMove, target: Pokemon, hit: u8) {
|
|
if !target
|
|
.battle()
|
|
.unwrap()
|
|
.choice_queue()
|
|
.move_pokemon_choice_next(&target)
|
|
{
|
|
mv.get_hit_data(&target, hit).fail()
|
|
}
|
|
}
|
|
|
|
fn as_any(&self) -> &dyn Any {
|
|
self
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use alloc::rc::Rc;
|
|
use pkmn_lib_interface::app_interface::{
|
|
MockBattle, MockChoiceQueue, MockExecutingMove, MockHitData, MockPokemon,
|
|
};
|
|
|
|
#[test]
|
|
fn move_pokemon_choice_next_gets_called_once() {
|
|
let mut battle = MockBattle::new();
|
|
battle.expect_choice_queue().once().return_once_st(move || {
|
|
let mut choice_queue = MockChoiceQueue::new();
|
|
choice_queue
|
|
.expect_move_pokemon_choice_next()
|
|
.once()
|
|
.return_const(true);
|
|
Rc::new(choice_queue)
|
|
});
|
|
|
|
let battle = Rc::new(battle);
|
|
let mut target = MockPokemon::new();
|
|
target
|
|
.expect_battle()
|
|
.once()
|
|
.return_once_st(move || Some(battle));
|
|
let target = Rc::new(target);
|
|
|
|
let script = AfterYou::new();
|
|
script.on_secondary_effect(Rc::new(MockExecutingMove::new()), target, 0);
|
|
}
|
|
|
|
#[test]
|
|
fn fails_when_move_pokemon_choice_next_can_not_move() {
|
|
let mut battle = MockBattle::new();
|
|
battle.expect_choice_queue().once().return_once_st(move || {
|
|
let mut choice_queue = MockChoiceQueue::new();
|
|
choice_queue
|
|
.expect_move_pokemon_choice_next()
|
|
.once()
|
|
.return_const(false);
|
|
Rc::new(choice_queue)
|
|
});
|
|
|
|
let battle = Rc::new(battle);
|
|
let mut target = MockPokemon::new();
|
|
target
|
|
.expect_battle()
|
|
.once()
|
|
.return_once_st(move || Some(battle));
|
|
let target = Rc::new(target);
|
|
|
|
let mut mv = MockExecutingMove::new();
|
|
mv.expect_get_hit_data().once().returning_st(move |_, _| {
|
|
let mut hit = MockHitData::new();
|
|
hit.expect_fail().once();
|
|
Rc::new(hit)
|
|
});
|
|
let mv = Rc::new(mv);
|
|
|
|
let script = AfterYou::new();
|
|
script.on_secondary_effect(mv, target, 0);
|
|
}
|
|
}
|