mirror of
https://gitlab.com/flectra-community/l10n-switzerland-flectra.git
synced 2024-11-17 03:22:03 +00:00
259 lines
10 KiB
Python
259 lines
10 KiB
Python
# Copyright 2015 Yannick Vaucher, Camptocamp SA
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
|
|
|
|
import psycopg2
|
|
|
|
from flectra.exceptions import ValidationError
|
|
from flectra.tests import Form, common, tagged
|
|
from flectra.tools.misc import mute_logger
|
|
|
|
|
|
@tagged("post_install", "-at_install")
|
|
class TestBaseLocation(common.SavepointCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super().setUpClass()
|
|
state_obj = cls.env["res.country.state"]
|
|
city_obj = cls.env["res.city"]
|
|
zip_obj = cls.env["res.city.zip"]
|
|
cls.partner_obj = cls.env["res.partner"]
|
|
cls.env = cls.env(context=dict(cls.env.context, tracking_disable=True))
|
|
cls.state_vd = state_obj.create(
|
|
{"name": "Vaud", "code": "VD", "country_id": cls.env.ref("base.ch").id}
|
|
)
|
|
cls.env.ref("base.es").write({"enforce_cities": True})
|
|
cls.company = cls.env.ref("base.main_company")
|
|
cls.country_es = cls.env.ref("base.es")
|
|
cls.state_bcn = state_obj.create(
|
|
{"name": "Barcelona", "code": "08", "country_id": cls.country_es.id}
|
|
)
|
|
cls.state_madrid = state_obj.create(
|
|
{"name": "Madrid", "code": "28", "country_id": cls.env.ref("base.es").id}
|
|
)
|
|
cls.city_bcn = city_obj.create(
|
|
{
|
|
"name": "Barcelona",
|
|
"state_id": cls.state_bcn.id,
|
|
"country_id": cls.env.ref("base.es").id,
|
|
}
|
|
)
|
|
cls.city_madrid = city_obj.create(
|
|
{
|
|
"name": "Madrid",
|
|
"state_id": cls.state_madrid.id,
|
|
"country_id": cls.env.ref("base.es").id,
|
|
}
|
|
)
|
|
cls.city_lausanne = city_obj.create(
|
|
{
|
|
"name": "Lausanne",
|
|
"state_id": cls.state_vd.id,
|
|
"country_id": cls.env.ref("base.ch").id,
|
|
}
|
|
)
|
|
cls.lausanne = zip_obj.create({"name": "666", "city_id": cls.city_lausanne.id})
|
|
cls.barcelona = zip_obj.create({"name": "444", "city_id": cls.city_bcn.id})
|
|
|
|
def test_onchange_partner_city_completion(self):
|
|
"""Test that partner data is filled accodingly"""
|
|
partner1 = Form(self.env["res.partner"])
|
|
partner1.zip_id = self.barcelona
|
|
self.assertEqual(partner1.zip, self.barcelona.name)
|
|
self.assertEqual(partner1.city, self.barcelona.city_id.name)
|
|
self.assertEqual(partner1.state_id, self.barcelona.city_id.state_id)
|
|
self.assertEqual(partner1.country_id, self.barcelona.city_id.country_id)
|
|
|
|
def test_onchange_company_city_completion(self):
|
|
"""Test that company data is filled accodingly"""
|
|
company = self.env["res.company"].new({"name": "Test"})
|
|
company.zip_id = self.lausanne
|
|
company._onchange_zip_id()
|
|
self.assertEqual(company.zip, self.lausanne.name)
|
|
self.assertEqual(company.city, self.lausanne.city_id.name)
|
|
self.assertEqual(company.state_id, self.lausanne.city_id.state_id)
|
|
self.assertEqual(company.country_id, self.lausanne.city_id.country_id)
|
|
|
|
def test_company_address_fields(self):
|
|
"""Test if the partner address fields changes when
|
|
changing the ones from the company"""
|
|
company = self.env["res.company"].create({"name": "Test"})
|
|
self.assertTrue(company.partner_id)
|
|
company.partner_id.write(
|
|
{
|
|
"zip_id": self.lausanne.id,
|
|
"state_id": self.lausanne.city_id.state_id.id,
|
|
"country_id": self.lausanne.city_id.country_id.id,
|
|
"city_id": self.lausanne.city_id.id,
|
|
"city": self.lausanne.city_id.name,
|
|
"zip": self.lausanne.name,
|
|
}
|
|
)
|
|
company._compute_address()
|
|
self.assertEqual(company.zip_id, company.partner_id.zip_id)
|
|
self.assertEqual(company.city_id, company.partner_id.city_id)
|
|
|
|
def test_company_address_fields_inverse(self):
|
|
"""Test inverse fields from res.company"""
|
|
company = self.env["res.company"].create({"name": "Test"})
|
|
company.zip_id = self.barcelona.id
|
|
company._inverse_city_id()
|
|
company._inverse_zip_id()
|
|
self.assertEqual(company.zip_id, company.partner_id.zip_id)
|
|
self.assertEqual(company.city_id, company.partner_id.city_id)
|
|
|
|
def test_onchange_company_city_id_completion(self):
|
|
"""Test city auto-completion when changing zip in a company"""
|
|
company = self.env["res.company"].new({"name": "Test"})
|
|
company.zip_id = self.barcelona
|
|
company._onchange_zip_id()
|
|
self.assertEqual(company.city_id, self.barcelona.city_id)
|
|
|
|
def test_constrains_partner_01(self):
|
|
"""Test zip_id constraints"""
|
|
with self.assertRaises(ValidationError):
|
|
self.partner_obj.create(
|
|
{"name": "P1", "zip_id": self.barcelona.id, "state_id": False}
|
|
)
|
|
with self.assertRaises(ValidationError):
|
|
self.partner_obj.create(
|
|
{"name": "P1", "zip_id": self.barcelona.id, "country_id": False}
|
|
)
|
|
with self.assertRaises(ValidationError):
|
|
self.partner_obj.create(
|
|
{"name": "P1", "zip_id": self.barcelona.id, "city_id": False}
|
|
)
|
|
with self.assertRaises(ValidationError):
|
|
self.partner_obj.create(
|
|
{"name": "P1", "zip_id": self.barcelona.id, "zip": False}
|
|
)
|
|
|
|
def test_writing_company(self):
|
|
self.company.zip_id = self.barcelona
|
|
|
|
def test_constrains_partner_country(self):
|
|
"""Test partner country constraints"""
|
|
partner = self.partner_obj.create(
|
|
{
|
|
"name": "P1",
|
|
"zip_id": self.barcelona.id,
|
|
"country_id": self.barcelona.city_id.country_id.id,
|
|
"state_id": self.barcelona.city_id.state_id.id,
|
|
"city_id": self.barcelona.city_id.id,
|
|
}
|
|
)
|
|
|
|
with self.assertRaises(ValidationError):
|
|
partner.country_id = self.ref("base.ch")
|
|
|
|
def test_constrains_partner_state(self):
|
|
"""Test partner state constraints"""
|
|
partner = self.partner_obj.create(
|
|
{
|
|
"name": "P1",
|
|
"zip_id": self.barcelona.id,
|
|
"country_id": self.barcelona.city_id.country_id.id,
|
|
"state_id": self.barcelona.city_id.state_id.id,
|
|
"city_id": self.barcelona.city_id.id,
|
|
}
|
|
)
|
|
|
|
with self.assertRaises(ValidationError):
|
|
partner.state_id = self.state_vd.id
|
|
|
|
def test_constrains_partner_city(self):
|
|
"""Test partner city constraints"""
|
|
partner = self.partner_obj.create(
|
|
{
|
|
"name": "P1",
|
|
"zip_id": self.barcelona.id,
|
|
"country_id": self.barcelona.city_id.country_id.id,
|
|
"state_id": self.barcelona.city_id.state_id.id,
|
|
"city_id": self.barcelona.city_id.id,
|
|
}
|
|
)
|
|
|
|
with self.assertRaises(ValidationError):
|
|
partner.city_id = self.city_lausanne
|
|
|
|
def test_partner_onchange_country(self):
|
|
"""Test partner onchange country_id"""
|
|
partner = self.partner_obj.new({"name": "TEST", "zip_id": self.lausanne.id})
|
|
partner.country_id = self.env.ref("base.es")
|
|
partner._onchange_country_id()
|
|
self.assertFalse(partner.zip_id)
|
|
|
|
def test_partner_onchange_city(self):
|
|
"""Test partner onchange city_id"""
|
|
partner = Form(self.env["res.partner"])
|
|
partner.zip_id = self.lausanne
|
|
self.city_bcn.country_id.enforce_cities = False
|
|
partner.city_id = self.city_bcn
|
|
self.assertFalse(partner.zip_id)
|
|
|
|
def test_partner_onchange_state(self):
|
|
"""Test partner onchange state_id"""
|
|
partner = Form(self.env["res.partner"])
|
|
partner.zip_id = self.lausanne
|
|
partner.state_id = self.state_bcn
|
|
self.assertFalse(partner.zip_id)
|
|
self.assertEqual(partner.country_id, self.country_es)
|
|
|
|
def test_company_onchange_state(self):
|
|
"""Test company onchange state_id"""
|
|
self.company.state_id = self.state_bcn
|
|
self.company._onchange_state_id()
|
|
self.assertEqual(self.company.country_id, self.company.state_id.country_id)
|
|
|
|
def test_display_name(self):
|
|
"""Test if the display_name is stored and computed properly"""
|
|
self.assertEqual(
|
|
self.lausanne.display_name,
|
|
"666, Lausanne, Vaud, " + self.browse_ref("base.ch").name,
|
|
)
|
|
|
|
def test_name_search(self):
|
|
"""Test that zips can be searched through both the name of the
|
|
city or the zip code"""
|
|
madrid_data = {"city_id": self.city_madrid.id, "name": "555"}
|
|
|
|
madrid = self.env["res.city.zip"].create(madrid_data)
|
|
|
|
found_recs = self.env["res.city.zip"].name_search(name="444")
|
|
self.assertEqual(len(found_recs), 1)
|
|
self.assertEqual(found_recs[0][0], self.barcelona.id)
|
|
found_recs = self.env["res.city.zip"].name_search(name="Barcelona")
|
|
self.assertEqual(len(found_recs), 1)
|
|
self.assertEqual(found_recs[0][0], self.barcelona.id)
|
|
|
|
found_recs = self.env["res.city.zip"].name_search(name="555")
|
|
self.assertEqual(len(found_recs), 1)
|
|
self.assertEqual(found_recs[0][0], madrid.id)
|
|
found_recs = self.env["res.city.zip"].name_search(name="Madrid")
|
|
self.assertEqual(len(found_recs), 1)
|
|
self.assertEqual(found_recs[0][0], madrid.id)
|
|
|
|
found_recs = self.env["res.city.zip"].name_search(name="666")
|
|
self.assertEqual(len(found_recs), 1)
|
|
self.assertEqual(found_recs[0][0], self.lausanne.id)
|
|
found_recs = self.env["res.city.zip"].name_search(name="Lausanne")
|
|
self.assertEqual(len(found_recs), 1)
|
|
self.assertEqual(found_recs[0][0], self.lausanne.id)
|
|
|
|
def test_zip_ql_constraints(self):
|
|
"""Test UNIQUE name within it's area for zips"""
|
|
with self.assertRaises(psycopg2.IntegrityError), mute_logger("flectra.sql_db"):
|
|
self.env["res.city.zip"].create(
|
|
{"name": "666", "city_id": self.city_lausanne.id}
|
|
)
|
|
|
|
def test_city_sql_contraint(self):
|
|
"""Test UNIQUE name within it's area for cities"""
|
|
with self.assertRaises(psycopg2.IntegrityError), mute_logger("flectra.sql_db"):
|
|
self.env["res.city"].create(
|
|
{
|
|
"name": "Barcelona",
|
|
"state_id": self.state_bcn.id,
|
|
"country_id": self.ref("base.es"),
|
|
}
|
|
)
|