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

use core::num::NonZeroU8;

use crate::{
    diedto::DiedTo,
    game::{Game, GameSettings, SetupRole},
    game_test::{ActionPromptTitleExt, ActionResultExt, GameExt, SettingsExt, gen_players},
    message::night::{ActionPrompt, ActionPromptTitle},
    role::{Alignment, Role},
};
#[allow(unused)]
use pretty_assertions::{assert_eq, assert_ne, assert_str_eq};

#[test]
fn elder_doesnt_die_first_try_night_doesnt_know() {
    let players = gen_players(1..10);
    let elder_player_id = players[0].player_id.clone();
    let wolf_player_id = players[2].player_id.clone();
    let mut settings = GameSettings::empty();
    settings.add_role(
        SetupRole::Elder {
            knows_on_night: NonZeroU8::new(3).unwrap(),
        },
        |slot| {
            slot.assign_to = Some(elder_player_id.clone());
        },
    );
    settings.add_role(SetupRole::Werewolf, |slot| {
        slot.assign_to = Some(wolf_player_id.clone())
    });
    settings.fill_remaining_slots_with_villagers(players.len());
    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();

    assert_eq!(game.execute().title(), ActionPromptTitle::CoverOfDarkness);
    game.r#continue().r#continue();
    assert_eq!(game.next().title(), ActionPromptTitle::WolfPackKill);
    let elder = game.character_by_player_id(elder_player_id);

    game.mark_and_check(elder.character_id());
    game.r#continue().sleep();

    game.next_expect_day();

    let elder = game.character_by_player_id(elder_player_id);
    assert_eq!(elder.died_to().cloned(), None);

    assert_eq!(game.execute(), ActionPrompt::CoverOfDarkness);
    game.r#continue().r#continue();

    assert_eq!(game.next().title(), ActionPromptTitle::WolfPackKill);
    game.mark_and_check(elder.character_id());
    game.r#continue().sleep();

    game.next_expect_day();
    assert_eq!(
        game.character_by_player_id(elder_player_id)
            .died_to()
            .cloned(),
        Some(DiedTo::Wolfpack {
            killing_wolf: game
                .character_by_player_id(wolf_player_id)
                .character_id()
                .clone(),
            night: NonZeroU8::new(2).unwrap(),
        })
    );
}

#[test]
fn elder_doesnt_die_first_try_night_knows() {
    let players = gen_players(1..10);
    let elder_player_id = players[0].player_id.clone();
    let wolf_player_id = players[2].player_id.clone();
    let mut settings = GameSettings::empty();
    settings.add_role(
        SetupRole::Elder {
            knows_on_night: NonZeroU8::new(1).unwrap(),
        },
        |slot| {
            slot.assign_to = Some(elder_player_id.clone());
        },
    );
    settings.add_role(SetupRole::Werewolf, |slot| {
        slot.assign_to = Some(wolf_player_id.clone())
    });
    settings.fill_remaining_slots_with_villagers(players.len());
    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();

    assert_eq!(game.execute().title(), ActionPromptTitle::CoverOfDarkness);
    game.r#continue().r#continue();
    assert_eq!(game.next().title(), ActionPromptTitle::WolfPackKill);
    let elder = game.character_by_player_id(elder_player_id);

    game.mark_and_check(elder.character_id());
    game.r#continue().sleep();

    assert_eq!(
        game.next(),
        ActionPrompt::ElderReveal {
            character_id: elder.identity()
        },
    );
    game.r#continue().sleep();

    game.next_expect_day();

    let elder = game.character_by_player_id(elder_player_id);
    assert_eq!(elder.died_to().cloned(), None);

    assert_eq!(game.execute(), ActionPrompt::CoverOfDarkness);
    game.r#continue().r#continue();

    assert_eq!(game.next().title(), ActionPromptTitle::WolfPackKill);
    game.mark_and_check(elder.character_id());
    game.r#continue().sleep();

    game.next_expect_day();
    assert_eq!(
        game.character_by_player_id(elder_player_id)
            .died_to()
            .cloned(),
        Some(DiedTo::Wolfpack {
            killing_wolf: game.character_by_player_id(wolf_player_id).character_id(),
            night: NonZeroU8::new(2).unwrap(),
        })
    );
}

