def setUp(self):
self.floor = Floor()
self.floor.width = 2
self.floor.height = 3
self.captive = Captive()
self.floor.add(self.captive, 0, 0)
self.explode = Explode(self.captive)
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.captive = Captive()
def setUp(self):
super(TestWithCaptive, self).setUp()
self.captive = Captive()
self.floor.add(self.captive, 0, 0)
self.space = self.floor.space(0, 0)
