def test_rifle(self):
# setup
player = Player("joe")
monster = Mob("darf")
starting_monster_hp = monster.hp
# player shoots monster with rifle
player.perform_attack(attacks['rifle'], [monster])
# assert the monster was hit and the weapon must cooldown
self.assertEqual(monster.hp, starting_monster_hp - attacks['rifle']['damage'])
self.assertNotIn('rifle', player.available_attacks)
def test_grenade(self):
# setup
player = Player("joe")
monster1 = Mob("darf")
starting_monster1_hp = monster1.hp
monster2 = Mob("weebo")
starting_monster2_hp = monster2.hp
# player throws grenade at 2 monsters
player.perform_attack(attacks['grenade'], [monster1, monster2])
# assert both monsters were hit and the weapon must cooldown
self.assertEqual(monster1.hp, starting_monster1_hp - attacks['grenade']['damage'])
self.assertEqual(monster2.hp, starting_monster2_hp - attacks['grenade']['damage'])
self.assertNotIn('grenade', player.available_attacks)
