|
|
# Copyright 2016-2019 Pedro M. Baeza <pedro.baeza@tecnativa.com># License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
from odoo.tests import commonfrom odoo.exceptions import UserError
class TestBaseLocationGeonamesImport(common.SavepointCase):
@classmethod def setUpClass(cls): super().setUpClass() cls.country = cls.env.ref('base.mc') cls.city = cls.env['res.city'].create({ 'name': 'Test city', 'country_id': cls.country.id, }) cls.wizard = cls.env['city.zip.geonames.import'].create({ 'country_id': cls.country.id, }) cls.wrong_country = cls.env['res.country'].create({ 'name': 'Wrong country', 'code': 'ZZYYXX', }) cls.wrong_wizard = cls.env['city.zip.geonames.import'].create({ 'country_id': cls.wrong_country.id, })
def test_import_country(self): max_import = 10 self.wizard.with_context(max_import=max_import).run_import() # Look if there are imported states for the country state_count = self.env['res.country.state'].search_count([ ('country_id', '=', self.country.id) ]) self.assertTrue(state_count) # Look if there are imported zips zip_count = self.env['res.city.zip'].search_count([ ('city_id.country_id', '=', self.country.id) ]) self.assertEqual(zip_count, max_import)
# Look if there are imported cities city_count = self.env['res.city'].search_count([ ('country_id', '=', self.country.id) ]) self.assertTrue(city_count)
# Reimport again to see that there's no duplicates self.wizard.with_context(max_import=max_import).run_import() state_count2 = self.env['res.country.state'].search_count([ ('country_id', '=', self.country.id) ]) self.assertEqual(state_count, state_count2)
city_count2 = self.env['res.city'].search_count([ ('country_id', '=', self.country.id) ]) self.assertEqual(city_count, city_count2)
zip_count = self.env['res.city.zip'].search_count([ ('city_id.country_id', '=', self.country.id) ]) self.assertEqual(zip_count, max_import)
def test_delete_old_entries(self): zip_entry = self.env['res.city.zip'].create({ 'name': 'Brussels', 'city_id': self.city.id, }) self.wizard.run_import() self.assertFalse(zip_entry.exists())
city_entry = self.env['res.city'].create({ 'name': 'Test city', 'country_id': self.country.id, }) self.wizard.run_import() self.assertFalse(city_entry.exists())
def test_import_title(self): self.wizard.letter_case = 'title' self.wizard.with_context(max_import=1).run_import() zip_entry = self.env['res.city.zip'].search( [('city_id.country_id', '=', self.country.id)], limit=1 ) self.assertEqual(zip_entry.city_id.name, zip_entry.city_id.name.title())
city = self.env['res.city'].search( [('country_id', '=', self.country.id)], limit=1 ) self.assertEqual(city.name, city.name.title())
def test_import_upper(self): self.wizard.letter_case = 'upper' self.wizard.with_context(max_import=1).run_import() zip_entry = self.env['res.city.zip'].search( [('city_id.country_id', '=', self.country.id)], limit=1 ) self.assertEqual(zip_entry.city_id.name, zip_entry.city_id.name.upper())
city = self.env['res.city'].search( [('country_id', '=', self.country.id)], limit=1 ) self.assertEqual(city.name, city.name.upper())
def test_download_error(self): """Check that we get an error when trying to download
with a wrong country code"""
with self.assertRaises(UserError): self.wrong_wizard.run_import()
def test_import_duplicated_city_name(self): country = self.env.ref('base.us') self.wizard.country_id = country.id parsed_csv = [ ['US', '95602', 'Auburn', ' California', 'CA', 'Placer', '61', '38.9829', '-121.0944', '4'], ['US', '95603', 'Auburn', ' California', 'CA', 'Placer', '61', '38.9115', '-121.08', '4'], ['US', '30011', 'Auburn', ' Georgia', 'GA', 'Barrow', '13', '34.0191', '-83.8261', '4'], ] self.wizard._process_csv(parsed_csv) cities = self.env['res.city'].search([('name', '=', 'Auburn')]) self.assertEqual(len(cities), 2) mapping = [ ['California', '95602'], ['California', '95603'], ['Georgia', '30011'], ] for state_name, zip_code in mapping: zip_entry = self.env['res.city.zip'].search([ ('city_id.country_id', '=', country.id), ('name', '=', zip_code), ]) state = self.env['res.country.state'].search([ ('country_id', '=', country.id), ('name', '=', state_name), ]) self.assertEqual( zip_entry.city_id.state_id, state, "Incorrect state for %s %s" % (state_name, zip_code), )
|
