def test_can_add_round(self):
"""
Test that:
1. We have not exceeded the total rounds for this battle
2. All combatants are alive
3. No combatant can attack the same target twice in a row
"""
with LogCapture():
logger = logging.getLogger()
battle = Battle(log=logger, total_rounds=1)
battlemaster = Battlemaster()
unit_charlie = self._make_unit(is_player=True, level=99)
unit_omega = self._make_unit(is_player=True, level=13)
battle.add_combatant(combatant=unit_charlie)
battle.add_combatant(combatant=unit_omega)
# Make sure battle was added
battlemaster.add_battle(battle=battle)
# Target's Scissors cut Attacker's Paper
attacker_weapon = self._make_item(item_type="lizard")
target_weapon = self._make_item(item_type="spock")
unit_charlie.equip_item(item=attacker_weapon)
unit_omega.equip_item(item=target_weapon)
"""
Sunny day scenario
"""
can_add_round = battle.can_add_round(attacker=unit_charlie,
target=unit_omega)
self.assertTrue(can_add_round)
"""
Test that we cannot exceed total rounds
"""
dungeon = self._make_dungeon()
battle.start_round(battle=battle,
ircutils="quux",
ircmsgs="quux",
irc="quux",
dungeon=dungeon)
self.assertEqual(len(battle.rounds), 1)
# Swap combatant order so it is someone else's turn
battle.combatants = list(reversed(battle.combatants))
cannot_exceed_rounds_error = battle.can_add_round(attacker=unit_charlie,
target=unit_omega)
self.assertEqual(cannot_exceed_rounds_error, "Cannot add round: maximum rounds reached.")
def test_add_battle_with_rounds(self):
combatant_1 = self._make_unit(level=13, is_player=False)
combatant_2 = self._make_unit(level=13, is_player=False)
with LogCapture():
logger = logging.getLogger()
battle = Battle(log=logger)
battle.add_combatant(combatant_1)
battle.add_combatant(combatant_2)
self.assertEqual(len(battle.combatants), 2)
battlemaster = Battlemaster()
battlemaster.add_battle(battle=battle)
self.assertEqual(len(battlemaster.battles), 1)
actual_1 = battlemaster.get_battle_by_combatant(combatant=combatant_1)
self.assertIsNotNone(actual_1)
self.assertEqual(battle, actual_1)
actual_2 = battlemaster.get_battle_by_combatant(combatant=combatant_2)
self.assertIsNotNone(actual_2)
self.assertEqual(battle, actual_2)
self.assertTrue(combatant_1.is_alive())
self.assertTrue(combatant_2.is_alive())
"""
Round 1
"""
attacker_weapon = self._make_item(item_type="rock")
target_weapon = self._make_item(item_type="scissors")
combatant_1.equip_item(item=attacker_weapon)
combatant_2.equip_item(item=target_weapon)
dungeon = self._make_dungeon()
battle.start_round(battle=battle,
irc="quux",
ircutils="quux",
ircmsgs="quux",
dungeon=dungeon)
self.assertEqual(len(battle.rounds), 1)
rounds_won_for_combatant_1 = battle.get_rounds_won(
combatant=combatant_1)
rounds_won_for_combatant_2 = battle.get_rounds_won(
combatant=combatant_2)
self.assertEqual(rounds_won_for_combatant_1, 1)
self.assertEqual(rounds_won_for_combatant_2, 0)
"""
Round 2
"""
round_2_attacker_weapon = self._make_item(item_type="paper")
round_2_target_weapon = self._make_item(item_type="rock")
combatant_2.equip_item(item=round_2_attacker_weapon)
combatant_1.equip_item(item=round_2_target_weapon)
battle.combatants = list(reversed(battle.combatants))
battle.start_round(battle=battle,
irc="quux",
ircutils="quux",
ircmsgs="quux",
dungeon=dungeon)
self.assertEqual(len(battle.rounds), 2)
rounds_won_for_combatant_1 = battle.get_rounds_won(
combatant=combatant_1)
rounds_won_for_combatant_2 = battle.get_rounds_won(
combatant=combatant_2)
self.assertEqual(rounds_won_for_combatant_1, 1)
self.assertEqual(rounds_won_for_combatant_2, 1)
"""
Round 3
"""
round_3_attacker_weapon = self._make_item(item_type="lizard")
round_3_target_weapon = self._make_item(item_type="spock")
combatant_1.equip_item(item=round_3_attacker_weapon)
combatant_2.equip_item(item=round_3_target_weapon)
battle.combatants = list(reversed(battle.combatants))
battle.start_round(battle=battle,
irc="quux",
ircutils="quux",
ircmsgs="quux",
dungeon=dungeon)
self.assertEqual(len(battle.rounds), 3)
rounds_won_for_combatant_1 = battle.get_rounds_won(
combatant=combatant_1)
rounds_won_for_combatant_2 = battle.get_rounds_won(
combatant=combatant_2)
self.assertEqual(rounds_won_for_combatant_1, 2)
self.assertEqual(rounds_won_for_combatant_2, 1)
"""
Make sure we can't exceed max rounds
"""
cannot_add_round_reason = battle.can_add_round(attacker=combatant_1,
target=combatant_2)
self.assertEqual(cannot_add_round_reason, "Cannot add round: maximum rounds reached.")
self.assertEqual(len(battle.rounds), 3)
def _start_player_battle_with_weapon(self, **kwargs):
"""
1. Get dungeon and player
2. Equip player with desired item type
3. Start battle and add combatants
4. Add round
"""
irc = kwargs["irc"]
player_user = kwargs["player_user"]
target_name = kwargs["target_name"]
weapon_type = kwargs["weapon_type"]
dungeon_info = self._get_dungeon_and_player(irc, player_user)
if dungeon_info is not None:
dungeon = dungeon_info["dungeon"]
player = dungeon_info["player"]
target_unit = dungeon.get_living_unit_by_name(target_name)
if target_unit is None:
irc.error("Invalid target")
return
equip_ok = player.equip_item_by_type(item_type=weapon_type)
if equip_ok is not None:
new_battle = SpiffyBattle(log=log)
can_start_round = new_battle.can_add_round(attacker=player,
target=target_unit)
if can_start_round is True:
"""
NPCs equip a random weapon that is not what the player
chose each battle. This could probably be moved somewhere
else. Perhaps into Battle?
"""
if target_unit.is_npc:
target_unit.equip_random_weapon(avoid_weapon_type=weapon_type)
"""
Find the last battle, or create a new one if there isn't
one already.
"""
bm = self.SpiffyWorld.battlemaster
battle = bm.get_battle_by_combatant(combatant=target_unit)
if battle is None:
battle = new_battle
battle.add_combatant(player)
battle.add_combatant(target_unit)
added_successfully = bm.add_battle(battle=battle)
"""
When a combatant is currently engaged in combat,
they cannot be added to another battle.
"""
if added_successfully is not True:
irc.error(added_successfully)
return
"""
Announce challenge accepted if this is a new
battle.
dungeon.announcer.challenge_accepted(attacker=player,
target=target_unit)
"""
battle.start_round(battle=battle,
irc=irc,
ircmsgs=ircmsgs,
dungeon=dungeon,
ircutils=ircutils)
else:
formatted_error \
= "Cannot start round {} vs {} because {}".format(player,
target_unit,
can_start_round)
log.error(formatted_error)
irc.error(can_start_round)
else:
irc.error("You can't equip that.")
else:
log.error("SpiffyWorld: Player %s not found!" % player_user)
irc.error("Player not found")
