107 lines
3.6 KiB
Rust
107 lines
3.6 KiB
Rust
// Copyright (C) 2025 Emilis Bliūdžius
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as
|
|
// published by the Free Software Foundation, either version 3 of the
|
|
// License, or (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
#[allow(unused)]
|
|
use pretty_assertions::{assert_eq, assert_ne, assert_str_eq};
|
|
|
|
use crate::{
|
|
game::{Game, GameSettings, SetupRole},
|
|
game_test::{
|
|
ActionPromptTitleExt, ActionResultExt, GameExt, SettingsExt, gen_players, init_log,
|
|
},
|
|
message::night::{ActionPrompt, ActionPromptTitle},
|
|
};
|
|
|
|
#[test]
|
|
fn cannot_protect_same_target() {
|
|
init_log();
|
|
let players = gen_players(1..21);
|
|
let mut player_ids = players.iter().map(|p| p.player_id);
|
|
let protector = player_ids.next().unwrap();
|
|
let wolf = player_ids.next().unwrap();
|
|
let mut settings = GameSettings::empty();
|
|
settings.add_and_assign(SetupRole::Protector, protector);
|
|
settings.add_and_assign(SetupRole::Werewolf, wolf);
|
|
settings.fill_remaining_slots_with_villagers(20);
|
|
let mut game = Game::new(&players, settings).unwrap();
|
|
game.r#continue().r#continue();
|
|
assert_eq!(game.next().title(), ActionPromptTitle::WolvesIntro);
|
|
game.r#continue().sleep();
|
|
|
|
game.next_expect_day();
|
|
game.execute().title().protector();
|
|
let prot = game.living_villager();
|
|
game.mark(prot.character_id());
|
|
game.r#continue().sleep();
|
|
|
|
game.next().title().wolf_pack_kill();
|
|
game.mark(prot.character_id());
|
|
game.r#continue().sleep();
|
|
|
|
game.next_expect_day();
|
|
|
|
assert_eq!(
|
|
game.character_by_player_id(protector)
|
|
.protector_mut()
|
|
.unwrap()
|
|
.last_protected
|
|
.clone(),
|
|
Some(prot.character_id())
|
|
);
|
|
|
|
match game.execute() {
|
|
ActionPrompt::Protector { targets, .. } => {
|
|
assert!(
|
|
!targets
|
|
.into_iter()
|
|
.any(|c| c.character_id == prot.character_id())
|
|
);
|
|
}
|
|
prompt => panic!("expected protector prompt, got {:?}", prompt.title()),
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn can_self_protect() {
|
|
init_log();
|
|
let players = gen_players(1..21);
|
|
let mut player_ids = players.iter().map(|p| p.player_id);
|
|
let protector = player_ids.next().unwrap();
|
|
let wolf = player_ids.next().unwrap();
|
|
let mut settings = GameSettings::empty();
|
|
settings.add_and_assign(SetupRole::Protector, protector);
|
|
settings.add_and_assign(SetupRole::Werewolf, wolf);
|
|
settings.fill_remaining_slots_with_villagers(20);
|
|
let mut game = Game::new(&players, settings).unwrap();
|
|
game.r#continue().r#continue();
|
|
assert_eq!(game.next().title(), ActionPromptTitle::WolvesIntro);
|
|
game.r#continue().sleep();
|
|
|
|
game.next_expect_day();
|
|
game.execute().title().protector();
|
|
game.mark(game.character_by_player_id(protector).character_id());
|
|
game.r#continue().sleep();
|
|
|
|
game.next().title().wolf_pack_kill();
|
|
game.mark(game.character_by_player_id(protector).character_id());
|
|
game.r#continue().sleep();
|
|
|
|
game.next_expect_day();
|
|
|
|
assert_eq!(
|
|
game.character_by_player_id(protector).died_to().cloned(),
|
|
None
|
|
);
|
|
}
|