class TestRescue(unittest.TestCase): def setUp(self): self.warrior = Warrior() self.rescue = Rescue(self.warrior) def test_should_rescue_captive(self): captive = Captive() captive.position = mock.Mock() mock_space = mock.Mock() mock_space.is_captive.return_value = True self.rescue.space = mock.Mock(return_value=mock_space) self.rescue.unit = mock.Mock(return_value=captive) self.warrior.earn_points = mock.Mock() self.rescue.perform() self.assertEqual(None, captive.position) self.warrior.earn_points.assert_called_once_with(20)
def setUp(self): self.warrior = Warrior() self.rescue = Rescue(self.warrior)