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'])