mirror of
https://gitlab.com/flectra-community/partner-contact.git
synced 2024-11-15 02:32:04 +00:00
118 lines
4.2 KiB
Python
118 lines
4.2 KiB
Python
# Copyright 2016 ACSONE SA/NV (<http://acsone.eu>)
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
|
|
from psycopg2._psycopg import IntegrityError
|
|
from flectra.tests import common
|
|
from flectra.exceptions import ValidationError
|
|
from flectra.tools import mute_logger
|
|
|
|
|
|
class TestPartnerIdentificationBase(common.TransactionCase):
|
|
|
|
def test_create_id_category(self):
|
|
partner_id_category = self.env['res.partner.id_category'].create({
|
|
'code': 'id_code',
|
|
'name': 'id_name',
|
|
})
|
|
self.assertEqual(partner_id_category.name, 'id_name')
|
|
self.assertEqual(partner_id_category.code, 'id_code')
|
|
|
|
@mute_logger('flectra.sql_db')
|
|
def test_update_partner_with_no_category(self):
|
|
partner_1 = self.env.ref('base.res_partner_1')
|
|
self.assertEqual(len(partner_1.id_numbers), 0)
|
|
# create without required category
|
|
with self.assertRaises(IntegrityError):
|
|
partner_1.write({'id_numbers': [(0, 0, {
|
|
'name': '1234',
|
|
})]})
|
|
|
|
def test_update_partner_with_category(self):
|
|
partner_1 = self.env.ref('base.res_partner_1')
|
|
partner_id_category = self.env['res.partner.id_category'].create({
|
|
'code': 'new_code',
|
|
'name': 'new_name',
|
|
})
|
|
# successful creation
|
|
partner_1.write({'id_numbers': [(0, 0, {
|
|
'name': '1234',
|
|
'category_id': partner_id_category.id
|
|
})]})
|
|
self.assertEqual(len(partner_1.id_numbers), 1)
|
|
self.assertEqual(partner_1.id_numbers.name, '1234')
|
|
# delete
|
|
partner_1.write({'id_numbers': [(5, 0, 0)]})
|
|
self.assertEqual(len(partner_1.id_numbers), 0)
|
|
|
|
|
|
class TestPartnerCategoryValidation(common.TransactionCase):
|
|
|
|
def test_partner_id_number_validation(self):
|
|
partner_id_category = self.env['res.partner.id_category'].create({
|
|
'code': 'id_code',
|
|
'name': 'id_name',
|
|
'validation_code': """
|
|
if id_number.name != '1234':
|
|
failed = True
|
|
"""
|
|
})
|
|
partner_1 = self.env.ref('base.res_partner_1')
|
|
with self.assertRaises(ValidationError), self.cr.savepoint():
|
|
partner_1.write({'id_numbers': [(0, 0, {
|
|
'name': '01234',
|
|
'category_id': partner_id_category.id
|
|
})]})
|
|
partner_1.write({'id_numbers': [(0, 0, {
|
|
'name': '1234',
|
|
'category_id': partner_id_category.id
|
|
})]})
|
|
self.assertEqual(len(partner_1.id_numbers), 1)
|
|
self.assertEqual(partner_1.id_numbers.name, '1234')
|
|
|
|
partner_id_category2 = self.env['res.partner.id_category'].create({
|
|
'code': 'id_code2',
|
|
'name': 'id_name2',
|
|
'validation_code': """
|
|
if id_number.name != '1235':
|
|
failed = True
|
|
"""})
|
|
# check that the constrains is also checked when we change the
|
|
# associated category
|
|
with self.assertRaises(ValidationError), self.cr.savepoint():
|
|
partner_1.id_numbers.write({
|
|
'category_id': partner_id_category2.id
|
|
})
|
|
|
|
def test_bad_validation_code(self):
|
|
partner_id_category = self.env['res.partner.id_category'].create({
|
|
'code': 'id_code',
|
|
'name': 'id_name',
|
|
'validation_code': """
|
|
if id_number.name != '1234' # missing :
|
|
failed = True
|
|
"""
|
|
})
|
|
partner_1 = self.env.ref('base.res_partner_1')
|
|
with self.assertRaises(ValidationError):
|
|
partner_1.write({'id_numbers': [(0, 0, {
|
|
'name': '1234',
|
|
'category_id': partner_id_category.id
|
|
})]})
|
|
|
|
def test_bad_validation_code_override(self):
|
|
""" It should allow a bad validation code if context overrides. """
|
|
partner_id_category = self.env['res.partner.id_category'].create({
|
|
'code': 'id_code',
|
|
'name': 'id_name',
|
|
'validation_code': """
|
|
if id_number.name != '1234' # missing :
|
|
failed = True
|
|
"""
|
|
})
|
|
partner_1 = self.env.ref('base.res_partner_1').with_context(
|
|
id_no_validate=True,
|
|
)
|
|
partner_1.write({'id_numbers': [(0, 0, {
|
|
'name': '1234',
|
|
'category_id': partner_id_category.id
|
|
})]})
|