def test_country(self, mock_result_to_region, mock_result_to_country): # If country in results, builds a Country object, passes result to result_to_region result = { 'country': {'id': 'mapbox_id', 'name': 'Petoria'}, } mock_result_to_country.return_value = CountryFactory.create(mapbox_id='mapbox_id', name='Petoria') mock_result_to_region.return_value = None country, region, city = result_to_country_region_city(result) ok_(region is None) ok_(city is None) ok_(country is not None) ok_(isinstance(country, Country)) eq_('mapbox_id', country.mapbox_id) eq_('Petoria', country.name) mock_result_to_region.assert_called_with(result, country)
def test_country(self, mock_result_to_region, mock_result_to_country): # If country in results, builds a Country object, passes result to result_to_region result = { 'country': { 'id': 'mapbox_id', 'name': 'Petoria' }, } mock_result_to_country.return_value = CountryFactory.create( mapbox_id='mapbox_id', name='Petoria') mock_result_to_region.return_value = None country, region, city = result_to_country_region_city(result) ok_(region is None) ok_(city is None) ok_(country is not None) ok_(isinstance(country, Country)) eq_('mapbox_id', country.mapbox_id) eq_('Petoria', country.name) mock_result_to_region.assert_called_with(result, country)
def test_no_country(self, mock_result_to_country): # If result_to_country returns None, None, None mock_result_to_country.return_value = None eq_((None, None, None), result_to_country_region_city({'foo': 1}))