manabrew_engine/ability/effects/
take_initiative_effect.rs1use super::{resolve_defined_player, EffectContext};
2
3#[manabrew_engine_macros::spell_effect(TakeInitiativeEffect)]
17fn resolve(ctx: &mut EffectContext, sa: &crate::spellability::SpellAbility) {
18 let controller = sa.activating_player;
19
20 let defined = sa.defined().unwrap_or("You");
21
22 let target = resolve_defined_player(defined, controller, ctx.game).unwrap_or(controller);
23
24 if !ctx.game.player(target).is_alive() {
25 return;
26 }
27
28 ctx.game
29 .player_take_initiative(target, Some(ctx.trigger_handler));
30}
31
32#[cfg(test)]
33mod tests {
34 use crate::ability::spell_ability_effect::SpellAbilityEffect;
35 use std::collections::HashMap;
36
37 use crate::ability::effects::EffectContext;
38 use crate::agent::PassAgent;
39 use crate::game::GameState;
40 use crate::ids::PlayerId;
41 use crate::mana::ManaPool;
42 use crate::spellability::SpellAbility;
43 use crate::trigger::handler::TriggerHandler;
44
45 #[test]
46 fn take_initiative_sets_state() {
47 let mut game = GameState::new(&["Alice", "Bob"], 20);
48 let p0 = PlayerId(0);
49 assert!(game.initiative_holder.is_none());
50
51 let sa = SpellAbility::new_simple(None, p0, "SP$ TakeInitiative | Defined$ You");
52
53 let mut th = TriggerHandler::new();
54 let mut agents: Vec<Box<dyn crate::agent::PlayerAgent>> =
55 vec![Box::new(PassAgent), Box::new(PassAgent)];
56 let mut mp = vec![ManaPool::default(), ManaPool::default()];
57 let templates = HashMap::new();
58 let templates_variants = HashMap::new();
59 let token_fallback = HashMap::new();
60 let edition_dates: HashMap<String, String> = HashMap::new();
61 let mut rng_adapter = crate::game_rng::ThreadRngAdapter;
62 let mut ctx = EffectContext {
63 game: &mut game,
64 combat: None,
65 agents: &mut agents,
66 trigger_handler: &mut th,
67 token_templates: &templates,
68 token_art_variants: &templates_variants,
69 token_fallback: &token_fallback,
70 edition_dates: &edition_dates,
71 mana_pools: &mut mp,
72 parent_target_card: None,
73 rng: &mut rng_adapter,
74 };
75 super::TakeInitiativeEffect::resolve(&mut ctx, &sa);
76
77 assert_eq!(ctx.game.initiative_holder, Some(p0));
78 }
79}