def take_turn(self, player: Player, game: Game, sink: CommandsSink):
self.game = game
if not self.possible_actions:
self.possible_actions = list(sink.possible_actions)
if sink.possible_actions:
sink.execute(sink.possible_actions[0])
else:
sink.end_turn()
def test_sink_is_done_when_end_turn_called(game):
# arrange
player = game.add_player('Player', char=Character.Assassin)
# act
sink = CommandsSink(player, game)
sink.execute(sink.possible_actions[0])
assert sink.can_end_turn
sink.end_turn()
# assert
assert sink.done
def take_turn(self, player: Player, game: Game, sink: CommandsSink):
""" Should execute commands via sink """
command = random.choice(list(sink.all_possible_commands))
if isinstance(command, InteractiveCommand):
while command.choices(player, game):
command.select(random.choice(command.choices(player, game)))
assert command.ready
if command:
sink.execute(command)
else:
sink.end_turn()
def test_warlord_abilities(game):
# arrange
player = game.add_player('Player', char=Character.Warlord)
victim = game.add_player('Victim', char=Character.King)
victim.build_district(District.Docks)
# assert
sink = CommandsSink(player, game)
assert not sink.possible_abilities
sink.execute(sink.possible_actions[0])
assert sink.possible_abilities == (commands.Destroy(), )
def test_possible_build(game):
# arrange
player = game.add_player('Player',
char=Character.King,
hand=[District.Tavern, District.Fortress])
player.cash_in(2)
# assert
sink = CommandsSink(player, game)
assert not sink.possible_builds
sink.execute(sink.possible_actions[0])
assert sink.possible_builds == (commands.Build(), )
def test_income_must_be_taken_after_action(game):
# arrange
player = game.add_player('Player',
char=Character.Warlord,
hand=[District.Watchtower])
player.cash_in(10)
# act
sink = CommandsSink(player, game)
sink.execute(sink.possible_actions[0])
# assert
assert sink.possible_builds
def test_architect_ability(game):
# arrange
player = game.add_player('Player',
char=Character.Architect,
hand=[District.Watchtower])
# act
sink = CommandsSink(player, game)
sink.execute(
next(
iter(action for action in sink.possible_actions
if isinstance(action, commands.CashIn))))
# assert
assert len(player.hand) == 3
def test_merchant_ability(game):
# arrange
player = game.add_player('Player', char=Character.Merchant)
player.cash_in(1)
# act
sink = CommandsSink(player, game)
draw_cards = next(
iter(action for action in sink.possible_actions
if isinstance(action, commands.DrawSomeCards)))
draw_cards.select(draw_cards.choices(player, game)[0])
sink.execute(draw_cards)
# assert
assert player.gold == 2
def test_income_is_taken_automatically_on_end_turn(game):
# arrange
player = game.add_player('Player',
char=Character.Warlord,
city=[District.Prison],
hand=[District.Watchtower])
sink = CommandsSink(player, game)
draw_cards = next(command for command in sink.possible_actions
if isinstance(command, commands.DrawSomeCards))
draw_cards.select(next(iter(draw_cards.choices(player, game))))
sink.execute(draw_cards)
# act
sink.end_turn()
# assert
assert player.gold == 1
def test_destroy_lead_player_district(bot, game, warlord, king, thief):
# arrange
for _ in range(6):
king.build_district(District.Docks)
for _ in range(7):
thief.build_district(District.Docks)
context = bot.create_context(warlord, game)
sink = CommandsSink(warlord, game)
sink.execute(sink.possible_actions[0])
assert sink.possible_abilities
# act
command = bot.destroy(sink.possible_abilities, context, warlord, game)
# assert
assert command == commands.Destroy(target=thief, card=District.Docks)
def test_warlord_ability_is_final(game):
# arrange
hand = [District.Tavern, District.Fortress]
city = [District.Watchtower]
player = game.add_player('Player',
char=Character.Warlord,
hand=hand,
city=city)
player.cash_in(2)
# act
sink = CommandsSink(player, game)
sink.execute(sink.possible_actions[0])
destroy = sink.possible_abilities[0]
while destroy.choices(player, game):
destroy.select(destroy.choices(player, game)[0])
sink.execute(destroy)
# assert
assert not sink.possible_builds
def take_turn(self, player: Player, game: Game, sink: CommandsSink):
if sink.possible_actions:
sink.execute(sink.possible_actions[0])
else:
sink.end_turn()
def take_turn(self, player: Player, game: Game, sink: CommandsSink):
command = self.decide(player, game, sink)
if command:
sink.execute(command)
else:
sink.end_turn()