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

// 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/>.
use core::num::NonZeroU8;

use crate::{
    diedto::DiedTo,
    error::GameError,
    game::{Game, GameSettings, SetupRole},
    game_test::{
        ActionPromptTitleExt, ActionResultExt, GameExt, SettingsExt, gen_players, init_log,
    },
    message::{
        host::{HostGameMessage, HostNightMessage},
        night::{ActionPromptTitle, ActionResponse},
    },
    role::{PreviousGuardianAction, Role},
};

#[test]
fn can_change_targets() {
    init_log();
    let players = gen_players(1..10);
    let hunter = players[0].player_id;
    let wolf = players[1].player_id;
    let mut settings = GameSettings::empty();
    settings.add_and_assign(SetupRole::Werewolf, wolf);
    settings.add_and_assign(SetupRole::Hunter, hunter);
    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();
    let hunt_target = game.living_villager();
    game.execute().title().wolf_pack_kill();
    game.mark(
        game.living_villager_excl(hunt_target.player_id())
            .character_id(),
    );
    game.r#continue().sleep();

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

    game.next_expect_day();

    match game.character_by_player_id(hunter).role() {
        Role::Hunter {
            target: Some(target),
        } => assert_eq!(*target, hunt_target.character_id()),
        Role::Hunter { target: None } => panic!("did not record target"),
        _ => panic!("not a guardian?"),
    }

    let hunt_target = game.living_villager_excl(hunt_target.player_id());

    game.execute().title().wolf_pack_kill();
    game.mark(
        game.living_villager_excl(hunt_target.player_id())
            .character_id(),
    );
    game.r#continue().sleep();

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

    game.next_expect_day();

    match game.character_by_player_id(hunter).role() {
        Role::Hunter {
            target: Some(target),
        } => assert_eq!(*target, hunt_target.character_id()),
        Role::Hunter { target: None } => panic!("did not record target"),
        _ => panic!("not a guardian?"),
    }
}