def test_fireball(self):
battle_abilities.FIREBALL().use(self.messenger, self.attacker,
self.defender)
self.assertTrue(self.defender.health < self.defender.max_health)
self.assertFalse(self.attacker.context.damage_queue_fire)
self.assertTrue(self.defender.context.damage_queue_fire)
self.assertEqual(self.messenger.messages, ['hero_ability_fireball'])
class BattleAbilityFireball(BaseBattleAbility):
TYPE = relations.EFFECT.BATTLE_ABILITY_FIREBALL
ABILITY = ability = battle_abilities.FIREBALL(5)