partner-contact/partner_coc/tests/test_partner_coc.py

65 lines
2.2 KiB
Python
Raw Normal View History

2021-03-23 19:36:29 +00:00
# Copyright 2018 Onestein (<http://www.onestein.eu>)
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
from flectra.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')