#[test]
fn elder_executed_doesnt_know() {
    let players = gen_players(1..10);
    let elder_player_id = players[0].player_id;
    let seer_player_id = players[1].player_id;
    let wolf_player_id = players[2].player_id;
    let mut settings = GameSettings::empty();
    settings.add_role(
        SetupRole::Elder {
            knows_on_night: NonZeroU8::new(3).unwrap(),
        },
        |slot| {
            slot.assign_to = Some(elder_player_id.clone());
        },
    );
    settings.add_and_assign(SetupRole::Seer, seer_player_id.clone());
    settings.add_role(SetupRole::Werewolf, |slot| {
        slot.assign_to = Some(wolf_player_id.clone())
    });
    settings.fill_remaining_slots_with_villagers(players.len());
    let mut game = Game::new(&players, settings).unwrap();
    let mut villagers = game.villager_character_ids().into_iter();
    game.r#continue().r#continue();
    assert_eq!(game.next().title(), ActionPromptTitle::WolvesIntro);
    game.r#continue().sleep();

    game.next().title().seer();
    game.mark(villagers.next().unwrap());
    assert_eq!(game.r#continue().seer(), Alignment::Village);

    game.next_expect_day();
    let elder = game.character_by_player_id(elder_player_id);

    game.mark_for_execution(elder.character_id());

    game.execute().title().cover_of_darkness();
    game.r#continue().r#continue();

    game.next().title().wolf_pack_kill();

    assert_eq!(
        game.character_by_player_id(elder_player_id)
            .died_to()
            .cloned(),
        Some(DiedTo::Execution {
            day: NonZeroU8::new(1).unwrap()
        })
    );

    assert_eq!(
        game.character_by_player_id(elder_player_id).role().clone(),
        Role::Elder {
            knows_on_night: NonZeroU8::new(3).unwrap(),
            woken_for_reveal: false,
            lost_protection_night: None
        }
    );

    game.mark(villagers.next().unwrap());
    game.r#continue().sleep();

    game.next().title().seer();
    game.mark(villagers.next().unwrap());
    assert_eq!(game.r#continue().seer(), Alignment::Village);

    game.next_expect_day();
}

#[test]
fn elder_executed_knows_no_powers_incl_hunter_activation() {
    let players = gen_players(1..10);
    let elder_player_id = players[0].player_id;
    let seer_player_id = players[1].player_id;
    let wolf_player_id = players[2].player_id;
    let hunter_player_id = players[3].player_id;
    let mut settings = GameSettings::empty();
    settings.add_role(
        SetupRole::Elder {
            knows_on_night: NonZeroU8::new(1).unwrap(),
        },
        |slot| {
            slot.assign_to = Some(elder_player_id.clone());
        },
    );
    settings.add_and_assign(SetupRole::Seer, seer_player_id.clone());
    settings.add_role(SetupRole::Werewolf, |slot| {
        slot.assign_to = Some(wolf_player_id.clone())
    });
    settings.add_and_assign(SetupRole::Hunter, hunter_player_id);
    settings.fill_remaining_slots_with_villagers(players.len());
    let mut game = Game::new(&players, settings).unwrap();
    let mut villagers = game.villager_character_ids().into_iter();
    game.r#continue().r#continue();
    assert_eq!(game.next().title(), ActionPromptTitle::WolvesIntro);
    game.r#continue().sleep();

    game.next().title().seer();
    game.mark(game.living_villager_excl(seer_player_id).character_id());
    assert_eq!(game.r#continue().seer(), Alignment::Village);

    game.next_expect_day();

    game.execute().title().cover_of_darkness();
    game.r#continue().r#continue();

    game.next().title().wolf_pack_kill();
    game.mark(villagers.next().unwrap());
    game.r#continue().sleep();

    game.next().title().seer();
    game.mark(game.living_villager_excl(seer_player_id).character_id());
    assert_eq!(game.r#continue().seer(), Alignment::Village);

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

    assert_eq!(
        game.next(),
        ActionPrompt::ElderReveal {
            character_id: game.character_by_player_id(elder_player_id).identity()
        }
    );
    game.r#continue().sleep();

    game.next_expect_day();

    assert_eq!(
        game.character_by_player_id(elder_player_id).role().clone(),
        Role::Elder {
            knows_on_night: NonZeroU8::new(1).unwrap(),
            woken_for_reveal: true,
            lost_protection_night: None
        }
    );

    game.mark_for_execution(game.character_by_player_id(elder_player_id).character_id());
    game.execute().title().cover_of_darkness();
    game.r#continue().r#continue();

    game.next().title().wolf_pack_kill();
    game.mark(game.character_by_player_id(hunter_player_id).character_id());
    game.r#continue().sleep();

    game.next_expect_day();

    assert_eq!(
        game.character_by_player_id(hunter_player_id)
            .died_to()
            .cloned(),
        Some(DiedTo::Wolfpack {
            killing_wolf: game.character_by_player_id(wolf_player_id).character_id(),
            night: NonZeroU8::new(2).unwrap()
        })
    );

    assert_eq!(
        game.character_by_player_id(wolf_player_id)
            .died_to()
            .cloned(),
        None
    );
}