class OpenWeatherMapTestCase(TestCase):
def setUp(self):
self.open_weather_map = OpenWeatherMap()
@patch("requests.get")
def test_failure(self, mock_get):
response = requests.Response()
response.status_code = requests.codes.bad_request
mock_get.return_value = response
forecast = self.open_weather_map.get_forecast()
self.assertEqual({}, forecast)
@patch("requests.get")
def test_success(self, mock_get):
response = requests.Response()
response.status_code = requests.codes.ok
response._content = b'{"city": {"name": "Sydney"}}'
mock_get.return_value = response
forecast = self.open_weather_map.get_forecast()
self.assertEqual("Sydney", forecast['city']['name'])
def setUp(self): self.open_weather_map = OpenWeatherMap()
