async def test_discard_supplies_move(): gs = GameState() gs.players = [PlayerShelter('0'), PlayerShelter('1')] gs.active_player = gs.players[0] shelter = gs.active_player shelter.print = dumper_factory() shelter.input = helper_factory(['3']) shelter.survivors = [CityCard()] shelter.supplies = [Supply.AXE, Supply.BARRICADES, Supply.ALARM] turn_end = await gs.discard_supplies_move(False) assert turn_end assert len(shelter.supplies) == 0 gs.active_player = shelter shelter.input = helper_factory(['1']) shelter.supplies = [Supply.AXE, Supply.BARRICADES, Supply.ALARM] turn_end = await gs.discard_supplies_move(False) assert not turn_end assert len(shelter.supplies) == 2 assert Supply.BARRICADES not in shelter.supplies gs.active_player = shelter shelter.input = helper_factory(['2']) turn_end = await gs.discard_supplies_move(False) assert turn_end assert len(shelter.supplies) == 2
async def test_end_active_player_turn_game_finished(fast_zombie, zombie): gs = GameState() gs.players = [ PlayerShelter(), PlayerShelter(), PlayerShelter(), PlayerShelter() ] gs.players[1].defeated = True gs.players[2].zombies = [fast_zombie] gs.players[2].survivors = [CityCard()] gs.players[3].survivors = [CityCard()] gs.active_player = gs.players[0] shelter = gs.active_player shelter.survivors = [CityCard()] shelter.zombies = [zombie] assert not shelter.defeated assert not gs.players[2].defeated assert not gs.players[3].defeated assert not gs.finished await gs.end_active_player_turn() assert shelter.defeated assert not gs.players[2].defeated assert not gs.players[3].defeated assert not gs.finished await gs.end_active_player_turn() assert shelter.defeated assert gs.players[2].defeated assert not gs.players[3].defeated assert gs.finished
async def test_play_sniper_rifle_zombie_in_city_and_shelter(gs, zombie, big_zombie): gs.active_player = PlayerShelter(print_foo=dumper_factory(), input_foo=helper_factory(['y'])) shelter = gs.active_player shelter.zombies = [big_zombie, zombie] gs.city_deck = [zombie, CityCard()] shelter.supplies = [Supply.SNIPER] await weapons.play_sniper_rifle(gs) assert len(gs.city_deck) == 1 assert len(shelter.zombies) == 2 assert zombie not in gs.city_deck assert len(shelter.supplies) == 0 assert len(gs.supply_graveyard) == 1 assert len(gs.city_graveyard) == 1 assert len(tests.common.outputs) == 3 gs.active_player = PlayerShelter(print_foo=dumper_factory(), input_foo=helper_factory(['n', '0'])) shelter = gs.active_player shelter.zombies = [big_zombie, zombie] gs.city_deck = [zombie, CityCard()] shelter.supplies = [Supply.SNIPER] await weapons.play_sniper_rifle(gs) assert len(gs.city_deck) == 2 assert len(shelter.zombies) == 1 assert big_zombie not in shelter.zombies assert zombie in gs.city_deck assert len(shelter.supplies) == 0 assert len(gs.supply_graveyard) == 2 assert len(gs.city_graveyard) == 2 assert len(tests.common.outputs) == 2
async def test_play_round_end_round_one_shelter_destroyed(zombie): gs = GameState() gs.players = [PlayerShelter('0'), PlayerShelter('1'), PlayerShelter('2')] gs.supply_deck = [ Supply.AXE, Supply.AXE, Supply.AXE, Supply.AXE, Supply.AXE, Supply.AXE ] for player in gs.players: player.survivors = [CityCard()] gs.players[0].zombies = [zombie] gs.active_player = gs.players[0] shelter = gs.active_player shelter.print = dumper_factory() shelter.input = helper_factory(['3', '3']) shelter.survivors = [CityCard()] shelter.supplies = [Supply.AXE, Supply.BARRICADES, Supply.ALARM] await gs.play_round() assert len(shelter.supplies) == 0 assert shelter.defeated assert gs.active_player != shelter shelter = gs.active_player shelter.print = dumper_factory() shelter.input = helper_factory(['3', '3']) await gs.play_round() assert len(shelter.supplies) == 3 assert not shelter.defeated assert gs.active_player != shelter shelter = gs.active_player shelter.print = dumper_factory() shelter.input = helper_factory(['3', '3']) await gs.play_round() assert len(shelter.supplies) == 3 assert not shelter.defeated assert gs.active_player != shelter
def test_gui_default(fast_zombie, zombie, big_zombie): gs = GameState() gs.players = [ PlayerShelter('Name_1'), PlayerShelter('Name_2'), PlayerShelter('Name_3') ] gs.players[1].zombies = [ zombie, fast_zombie, fast_zombie, big_zombie, zombie ] gs.players[1].obstacles = [Supply.BARRICADES, Supply.ALARM] gs.players[1].survivors = [CityCard(), CityCard()] gs.players[2].zombies = [big_zombie, big_zombie] gs.players[2].obstacles = [Supply.MINE_FILED] gs.players[2].survivors = [CityCard(), CityCard(), CityCard()] gs.active_player = gs.players[0] shelter = gs.active_player shelter.print = dumper_factory() shelter.zombies = [zombie] shelter.supplies = [Supply.AXE, Supply.MINE_FILED, Supply.LURE_OUT] shelter.obstacles = [Supply.MINE_FILED, Supply.ALARM] shelter.survivors = [CityCard(), CityCard(), CityCard(), CityCard()] shelter.gui_default(gs) assert len(tests.common.outputs) == 1 assert len(tests.common.outputs[0]) == 684 gs.city_deck = [zombie, CityCard()] shelter.gui_default(gs) assert len(tests.common.outputs) == 2 assert len(tests.common.outputs[1]) == 706
async def test_play_flare_gun(gs): shelter = gs.active_player gs.city_deck = [CityCard()] shelter.supplies = [Supply.FLARE_GUN] await summons.play_flare_gun(gs) assert len(shelter.survivors) == 1 assert len(shelter.supplies) == 0 assert len(gs.city_deck) == 0 assert len(gs.supply_graveyard) == 1 assert len(tests.common.outputs) == 5 gs.active_player = PlayerShelter(print_foo=dumper_factory()) shelter = gs.active_player gs.city_deck = [CityCard(), CityCard()] shelter.supplies = [Supply.FLARE_GUN] await summons.play_flare_gun(gs) assert len(shelter.survivors) == 2 assert len(shelter.supplies) == 0 assert len(gs.city_deck) == 0 assert len(gs.supply_graveyard) == 2 assert len(tests.common.outputs) == 5 gs.active_player = PlayerShelter(print_foo=dumper_factory()) shelter = gs.active_player gs.city_deck = [] shelter.supplies = [Supply.FLARE_GUN] await summons.play_flare_gun(gs) assert len(shelter.survivors) == 0 assert len(shelter.supplies) == 0 assert len(gs.city_deck) == 0 assert len(gs.supply_graveyard) == 3 assert len(tests.common.outputs) == 4
def __init__(self, name: str = '', print_foo: Callable[[str], None] = print): PlayerShelter.__init__(self, name, self.cpu_input, print_foo) self.gui = self.cpu_ui self.move_counter = -1 self.planned_moves = [] self.game_state = None
async def test_end_active_player_turn_no_zombies(): gs = GameState() gs.players = [PlayerShelter(), PlayerShelter(), PlayerShelter()] survivor_card = CityCard(ZombieType.ZOMBIE) survivor_card.flip() for player in gs.players: player.survivors.append(survivor_card) gs.active_player = gs.players[2] gs.supply_deck = [Supply.AXE, Supply.ALARM, Supply.AXE, Supply.GUN] await gs.end_active_player_turn() assert gs.active_player == gs.players[0] assert len(gs.players[2].supplies) == 3
async def test_play_round_end_round(): gs = GameState() gs.players = [PlayerShelter('0'), PlayerShelter('1')] gs.players[1].survivors = [CityCard()] gs.active_player = gs.players[0] shelter = gs.active_player shelter.print = dumper_factory() shelter.input = helper_factory(['3', '3']) shelter.survivors = [CityCard()] shelter.supplies = [Supply.AXE, Supply.BARRICADES, Supply.ALARM] await gs.play_round() assert len(shelter.supplies) == 3 assert gs.active_player != shelter
async def test_play_chainsaw_no_one_has_defence(gs): gs.players = [ PlayerShelter('ZERO'), PlayerShelter('FIRST'), PlayerShelter('SECOND') ] gs.active_player = gs.players[0] shelter = gs.active_player shelter.print = dumper_factory() shelter.supplies = [Supply.CHAINSAW] await counters.play_chainsaw(gs) assert len(shelter.supplies) == 0 assert len(gs.supply_graveyard) == 1 assert len(tests.common.outputs) == 2
async def test_play_round_win_by_takeover(zombie): gs = GameState() gs.players = [PlayerShelter('0'), PlayerShelter('1')] gs.players[1].survivors = [CityCard()] gs.active_player = gs.players[0] shelter = gs.active_player shelter.print = dumper_factory() shelter.input = helper_factory(['0']) shelter.zombies = [zombie] shelter.survivors = [CityCard()] shelter.supplies = [Supply.TAKEOVER, Supply.BARRICADES, Supply.ALARM] await gs.play_round() assert not shelter.defeated assert gs.players[1].defeated assert gs.finished
async def test_defend_with_mine_field(gs, fast_zombie, zombie, big_zombie): shelter = gs.active_player shelter.zombies = [zombie, big_zombie] shelter.obstacles = [Supply.MINE_FILED] await defences.defend_with_mine_field(gs) assert len(shelter.zombies) == 0 assert len(shelter.obstacles) == 0 assert len(gs.supply_graveyard) == 1 assert len(gs.city_graveyard) == 2 assert len(tests.common.outputs) == 5 gs.active_player = PlayerShelter(print_foo=dumper_factory(), input_foo=helper_factory(['0'])) shelter = gs.active_player shelter.zombies = [zombie, big_zombie, fast_zombie] shelter.obstacles = [Supply.MINE_FILED] await defences.defend_with_mine_field(gs) assert len(shelter.zombies) == 1 assert fast_zombie in shelter.zombies assert len(shelter.obstacles) == 0 assert len(gs.supply_graveyard) == 2 assert len(gs.city_graveyard) == 4 assert len(tests.common.outputs) == 5 gs.active_player = PlayerShelter(print_foo=dumper_factory(), input_foo=helper_factory(['1', '1'])) shelter = gs.active_player shelter.zombies = [zombie, big_zombie, fast_zombie] shelter.obstacles = [Supply.MINE_FILED] await defences.defend_with_mine_field(gs) assert len(shelter.zombies) == 1 assert big_zombie in shelter.zombies assert len(shelter.obstacles) == 0 assert len(gs.supply_graveyard) == 3 assert len(gs.city_graveyard) == 6 assert len(tests.common.outputs) == 5 gs.active_player = PlayerShelter(print_foo=dumper_factory(), input_foo=helper_factory(['1', '0'])) shelter = gs.active_player shelter.zombies = [zombie, big_zombie, fast_zombie] shelter.obstacles = [Supply.MINE_FILED] await defences.defend_with_mine_field(gs) assert len(shelter.zombies) == 1 assert fast_zombie in shelter.zombies assert len(shelter.obstacles) == 0 assert len(gs.supply_graveyard) == 4 assert len(gs.city_graveyard) == 8 assert len(tests.common.outputs) == 5
async def test_play_round_use_loud_tool(zombie): gs = GameState() gs.players = [PlayerShelter('0'), PlayerShelter('1')] gs.players[1].survivors = [CityCard()] gs.players[1].zombies = [zombie] gs.active_player = gs.players[0] gs.city_deck = [CityCard()] shelter = gs.active_player shelter.print = dumper_factory() shelter.input = helper_factory(['0', '2']) shelter.survivors = [CityCard()] shelter.supplies = [Supply.GUN, Supply.BARRICADES, Supply.ALARM] await gs.play_round() assert len(shelter.supplies) == 3 assert gs.city_deck[0].top == ZombieType.ZOMBIE assert gs.active_player != shelter
def test_get_supplies(): gs = GameState() gs.active_player = PlayerShelter() gs.supply_deck = [Supply.AXE, Supply.ALARM, Supply.AXE, Supply.GUN] gs.get_supplies() assert len(gs.supply_deck) == 1 assert gs.supply_deck[0] == Supply.GUN assert len(gs.active_player.supplies) == 3 gs = GameState() gs.active_player = PlayerShelter() gs.supply_deck = [Supply.AXE, Supply.ALARM, Supply.AXE, Supply.GUN] gs.active_player.supplies = [Supply.RADIO, Supply.DRONE] gs.get_supplies() assert len(gs.supply_deck) == 3 assert gs.supply_deck[0] == Supply.ALARM assert len(gs.active_player.supplies) == 3
async def test_play_drone_for_nothing(gs): gs.players = [ PlayerShelter('ZERO'), PlayerShelter('FIRST'), PlayerShelter('SECOND') ] gs.active_player = gs.players[0] gs.players[1].defeated = True shelter = gs.active_player shelter.print = dumper_factory() shelter.zombies = [] shelter.supplies = [Supply.DRONE] await counters.play_drone(gs) assert len(shelter.zombies) == 0 assert len(shelter.supplies) == 1 assert len(gs.supply_graveyard) == 0 assert len(tests.common.outputs) == 1
def clean_up_shelter(self, shelter: PlayerShelter) -> None: """ Method used to purge all card list of defeated player. :param shelter: PlayerShelter object of defeated player """ self.supply_graveyard += shelter.supplies + shelter.obstacles self.city_graveyard += shelter.zombies + shelter.survivors shelter.supplies, shelter.obstacles, shelter.zombies, shelter.survivors = [], [], [], []
async def test_play_round_play_axe_barricades_and_end_round( zombie, fast_zombie): gs = GameState() gs.players = [PlayerShelter('0'), PlayerShelter('1')] gs.players[1].survivors = [CityCard()] gs.active_player = gs.players[0] shelter = gs.active_player shelter.print = dumper_factory() shelter.input = helper_factory(['0', '0', '1', 'y']) shelter.zombies = [zombie, fast_zombie] shelter.survivors = [CityCard()] shelter.supplies = [Supply.AXE, Supply.BARRICADES, Supply.ALARM] await gs.play_round() assert len(shelter.supplies) == 3 assert len(shelter.zombies) == 1 assert len(shelter.survivors) == 1 assert gs.active_player != shelter
def test_fast_zombie_show_up(): gs = GameState() gs.active_player = PlayerShelter() gs.city_deck = [CityCard(ZombieType.FAST)] gs.zombie_show_up() assert len(gs.active_player.zombies) == 1 assert gs.active_player.zombies[0].top == ZombieType.FAST assert len(gs.city_deck) == 0
async def test_end_active_player_turn_zombies_no_obstacles(): gs = GameState() gs.players = [PlayerShelter(), PlayerShelter(), PlayerShelter()] survivor_card = CityCard(ZombieType.ZOMBIE) survivor_card.flip() for player in gs.players: player.survivors.append(survivor_card) gs.active_player = gs.players[2] gs.active_player.print = dumper_factory() gs.active_player.supplies = [Supply.AXE, Supply.ALARM] zombie_card = CityCard(ZombieType.ZOMBIE) zombie_card.flip() gs.active_player.zombies = [zombie_card] gs.supply_deck = [Supply.AXE, Supply.ALARM, Supply.AXE, Supply.GUN] await gs.end_active_player_turn() assert gs.active_player == gs.players[0] assert len(gs.players[2].supplies) == 0 assert len(tests.common.outputs) == 2
async def test_play_round_last_supplies(zombie): gs = GameState() gs.players = [PlayerShelter('0'), PlayerShelter('1')] gs.supply_deck = [Supply.SACRIFICE, Supply.DRONE, Supply.GUN] gs.city_deck = [CityCard()] gs.players[1].survivors = [CityCard()] gs.active_player = gs.players[0] shelter = gs.active_player shelter.print = dumper_factory() shelter.input = helper_factory(['0', '0', '0', 'y']) shelter.survivors = [CityCard()] shelter.zombies = [zombie] shelter.supplies = [Supply.RADIO, Supply.BARRICADES, Supply.ALARM] await gs.play_round() assert len(shelter.supplies) == 3 assert gs.final_attack assert gs.last_supplies_taken assert gs.active_player != shelter
async def test_play_drone_only_big(gs, big_zombie): gs.players = [ PlayerShelter('ZERO'), PlayerShelter('FIRST'), PlayerShelter('SECOND') ] gs.active_player = gs.players[0] shelter = gs.active_player shelter.print = dumper_factory() shelter.input = helper_factory(['0']) shelter.zombies = [big_zombie, big_zombie, big_zombie] shelter.supplies = [Supply.DRONE] await counters.play_drone(gs) assert len(shelter.zombies) == 2 assert len(shelter.supplies) == 0 assert len(gs.players[1].zombies) == 1 assert len(gs.players[2].zombies) == 0 assert len(gs.supply_graveyard) == 1 assert len(tests.common.outputs) == 3
async def test_play_swap(gs, fast_zombie, zombie, big_zombie): gs.players = [ PlayerShelter('ZERO'), PlayerShelter('FIRST'), PlayerShelter('SECOND') ] gs.active_player = gs.players[0] shelter = gs.active_player shelter.print = dumper_factory() shelter.input = helper_factory(['0']) shelter.supplies = [Supply.SWAP] shelter.zombies = [fast_zombie, big_zombie, fast_zombie, zombie] gs.players[1].obstacles = [Supply.BARRICADES, Supply.ALARM] await counters.play_swap(gs) assert len(shelter.zombies) == 0 assert len(shelter.obstacles) == 2 assert len(gs.players[1].zombies) == 4 assert len(gs.players[1].obstacles) == 0 assert len(tests.common.outputs) == 3
async def test_play_lure_for_nothing(gs): gs.players = [ PlayerShelter('ZERO'), PlayerShelter('FIRST'), PlayerShelter('SECOND') ] gs.active_player = gs.players[0] shelter = gs.active_player shelter.print = dumper_factory() shelter.input = helper_factory(['y', '1', '0', '0']) shelter.supplies = [Supply.LURE_OUT] shelter.zombies = [] gs.city_deck = [CityCard()] await counters.play_lure_out(gs) assert len(shelter.zombies) == 0 assert len(shelter.supplies) == 1 assert len(gs.city_deck) == 1 assert len(gs.supply_graveyard) == 0 assert len(tests.common.outputs) == 1
async def test_play_takeover(gs): gs.players = [ PlayerShelter('ZERO'), PlayerShelter('FIRST'), PlayerShelter('SECOND') ] gs.active_player = gs.players[0] shelter = gs.active_player shelter.print = dumper_factory() shelter.input = helper_factory(['0']) shelter.supplies = [Supply.TAKEOVER] gs.players[1].survivors = [CityCard()] gs.players[2].survivors = [CityCard()] await counters.play_takeover(gs) assert len(shelter.survivors) == 1 assert len(gs.players[1].survivors) == 0 assert gs.players[1].defeated assert len(gs.players[2].survivors) == 1 assert len(gs.supply_graveyard) == 1 assert len(tests.common.outputs) == 3
async def test_play_shotgun_one_lesser_one_big(gs, zombie, big_zombie): gs.active_player = PlayerShelter(print_foo=dumper_factory(), input_foo=helper_factory(['1'])) shelter = gs.active_player shelter.zombies = [big_zombie, zombie] shelter.supplies = [Supply.SHOTGUN] await weapons.play_shotgun(gs) assert len(shelter.zombies) == 1 assert zombie not in shelter.zombies assert len(shelter.supplies) == 0 assert len(gs.supply_graveyard) == 1 assert len(gs.city_graveyard) == 1 assert len(tests.common.outputs) == 3
def test_player_shelter_class(): shelter = PlayerShelter() assert shelter.name == '' assert not shelter.defeated assert len(shelter.zombies) == 0 assert len(shelter.supplies) == 0 assert len(shelter.obstacles) == 0 assert len(shelter.survivors) == 0 assert type(shelter.zombies) is list assert type(shelter.supplies) is list assert type(shelter.obstacles) is list assert type(shelter.survivors) is list
def test_get_winners(): gs = GameState() gs.players = [PlayerShelter('0'), PlayerShelter('1'), PlayerShelter('2')] gs.players[0].survivors = [CityCard(), CityCard(), CityCard()] gs.players[1].survivors = [CityCard()] gs.players[2].survivors = [CityCard(), CityCard(), CityCard()] winners = gs.get_winners() assert len(winners) == 2 assert winners == ['0', '2'] gs.players[2].survivors = [CityCard(), CityCard(), CityCard(), CityCard()] winners = gs.get_winners() assert len(winners) == 1 assert winners == ['2'] gs.players[0].survivors = [] gs.players[1].survivors = [] gs.players[2].survivors = [] winners = gs.get_winners() assert len(winners) == 0 assert winners == []
async def test_play_shotgun_many_lesser_one_big_wrong_input(gs, fast_zombie, zombie, big_zombie): gs.active_player = PlayerShelter(print_foo=dumper_factory(), input_foo=helper_factory(['6', 'rgdfrbw', '0'])) shelter = gs.active_player shelter.zombies = [fast_zombie, zombie, fast_zombie, big_zombie, fast_zombie, zombie, zombie] shelter.supplies = [Supply.SHOTGUN] await weapons.play_shotgun(gs) assert len(shelter.zombies) == 6 assert big_zombie not in shelter.zombies assert len(shelter.supplies) == 0 assert len(gs.supply_graveyard) == 1 assert len(gs.city_graveyard) == 1 assert len(tests.common.outputs) == 5
async def test_play_round_play_sacrifice_wit_last_survivor( zombie, fast_zombie): gs = GameState() gs.players = [PlayerShelter('0'), PlayerShelter('1'), PlayerShelter('2')] gs.players[1].survivors = [CityCard()] gs.players[2].survivors = [CityCard()] gs.active_player = gs.players[0] shelter = gs.active_player shelter.print = dumper_factory() shelter.input = helper_factory(['0', '1', '0']) shelter.zombies = [zombie, fast_zombie] shelter.survivors = [CityCard()] shelter.supplies = [Supply.SACRIFICE, Supply.BARRICADES, Supply.ALARM] await gs.play_round() assert len(shelter.supplies) == 0 assert len(shelter.zombies) == 0 assert len(shelter.survivors) == 0 assert shelter.defeated assert len(gs.city_graveyard) == 1 assert len(gs.supply_graveyard) == 3 assert gs.active_player != shelter
async def test_play_chainsaw(gs): gs.players = [ PlayerShelter('ZERO'), PlayerShelter('FIRST'), PlayerShelter('SECOND') ] gs.active_player = gs.players[0] gs.players[1].obstacles = [ Supply.ALARM, Supply.BARRICADES, Supply.BARRICADES ] gs.players[2].obstacles = [Supply.MINE_FILED] shelter = gs.active_player shelter.print = dumper_factory() shelter.input = helper_factory(['0']) shelter.supplies = [Supply.CHAINSAW] await counters.play_chainsaw(gs) assert len(shelter.supplies) == 0 assert len(gs.players[1].obstacles) == 0 assert len(gs.players[2].obstacles) == 1 assert len(gs.supply_graveyard) == 4 assert len(tests.common.outputs) == 3