class attemptAfterTurn(unittest.TestCase):
    """ Test cases of attemptAfterTurn """
    
    def  setUp(self):
        """ Build the Pkmn and AfterTurnEffect for the test """
        self.effect = AfterTurnEffect()
        self.effect.faintHandler = BuildFaintHandler("USER")
        self.pkmn = BuildPokemonBattleWrapper()
        
    def fainted(self):
        """ Test that the Effect is not performed when the pkmn has fainted """
        self.pkmn.faint()
        messages = self.effect.attemptAfterTurn(self.pkmn)
        assert messages == [], "Should receive no messages since nothing was performed"
        
    def notFainted(self):
        """ Test that the Effect is performed when the pkmn has fainted """
        messages = self.effect.attemptAfterTurn(self.pkmn)
        assert messages == [AfterTurnEffect.message], "Should receive messages from afterTurn function"
class attemptAfterTurn(unittest.TestCase):
    """ Test cases of attemptAfterTurn """
    def setUp(self):
        """ Build the Pkmn and AfterTurnEffect for the test """
        self.effect = AfterTurnEffect()
        self.effect.faintHandler = BuildFaintHandler("USER")
        self.pkmn = BuildPokemonBattleWrapper()

    def fainted(self):
        """ Test that the Effect is not performed when the pkmn has fainted """
        self.pkmn.faint()
        messages = self.effect.attemptAfterTurn(self.pkmn)
        assert messages == [], "Should receive no messages since nothing was performed"

    def notFainted(self):
        """ Test that the Effect is performed when the pkmn has fainted """
        messages = self.effect.attemptAfterTurn(self.pkmn)
        assert messages == [
            AfterTurnEffect.message
        ], "Should receive messages from afterTurn function"
 def  setUp(self):
     """ Build the Pkmn and AfterTurnEffect for the test """
     self.effect = AfterTurnEffect()
     self.effect.faintHandler = BuildFaintHandler("USER")
     self.pkmn = BuildPokemonBattleWrapper()
 def setUp(self):
     """ Build the Pkmn and AfterTurnEffect for the test """
     self.effect = AfterTurnEffect()
     self.effect.faintHandler = BuildFaintHandler("USER")
     self.pkmn = BuildPokemonBattleWrapper()