mirror of
https://gitlab.com/flectra-community/partner-contact.git
synced 2024-12-24 13:21:48 +00:00
65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
|
# Copyright 2015 Grupo ESOC Ingeniería de Servicios, S.L. - Jairo Llopis.
|
||
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
|
||
|
|
||
|
"""Test default values for models."""
|
||
|
|
||
|
from flectra.tests.common import TransactionCase
|
||
|
|
||
|
from .base import MailInstalled
|
||
|
|
||
|
|
||
|
class PersonCase(TransactionCase):
|
||
|
"""Test ``res.partner`` when it is a person."""
|
||
|
|
||
|
context = {"default_is_company": False}
|
||
|
model = "res.partner"
|
||
|
|
||
|
def setUp(self):
|
||
|
super(PersonCase, self).setUp()
|
||
|
self.values = {"firstname": "Núñez", "lastname": "Fernán"}
|
||
|
self.values["name"] = "{} {}".format(
|
||
|
self.values["firstname"], self.values["lastname"]
|
||
|
)
|
||
|
if "default_is_company" in self.context:
|
||
|
self.values["is_company"] = self.context["default_is_company"]
|
||
|
|
||
|
def tearDown(self):
|
||
|
for key, value in self.values.items():
|
||
|
self.assertEqual(self.defaults.get(key), value, "Checking key %s" % key)
|
||
|
|
||
|
return super(PersonCase, self).tearDown()
|
||
|
|
||
|
def test_default_get(self):
|
||
|
"""Getting default values for fields includes new fields."""
|
||
|
self.defaults = (
|
||
|
self.env[self.model]
|
||
|
.with_context(self.context, default_name=self.values["name"])
|
||
|
.default_get(list(self.values.keys()))
|
||
|
)
|
||
|
|
||
|
|
||
|
class CompanyCase(PersonCase):
|
||
|
"""Test ``res.partner`` when it is a company."""
|
||
|
|
||
|
context = {"default_is_company": True}
|
||
|
|
||
|
def tearDown(self):
|
||
|
self.values.update(lastname=self.values["name"], firstname=False)
|
||
|
return super(CompanyCase, self).tearDown()
|
||
|
|
||
|
|
||
|
class UserCase(PersonCase, MailInstalled):
|
||
|
"""Test ``res.users``."""
|
||
|
|
||
|
model = "res.users"
|
||
|
context = {"default_login": "user@example.com"}
|
||
|
|
||
|
def tearDown(self):
|
||
|
# Cannot create users if ``mail`` is installed
|
||
|
if self.mail_installed():
|
||
|
# Skip tests
|
||
|
super(PersonCase, self).tearDown()
|
||
|
else:
|
||
|
# Run tests
|
||
|
super(UserCase, self).tearDown()
|