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
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_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_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
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 prepare_city_deck(self) -> None:
"""
Method used to prepare city deck for new game.
"""
horde = [CityCard(ZombieType.HORDE) for _ in range(2)]
zombie = [CityCard(ZombieType.ZOMBIE) for _ in range(23)]
fast = [CityCard(ZombieType.FAST) for _ in range(18)]
big = [CityCard(ZombieType.BIG) for _ in range(12)]
deck = horde + zombie + fast + big
shuffle(deck)
self.city_deck = deck
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_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_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
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
def test_city_card_ctor_exception():
thrown = False
try:
CityCard(ZombieType.SURVIVOR)
except Exception as ex:
assert str(ex) == 'Card cannot be init with survivor on top!'
thrown = True
assert thrown
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_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
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_defend_with_alarm(gs, shelter):
shelter.survivors = [CityCard()]
shelter.obstacles = [Supply.ALARM]
await defences.defend_with_alarm(gs)
assert len(tests.common.outputs) == 4
assert len(shelter.supplies) == 0
assert len(shelter.obstacles) == 0
assert len(gs.supply_graveyard) == 1
for hungry_zombie in shelter.zombies:
assert not hungry_zombie.active
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_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
def test_horde_show_up_two_zombies_left_in_city_no_graveyard():
gs = GameState()
gs.players = [PlayerShelter(), PlayerShelter(), PlayerShelter()]
gs.active_player = gs.players[1]
gs.city_deck = [
CityCard(ZombieType.HORDE),
CityCard(ZombieType.ZOMBIE),
CityCard(ZombieType.ZOMBIE)
]
gs.zombie_show_up()
assert len(gs.city_deck) == 0
assert len(gs.players[0].zombies) == 1
assert len(gs.players[1].zombies) == 0
assert len(gs.players[2].zombies) == 1
assert len(gs.active_player.zombies) == 0
gs = GameState()
gs.players = [PlayerShelter(), PlayerShelter(), PlayerShelter()]
gs.active_player = gs.players[2]
gs.city_deck = [
CityCard(ZombieType.HORDE),
CityCard(ZombieType.ZOMBIE),
CityCard(ZombieType.ZOMBIE)
]
gs.zombie_show_up()
assert len(gs.city_deck) == 0
assert len(gs.players[0].zombies) == 1
assert len(gs.players[1].zombies) == 1
assert len(gs.players[2].zombies) == 0
assert len(gs.active_player.zombies) == 0
def test_zombie_show_up():
gs = GameState()
gs.active_player = PlayerShelter()
gs.city_deck = [CityCard(ZombieType.ZOMBIE)]
gs.zombie_show_up()
assert gs.city_deck[0].top == ZombieType.ZOMBIE
assert gs.city_deck[0].bottom == ZombieType.SURVIVOR
assert len(gs.city_deck) == 1
gs.zombie_show_up()
assert len(gs.active_player.zombies) == 1
assert gs.active_player.zombies[0].top == ZombieType.ZOMBIE
assert len(gs.city_deck) == 0
gs.city_deck = [CityCard(ZombieType.BIG)]
gs.zombie_show_up()
assert gs.city_deck[0].top == ZombieType.BIG
assert gs.city_deck[0].bottom == ZombieType.SURVIVOR
assert len(gs.city_deck) == 1
gs.zombie_show_up()
assert len(gs.active_player.zombies) == 2
assert gs.active_player.zombies[1].top == ZombieType.BIG
assert len(gs.city_deck) == 0
async def test_defend_with_barricades(gs, shelter):
shelter.survivors = [CityCard()]
shelter.obstacles = [Supply.BARRICADES]
await defences.defend_with_barricades(gs)
assert len(tests.common.outputs) == 3
assert len(shelter.supplies) == 0
assert len(shelter.obstacles) == 0
assert len(gs.supply_graveyard) == 1
number_of_stopped_zombies = 0
for hungry_zombie in shelter.zombies:
if not hungry_zombie.active:
number_of_stopped_zombies += 1
assert number_of_stopped_zombies == 1
async def test_play_sacrifice(gs, fast_zombie, zombie, big_zombie):
gs.players = [
PlayerShelter('ZERO'),
PlayerShelter('FIRST'),
PlayerShelter('SECOND')
]
gs.active_player = gs.players[1]
shelter = gs.active_player
shelter.print = dumper_factory()
shelter.input = helper_factory(['0', '1', '0'])
shelter.zombies = [zombie, big_zombie, fast_zombie]
shelter.survivors = [CityCard(), CityCard()]
shelter.supplies = [Supply.SACRIFICE]
await counters.play_sacrifice(gs)
assert len(shelter.zombies) == 0
assert len(shelter.survivors) == 1
assert len(shelter.supplies) == 0
assert len(gs.players[0].zombies) == 2
assert len(gs.players[2].zombies) == 1
assert big_zombie in gs.players[2].zombies
assert len(gs.city_graveyard) == 1
assert len(gs.supply_graveyard) == 1
assert len(tests.common.outputs) == 2
async def test_play_sacrifice_when_one_rivals_defeated(gs, fast_zombie, zombie,
big_zombie):
gs.players = [
PlayerShelter('ZERO'),
PlayerShelter('FIRST'),
PlayerShelter('SECOND')
]
gs.players[0].defeated = True
gs.active_player = gs.players[1]
shelter = gs.active_player
shelter.print = dumper_factory()
shelter.zombies = [zombie, big_zombie, fast_zombie]
shelter.survivors = [CityCard(), CityCard()]
shelter.supplies = [Supply.SACRIFICE]
await counters.play_sacrifice(gs)
assert len(shelter.zombies) == 0
assert len(shelter.survivors) == 1
assert len(shelter.supplies) == 0
assert len(gs.players[0].zombies) == 0
assert len(gs.players[2].zombies) == 3
assert len(gs.city_graveyard) == 1
assert len(gs.supply_graveyard) == 1
assert len(tests.common.outputs) == 2
async def test_end_active_player_turn_zombies_and_alarm(
fast_zombie, zombie, big_zombie):
gs = GameState()
gs.players = [PlayerShelter(), PlayerShelter(), PlayerShelter()]
survivor_card = CityCard(ZombieType.ZOMBIE)
for player in gs.players:
player.survivors.append(survivor_card)
survivor_card.flip()
gs.city_deck.append(survivor_card)
gs.supply_deck = [Supply.AXE, Supply.ALARM, Supply.AXE, Supply.GUN]
gs.active_player = gs.players[2]
shelter = gs.active_player
shelter.obstacles = [Supply.ALARM, Supply.ALARM]
shelter.zombies = [fast_zombie, zombie, big_zombie]
shelter.print = dumper_factory()
shelter.input = helper_factory(['y'])
await gs.end_active_player_turn()
assert len(shelter.survivors) == 1
assert len(shelter.zombies) == 4
assert len(gs.city_deck) == 0
assert len(shelter.obstacles) == 1
assert len(gs.supply_graveyard) == 1
assert len(shelter.supplies) == 3
assert len(tests.common.outputs) == 7
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_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_radio(gs):
shelter = gs.active_player
gs.city_deck = [CityCard()]
shelter.supplies = [Supply.RADIO]
await summons.play_radio(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) == 3
gs.active_player = PlayerShelter(print_foo=dumper_factory())
shelter = gs.active_player
shelter.supplies = [Supply.RADIO]
await summons.play_radio(gs)
assert len(shelter.survivors) == 0
assert len(shelter.supplies) == 1
assert len(gs.city_deck) == 0
assert len(gs.supply_graveyard) == 1
assert len(tests.common.outputs) == 2
def test_horde_show_up():
gs = GameState()
gs.players = [PlayerShelter(), PlayerShelter(), PlayerShelter()]
gs.active_player = gs.players[1]
horde = ZombieType.HORDE
zombie = ZombieType.ZOMBIE
gs.city_deck = [
CityCard(horde),
CityCard(zombie),
CityCard(zombie),
CityCard(zombie),
CityCard(zombie),
CityCard(zombie),
CityCard(zombie),
CityCard(zombie)
]
gs.zombie_show_up()
assert len(gs.city_deck) == 4
assert len(gs.players[0].zombies) == 1
assert len(gs.players[1].zombies) == 1
assert len(gs.players[2].zombies) == 1
assert len(gs.active_player.zombies) == 1
async def test_play_megaphone(gs):
shelter = gs.active_player
gs.city_deck = [CityCard()]
shelter.supplies = [Supply.MEGAPHONE]
await summons.play_megaphone(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) == 4
gs.active_player = PlayerShelter(print_foo=dumper_factory())
shelter = gs.active_player
gs.city_deck = []
shelter.supplies = [Supply.MEGAPHONE]
await summons.play_megaphone(gs)
assert len(shelter.survivors) == 0
assert len(shelter.supplies) == 0
assert len(gs.city_deck) == 0
assert len(gs.supply_graveyard) == 2
assert len(tests.common.outputs) == 4
async def test_play_lure_only_big(gs, 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(['y', '1', '0'])
shelter.supplies = [Supply.LURE_OUT]
shelter.zombies = [big_zombie, big_zombie, big_zombie, big_zombie]
gs.city_deck = [zombie, CityCard()]
await counters.play_lure_out(gs)
assert len(shelter.zombies) == 3
assert len(gs.players[1].zombies) == 1
assert big_zombie in gs.players[1].zombies
assert len(gs.players[2].zombies) == 1
assert zombie in gs.players[2].zombies
assert len(shelter.supplies) == 0
assert len(gs.supply_graveyard) == 1
assert len(tests.common.outputs) == 4