partner-contact/partner_ref_unique/tests/test_res_partner_ref.py

# Copyright 2017 Tecnativa - Vicent Cubells
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).

from odoo.exceptions import ValidationError
from odoo.tests import common


class TestResPartnerRefUnique(common.SavepointCase):
    @classmethod
    def setUpClass(cls):
        super(TestResPartnerRefUnique, cls).setUpClass()
        cls.company = cls.env.ref("base.main_company")
        cls.partner_obj = cls.env["res.partner"]
        cls.partner1 = cls.partner_obj.create({"name": "Partner1",})
        cls.partner2 = cls.partner_obj.create({"name": "Partner2",})

    def test_check_ref_company(self):
        # Test can create/modify partners with same ref
        self.company.partner_ref_unique = "none"
        self.partner1.ref = "same_ref"
        self.partner2.ref = "same_ref"
        self.assertEqual(self.partner1.ref, self.partner2.ref)
        # Here there shouldn't be any problem
        self.partner_obj.create(
            {"name": "other", "ref": "same_ref",}
        )
        self.partner2.ref = False
        with self.assertRaises(ValidationError):
            self.company.partner_ref_unique = "all"

    def test_check_ref(self):
        self.partner1.ref = "same_ref"
        # Test can't create/modify partner with same ref
        self.company.partner_ref_unique = "all"
        with self.assertRaises(ValidationError):
            self.partner2.ref = "same_ref"
        with self.assertRaises(ValidationError):
            self.partner_obj.create(
                {"name": "other", "ref": "same_ref",}
            )
        # Test can't create/modify companies with same ref
        self.company.partner_ref_unique = "companies"
        self.partner2.ref = "same_ref"
        self.assertEqual(self.partner1.ref, self.partner2.ref)
        self.partner2.ref = False
        self.partner1.is_company = True
        self.partner2.is_company = True
        with self.assertRaises(ValidationError):
            self.partner2.ref = "same_ref"
        with self.assertRaises(ValidationError):
            self.partner_obj.create(
                {"is_company": True, "name": "other", "ref": "same_ref",}
            )
        # Here there shouldn't be any problem
        self.partner_obj.create(
            {"is_company": False, "name": "other", "ref": "same_ref",}
        )