class TestRest(unittest.TestCase):
def setUp(self):
self.warrior = Warrior()
self.rest = Rest(self.warrior)
def test_should_give_10_percent_health_back(self):
self.warrior.max_health = 20
self.warrior._health = 10
self.rest.perform()
self.assertEqual(12, self.warrior.health)
def test_should_not_add_health_when_at_max(self):
self.warrior.max_health = 20
self._health = 20
self.rest.perform()
self.assertEqual(20, self.warrior.health)
def test_should_not_go_over_max_health(self):
self.warrior.max_health = 20
self._health = 19
self.rest.perform()
self.assertEqual(20, self.warrior.health)
def test_should_say_something_if_at_full_health(self):
self.warrior.max_health = 20
self.warrior.say = mock.Mock()
self._health = 20
self.rest.perform()
self.assertTrue(self.warrior.say.called)
def setUp(self):
self.warrior = Warrior()
self.rest = Rest(self.warrior)
