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
