partner-contact/partner_firstname/tests/test_create.py

# -*- coding: utf-8 -*-
# © 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 openerp.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.good_values = {
            "firstname": u"Núñez",
            "lastname": u"Fernán",
        }
        self.good_values["name"] = "%s %s" % (self.good_values["lastname"],
                                              self.good_values["firstname"])
        if "default_is_company" in self.context:
            self.good_values["is_company"] = self.context["default_is_company"]
        self.values = self.good_values.copy()

    def tearDown(self):
        self.record = (self.env[self.model]
                       .with_context(self.context)
                       .create(self.values))
        for key, value in self.good_values.iteritems():
            self.assertEqual(
                self.record[key],
                value,
                "Checking key %s" % key)

        super(PersonCase, self).tearDown()

    def test_no_name(self):
        """Name is calculated."""
        del self.values["name"]

    def test_wrong_name_value(self):
        """Wrong name value is ignored, name is calculated."""
        self.values["name"] = u"BÄD"

    def test_wrong_name_context(self):
        """Wrong name context is ignored, name is calculated."""
        del self.values["name"]
        self.context["default_name"] = u"BÄD"

    def test_wrong_name_value_and_context(self):
        """Wrong name value and context is ignored, name is calculated."""
        self.values["name"] = u"BÄD1"
        self.context["default_name"] = u"BÄD2"


class CompanyCase(PersonCase):
    """Test ``res.partner`` when it is a company."""
    context = {"default_is_company": True}

    def setUp(self):
        super(CompanyCase, self).setUp()
        self.good_values.update(lastname=self.values["name"], firstname=False)
        self.values = self.good_values.copy()


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()