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)
