def test_geocode_returns_invalid_format(self, mock):
"""Test geocode_location returns None if the response is unusual."""
mock_geocoder = Mock()
mock_geocoder.geocode.return_value = "this isn't right!"
mock.return_value = mock_geocoder
self.assertEqual(
location.geocode_location('some location'),
(None, None)
)
def test_geocode_location_geocoder_raises_exception(self, mock):
"""Test that geocode_location returns None if there's an exception."""
exception = ValueError('something went wrong')
mock_geocoder = Mock()
mock_geocoder.geocode.side_effect = exception
mock.return_value = mock_geocoder
self.assertEqual(
location.geocode_location('some location'),
(None, None)
)
def test_geocode_location(self, mock):
"""Test that a location is geocoded correctly."""
mock_geocoder = Mock()
mock_geocoder.geocode.return_value = [
(u'North Desplaines St & West Fulton Mkt, Chicago, IL 60661, USA',
(41.8868014, -87.6442893))
]
mock.return_value = mock_geocoder
self.assertEqual(
location.geocode_location('something'),
(41.8868014, -87.6442893)
)