werewolves/werewolves-proto/src/game_test/role/beholder.rs

use core::num::NonZeroU8;
#[allow(unused)]
use pretty_assertions::{assert_eq, assert_ne, assert_str_eq};

use crate::{
    game::{Game, GameSettings, SetupRole},
    game_test::{
        ActionPromptTitleExt, ActionResultExt, AlignmentExt, GameExt, SettingsExt, gen_players,
        init_log,
    },
    message::night::{ActionPrompt, ActionPromptTitle},
};

#[test]
fn beholding_seer() {
    init_log();
    let players = gen_players(1..10);
    let seer_player_id = players[0].player_id;
    let wolf_player_id = players[1].player_id;
    let beholder_player_id = players[2].player_id;
    let mut settings = GameSettings::empty();
    settings.add_and_assign(SetupRole::Seer, seer_player_id);
    settings.add_and_assign(SetupRole::Werewolf, wolf_player_id);
    settings.add_and_assign(SetupRole::Beholder, beholder_player_id);
    settings.fill_remaining_slots_with_villagers(9);
    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().title().seer();
    game.mark(game.character_by_player_id(wolf_player_id).character_id());
    game.r#continue().seer().wolves();
    game.r#continue().sleep();

    game.next_expect_day();
    game.execute().title().wolf_pack_kill();
    game.mark(game.living_villager().character_id());
    game.r#continue().sleep();

    game.next().title().seer();
    game.mark(game.character_by_player_id(wolf_player_id).character_id());
    game.r#continue().seer().wolves();
    game.r#continue().sleep();

    game.next().title().beholder();
    game.mark(game.character_by_player_id(seer_player_id).character_id());
    game.r#continue().seer().wolves();
    game.r#continue().sleep();

    game.next_expect_day();
}

#[test]
fn beholding_wolf() {
    let players = gen_players(1..10);
    let wolf_player_id = players[1].player_id;
    let beholder_player_id = players[2].player_id;
    let mut settings = GameSettings::empty();
    settings.add_and_assign(SetupRole::Werewolf, wolf_player_id);
    settings.add_and_assign(SetupRole::Beholder, beholder_player_id);
    settings.fill_remaining_slots_with_villagers(9);
    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().wolf_pack_kill();
    game.mark(game.living_villager_excl(beholder_player_id).character_id());
    game.r#continue().sleep();

    game.next().title().beholder();
    game.mark(game.character_by_player_id(wolf_player_id).character_id());
    game.r#continue().sleep();

    game.next_expect_day();
}