Esempio n. 1
0
class applyEffect(unittest.TestCase):
    """ Test that applyEffect actually adds a leech effect """
    
    def setUp(self):
        """ Builds the delegate and pkmn for use in the tests """
        self.user = BuildPokemonBattleWrapper(pkmn = "CHARMANDER")
        self.target = BuildPokemonBattleWrapper(pkmn = "CHARMANDER")
        self.target2 = BuildPokemonBattleWrapper(pkmn = "BULBASAUR")
        
        self.delegate = LeechDelegate("", "", "FIRE")
        
    def appliesLeech(self):
        """ Tests if applyEffect applies the leech """
        self.target.secondaryEffects = []
        self.delegate.applyEffect(self.user, self.target, None)
        
        assert isinstance(self.target.secondaryEffects[0], Leech), "Target should have a Leech effect"
        assert self.target.secondaryEffects[0].source is self.user, "Leech should have the Pkmn as the Leech's Source"
        
    def immune(self):
        """ Tests if applyEffect applies the leech """
        self.target2.secondaryEffects = []
        self.delegate.applyEffect(self.user, self.target2, None)
        
        assert len(self.target2.secondaryEffects) == 0, "Should have no Leech effects when the target is immune"
Esempio n. 2
0
class hasThisLeech(unittest.TestCase):
    """ Test that hasThisLeech actually returns when it has that type of trap """
    
    def setUp(self):
        """ Builds the delegate and pkmn for use in the tests """
        self.wrapper = BuildPokemonBattleWrapper()
        self.delegate = LeechDelegate("", "", "FIRE")
        self.leech = Leech(None, "")
        self.otherLeech = Leech(None, "other.")
        
    def hasLeech(self):
        """ Tests if hasThisLeech returns true when there is an object of this leech """
        self.wrapper.secondaryEffects = [self.leech]
        
        assert self.delegate.hasThisLeech(self.wrapper), "Should have a leech effect"
        
    def noLeech(self):
        """ Tests if hasThisLeech returns false when there is no trap """
        self.wrapper.secondaryEffects = []
        
        assert not self.delegate.hasThisLeech(self.wrapper), "Should not have a leech effect"
        
    def otherLeech(self):
        """ Tests if hasThisLeech returns false when there is a different leech """
        self.wrapper.secondaryEffects = [self.otherLeech]
        
        assert not self.delegate.hasThisLeech(self.wrapper), "Should not have a leech effect if there is a different leech effect there"
Esempio n. 3
0
 def setUp(self):
     """ Builds the delegate and pkmn for use in the tests """
     self.user = BuildPokemonBattleWrapper(pkmn = "CHARMANDER")
     self.target = BuildPokemonBattleWrapper(pkmn = "CHARMANDER")
     self.target2 = BuildPokemonBattleWrapper(pkmn = "BULBASAUR")
     
     self.delegate = LeechDelegate("", "", "FIRE")
Esempio n. 4
0
class removePreviousLeech(unittest.TestCase):
    """ Test that removePreviousLeech actually removes a leech effect """
    
    def setUp(self):
        """ Builds the delegate and pkmn for use in the tests """
        self.wrapper = BuildPokemonBattleWrapper()
        self.delegate = LeechDelegate("", "", "FIRE")
        self.leech = Leech(None, "")
        
    def removeLeech(self):
        """ Tests if the leech is removed """
        self.wrapper.secondaryEffects = [self.leech]
        self.delegate.removePreviousLeech(self.wrapper)
        
        assert not self.leech in self.wrapper.secondaryEffects, "Should have removed original trap effect"
Esempio n. 5
0
 def setUp(self):
     """ Builds the delegate and pkmn for use in the tests """
     self.wrapper = BuildPokemonBattleWrapper()
     self.delegate = LeechDelegate("", "", "FIRE")
     self.leech = Leech(None, "")
     self.otherLeech = Leech(None, "other.")