partner-contact/partner_firstname/tests/test_copy.py

# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
"""Test copy function for models."""
from flectra.tests.common import TransactionCase

from .base import MailInstalled


class UserCase(TransactionCase, MailInstalled):
    """Test ``res.users``."""

    def setUp(self):
        super(UserCase, self).setUp()
        self.create_original()

    def create_original(self):
        self.original = self.env["res.users"].create(
            {
                "firstname": u"Firstname",
                "lastname": u"Lastname",
                "name": u"Firstname Lastname",
                "login": u"firstname.lastname",
            }
        )

    def tearDown(self):
        super(UserCase, self).tearDown()

    def compare(self, copy):
        self.assertEqual(copy.lastname, u"Lastname2")
        self.assertEqual(copy.firstname, u"Firstname2")
        self.assertEqual(copy.name, u"Firstname2 Lastname2")

    def test_copy_name(self):
        """Copy original with default name set - firstname lastname not set."""
        copy = self.original.copy({"name": u"Firstname2 Lastname2"})
        self.compare(copy)

    def test_copy_firstname_lastname(self):
        """Copy original with default firstname and lastname set"""
        copy = self.original.copy(
            {"firstname": u"Firstname2", "lastname": u"Lastname2"}
        )
        self.compare(copy)

    def test_copy_firstname_lastname_name(self):
        """Copy original with default firstname, lastname and name set"""
        copy = self.original.copy(
            {
                "firstname": u"Firstname2",
                "lastname": u"Lastname2",
                "name": u"Firstname2 Lastname2",
            }
        )
        self.compare(copy)