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.tests import common
from odoo.exceptions import ValidationError


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(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
        # 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',
        })