def import_file(self, json_file, user):
countries_json = json.loads(json_file.read())
all_countries_codes = countries.countries
for country in countries_json:
name = countries_json.get(country, None)
country = country.upper()
if name and country in all_countries_codes:
CountryAlias.get_or_create(country, name, user)
def import_file(self, json_file, user):
countries_json = json.loads(json_file.read())
all_countries_codes = countries.countries
for country in countries_json:
name = countries_json.get(country, None)
country = country.upper()
if name and country in all_countries_codes:
CountryAlias.get_or_create(country, name, user)
def test_country_alias_get_or_create(self):
self.assertFalse(CountryAlias.objects.all())
alias1 = CountryAlias.get_or_create('RW', 'Awesome', self.admin)
self.assertEqual(CountryAlias.objects.all().count(), 1)
self.assertEqual(CountryAlias.objects.all().first(), alias1)
alias2 = CountryAlias.get_or_create('RW', 'Awesome', self.admin)
self.assertEqual(CountryAlias.objects.all().count(), 1)
self.assertEqual(CountryAlias.objects.all().first(), alias1)
self.assertEqual(alias1, alias2)
alias3 = CountryAlias.get_or_create('RW', '1kHills', self.admin)
self.assertEqual(CountryAlias.objects.all().count(), 2)
def test_country_alias_get_or_create(self):
self.assertFalse(CountryAlias.objects.all())
alias1 = CountryAlias.get_or_create('RW', 'Awesome', self.admin)
self.assertEqual(CountryAlias.objects.all().count(), 1)
self.assertEqual(CountryAlias.objects.all().first(), alias1)
alias2 = CountryAlias.get_or_create('RW', 'Awesome', self.admin)
self.assertEqual(CountryAlias.objects.all().count(), 1)
self.assertEqual(CountryAlias.objects.all().first(), alias1)
self.assertEqual(alias1, alias2)
CountryAlias.get_or_create('RW', '1kHills', self.admin)
self.assertEqual(CountryAlias.objects.all().count(), 2)
def test_country_alias_get_or_create(self):
self.assertFalse(CountryAlias.objects.all())
alias1 = CountryAlias.get_or_create("RW", "Awesome", self.admin)
self.assertEqual(CountryAlias.objects.all().count(), 1)
self.assertEqual(CountryAlias.objects.all().first(), alias1)
alias2 = CountryAlias.get_or_create("RW", "Awesome", self.admin)
self.assertEqual(CountryAlias.objects.all().count(), 1)
self.assertEqual(CountryAlias.objects.all().first(), alias1)
self.assertEqual(alias1, alias2)
CountryAlias.get_or_create("RW", "1kHills", self.admin)
self.assertEqual(CountryAlias.objects.all().count(), 2)
def test_list(self):
list_url = reverse('countries.countryalias_list')
response = self.client.get(list_url, SERVER_NAME='nigeria.ureport.io')
self.assertLoginRedirect(response)
self.login(self.superuser)
response = self.client.get(list_url, SERVER_NAME='nigeria.ureport.io')
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.context['object_list']), 0)
alias1 = CountryAlias.get_or_create('RW', 'Awesome', self.admin)
response = self.client.get(list_url, SERVER_NAME='nigeria.ureport.io')
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.context['object_list']), 1)
self.assertContains(response, "Rwanda (RW)")
self.assertTrue(alias1 in response.context['object_list'])
def test_list(self):
list_url = reverse('countries.countryalias_list')
response = self.client.get(list_url, SERVER_NAME='nigeria.ureport.io')
self.assertLoginRedirect(response)
self.login(self.superuser)
response = self.client.get(list_url, SERVER_NAME='nigeria.ureport.io')
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.context['object_list']), 0)
alias1 = CountryAlias.get_or_create('RW', 'Awesome', self.admin)
response = self.client.get(list_url, SERVER_NAME='nigeria.ureport.io')
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.context['object_list']), 1)
self.assertContains(response, "Rwanda (RW)")
self.assertTrue(alias1 in response.context['object_list'])
