def test_stormy_weather(self):
airport = Airport()
plane = Mock()
weather = Mock(stormy=True)
airport.planes = [plane]
self.assertEqual(airport.release_plane(plane, weather),
"Weather is Stormy")
def test_airport_take_off_plane(self):
airport = Airport()
plane = Mock()
weather = Mock()
weather.stormy.return_value = False
airport.planes = [plane]
airport.release_plane(plane, weather)
plane.take_off.assert_called_once_with()
def test_airport_take_off_plane(self):
airport = Airport()
plane = Mock()
weather = Mock()
weather.stormy.return_value = False
airport.planes = [plane]
airport.release_plane(plane,weather)
plane.take_off.assert_called_once_with()
def test_stormy_weather(self):
airport = Airport()
plane = Mock()
weather = Mock(stormy=True)
airport.planes = [plane]
self.assertEqual(airport.release_plane(plane,weather), "Weather is Stormy")
