class applyEffect(unittest.TestCase):
""" Test cases of applyEffect """
def setUp(self):
""" Build the Pkmn, Abilities and Effect for the test """
self.user = BuildPokemonBattleWrapper()
self.target = BuildPokemonBattleWrapper()
self.userAbility = Ability(None)
self.targetAbility = Ability(None)
self.user.setAbility(self.userAbility)
self.target.setAbility(self.targetAbility)
self.delegate = SwapAbilityDelegate()
def swapped(self):
""" Test that the abilities are swapped """
self.delegate.applyEffect(self.user, self.target, None)
assert self.user.getAbility() is self.targetAbility, "User should have target's ability"
assert self.target.getAbility() is self.userAbility, "Target should have user's ability"
def message(self):
""" Test message is returned correctly """
messages = self.delegate.applyEffect(self.user, self.target, None)
assert messages == [SwapAbilityDelegate.message % (self.user.getHeader(), self.target.getHeader())], "Should have the Effect's message"
def setUp(self):
""" Build the Pkmn, Abilities and Effect for the test """
self.user = BuildPokemonBattleWrapper()
self.target = BuildPokemonBattleWrapper()
self.userAbility = Ability(None)
self.targetAbility = Ability(None)
self.user.setAbility(self.userAbility)
self.target.setAbility(self.targetAbility)
self.delegate = SwapAbilityDelegate()
