def test_update_pre_existing_data(self): self.assertTrue(Country.objects.all().count() == 0) ua = Country.objects.create(name='Ukraine', code='UA') ru = Country.objects.create(name='Russia', code='RU') kemerovo = Region.objects.create(name='Кемеровская область', country=ru) City.objects.create(name='Березовский', id=1057, region=kemerovo) backend = IpGeobase() backend._update_geography(self.countries, self.regions, self.cities) self.assertEqual(set(Country.objects.all().values_list('code', flat=True)), self.countries) self.assertCountEqual(list(Region.objects.all().values('name', 'country__code')), self.regions) self.assertEqual(list(City.objects.all().values('name', 'id', 'region__name')), self.cities)
def test_update_geography_empty_data(self): command = IpGeobase() cities_info = command._update_geography(self.countries, self.regions, self.cities) self.assertEqual(set(Country.objects.all().values_list('code', flat=True)), self.countries) self.assertEqual(list(Region.objects.all().values('name', 'country__code')), self.regions) self.assertEqual(list(City.objects.all().values('name', 'id', 'region__name')), self.cities)
def test_update_geography_empty_data(self): command = IpGeobase() cities_info = command._update_geography( self.countries, self.regions, self.cities, self.cidr['city_country_mapping']) check_against_countries = [{ 'code': 'FR', 'name': 'France' }, { 'code': 'UA', 'name': 'Ukraine' }, { 'code': 'RU', 'name': 'Russian Federation' }] self.assertCountEqual(Country.objects.all().values('code', 'name'), check_against_countries) self.assertEqual( list(Region.objects.all().values('name', 'country__code')), self.regions) self.assertEqual( list(City.objects.all().values('name', 'id', 'region__name')), self.cities)
def test_update_pre_existing_data(self): self.assertTrue(Country.objects.all().count() == 0) ua = Country.objects.create(name='Ukraine', code='UA') ru = Country.objects.create(name='Russia', code='RU') kemerovo = Region.objects.create(name='Кемеровская область', country=ru) City.objects.create(name='Березовский', id=1057, region=kemerovo) backend = IpGeobase() backend._update_geography(self.countries, self.regions, self.cities) self.assertEqual( set(Country.objects.all().values_list('code', flat=True)), self.countries) self.assertCountEqual( list(Region.objects.all().values('name', 'country__code')), self.regions) self.assertEqual( list(City.objects.all().values('name', 'id', 'region__name')), self.cities)
def test_update_geography_empty_data(self): command = IpGeobase() cities_info = command._update_geography(self.countries, self.regions, self.cities) check_against_countries = [ {'code':'FR', 'name':'France'}, {'code':'UA', 'name':'Ukraine'}, {'code':'RU', 'name':'Russian Federation'} ] self.assertCountEqual(Country.objects.all().values('code', 'name'), check_against_countries) self.assertEqual(list(Region.objects.all().values('name', 'country__code')), self.regions) self.assertEqual(list(City.objects.all().values('name', 'id', 'region__name')), self.cities)