# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html). # Copyright 2015 Grupo ESOC Ingeniería de Servicios, S.L.U. # Copyright 2015 Antiun Ingenieria S.L. - Antonio Espinosa from flectra.tests.common import TransactionCase from flectra.addons.partner_firstname.tests.base import MailInstalled class CompanyCase(TransactionCase): """Test ``res.partner`` when it is a company.""" def setUp(self): super(CompanyCase, self).setUp() self.env["ir.config_parameter"].set_param("partner_names_order", "first_last") def tearDown(self): try: new = self.env["res.partner"].create( {"is_company": True, "name": self.name} ) # Name should be cleaned of unneeded whitespace clean_name = " ".join(self.name.split(None)) # Check it's saved OK self.assertEqual(new.name, clean_name, "Saved company name is wrong.") # Check it's saved in the lastname self.assertEqual( new.lastname, clean_name, "Company name should be saved in the lastname field.", ) # Check that other fields are empty self.assertEqual( new.firstname, False, "Company first name must always be empty." ) self.assertEqual( new.lastname2, False, "Company last name 2 must always be empty." ) finally: super(CompanyCase, self).tearDown() def test_long_name(self): """Create a company with a long name.""" self.name = "Söme very lóng nâme" def test_short_name(self): """Create a company with a short name.""" self.name = "Shoŕt" def test_whitespace_before(self): """Create a company with name prefixed with whitespace.""" self.name = " Wĥitespace befòre" def test_whitespace_after(self): """Create a company with name suffixed with whitespace.""" self.name = "Whitespâce aftér " def test_whitespace_inside(self): """Create a company with whitespace inside the name.""" self.name = "Whitespacé ïnside" def test_whitespace_everywhere(self): """Create a company with whitespace everywhere in the name.""" self.name = " A lot öf whitespace " class PersonCase(TransactionCase): """Test ``res.partner`` when it is a person.""" model = "res.partner" context = dict() def setUp(self): super(PersonCase, self).setUp() self.env["ir.config_parameter"].set_param( "partner_names_order", "last_first_comma" ) self.firstname = "Fírstname" self.lastname = "Làstname1" self.lastname2 = "Lâstname2" self.template = "%(last1)s %(last2)s, %(first)s" def tearDown(self): try: new = self.env[self.model].with_context(self.context).create(self.params) # Check that each individual field matches self.assertEqual(self.firstname, new.firstname, "First name saved badly.") self.assertEqual(self.lastname, new.lastname, "Last name 1 saved badly.") self.assertEqual(self.lastname2, new.lastname2, "Last name 2 saved badly.") # Check that name gets saved fine self.assertEqual( self.template % ( { "last1": self.lastname, "last2": self.lastname2, "first": self.firstname, } ), new.name, "Name saved badly.", ) finally: super(PersonCase, self).tearDown() def test_firstname_first(self): """Create a person setting his first name first.""" self.env["ir.config_parameter"].set_param("partner_names_order", "first_last") self.template = "%(first)s %(last1)s %(last2)s" self.params = { "is_company": False, "name": "{} {} {}".format(self.firstname, self.lastname, self.lastname2), } def test_firstname_last(self): """Create a person setting his first name last.""" self.params = { "is_company": False, "name": "{} {}, {}".format(self.lastname, self.lastname2, self.firstname), } def test_firstname_last_wo_comma(self): """Create a person setting his first name last and the order as 'last_first'""" self.env["ir.config_parameter"].set_param("partner_names_order", "last_first") self.template = "%(last1)s %(last2)s %(first)s" self.params = { "is_company": False, "name": "{} {} {}".format(self.lastname, self.lastname2, self.firstname), } def test_firstname_only(self): """Create a person setting his first name only.""" self.env["ir.config_parameter"].set_param("partner_names_order", "first_last") self.firstname = self.lastname2 = False self.template = "%(last1)s" self.params = { "is_company": False, "name": self.lastname, } def test_firstname_lastname_only(self): """Create a person setting his first name and last name 1 only.""" self.env["ir.config_parameter"].set_param("partner_names_order", "first_last") self.lastname2 = False self.template = "%(first)s %(last1)s" self.params = { "is_company": False, "name": "{} {}".format(self.firstname, self.lastname), } def test_lastname_firstname_only(self): """Create a person setting his last name 1 and first name only.""" self.lastname2 = False self.template = "%(last1)s, %(first)s" self.params = { "is_company": False, "name": "{}, {}".format(self.lastname, self.firstname), } def test_lastname_firstname_only_wo_comma(self): """Create a person setting his last name 1 and first name only. Set order to 'last_first' to test name split without comma""" self.env["ir.config_parameter"].set_param("partner_names_order", "last_first") self.lastname2 = False self.template = "%(last1)s %(first)s" self.params = { "is_company": False, "name": "{} {}".format(self.lastname, self.firstname), } def test_separately(self): """Create a person setting separately all fields.""" self.params = { "is_company": False, "firstname": self.firstname, "lastname": self.lastname, "lastname2": self.lastname2, } class UserCase(PersonCase, MailInstalled): """Test ``res.users``.""" model = "res.users" context = {"default_login": "user@example.com"} def tearDown(self): # Skip if ``mail`` is installed if not self.mail_installed(): super(UserCase, self).tearDown() class TestRecalculateNames(TransactionCase): def setUp(self): super().setUp() self.config_settings = self.env["res.config.settings"].create({}) def test_recalculate_names(self): firstname = "Xavier De Jesús" lastname = "Payen" lastname2 = "Sandoval" correct_names = { "first_last": f"{firstname} {lastname} {lastname2}", "last_first": f"{lastname} {lastname2} {firstname}", "last_first_comma": f"{lastname} {lastname2}, {firstname}", } partner = self.env["res.partner"].create( { "firstname": firstname, "lastname": lastname, "lastname2": lastname2, } ) for order in correct_names: self.config_settings.partner_names_order = order self.config_settings.action_recalculate_partners_name() self.assertEqual(partner.name, correct_names[order]) self.assertEqual(partner.firstname, firstname) self.assertEqual(partner.lastname, lastname) self.assertEqual(partner.lastname2, lastname2)