partner-contact/partner_coc/tests/test_partner_coc.py

# Copyright 2018 Onestein (<http://www.onestein.eu>)
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).

from openerp.tests.common import TransactionCase


class TestPartnerCoC(TransactionCase):
    def setUp(self):
        super(TestPartnerCoC, self).setUp()
        self.main_partner = self.env.ref('base.main_partner')
        self.partner_id_category = self.env.ref('partner_coc.id_category_coc')

    def test_01_id_number_new(self):
        # Good CoC
        vals = {
            'name': '56048785',
            'category_id': self.partner_id_category.id,
        }
        self.main_partner.write({'id_numbers': [(0, 0, vals)]})
        id_number = self.main_partner.id_numbers[0]

        self.assertEqual(id_number.name, '56048785')

    def test_02_id_number_duplicate(self):
        # Duplicate CoC
        vals = {
            'name': '56048785',
            'category_id': self.partner_id_category.id,
        }

        self.main_partner.write({'id_numbers': [(0, 0, vals)]})
        id_number_name = self.main_partner.id_numbers[0].name
        self.assertEqual(id_number_name, '56048785')

        new_partner = self.env['res.partner'].create({'name': 'Test Partner'})

        new_partner.write({'id_numbers': [(0, 0, vals)]})
        id_number_name = new_partner.id_numbers[0].name
        self.assertEqual(id_number_name, '56048785')

    def test_03_coc_registration_number_create(self):
        new_partner = self.env['res.partner'].create({
            'name': 'Test Partner',
            'coc_registration_number': '56048785',
        })

        self.assertEqual(new_partner.coc_registration_number, '56048785')

        id_numbers = new_partner.id_numbers
        self.assertTrue(id_numbers)
        self.assertEqual(len(id_numbers), 1)
        self.assertEqual(id_numbers.name, '56048785')

    def test_04_coc_registration_number_write(self):
        self.main_partner.write({
            'coc_registration_number': '56048785'
        })
        coc = self.main_partner.coc_registration_number
        self.assertEqual(coc, '56048785')

        id_numbers = self.main_partner.id_numbers
        self.assertTrue(id_numbers)
        self.assertEqual(len(id_numbers), 1)
        self.assertEqual(id_numbers.name, '56048785')