Files
enterprise-addons/data_cleaning/tests/test_common.py
T

89 lines
3.4 KiB
Python

# Part of Odoo. See LICENSE file for full copyright and licensing details.
import unittest
from odoo.tests.common import TransactionCase
from odoo.tools.func import reset_cached_properties
from odoo.tools.sql import convert_column
class TestCommon(TransactionCase):
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.addClassCleanup(lambda: cls.registry._setup_models__(cls.cr))
def setUp(self):
super(TestCommon, self).setUp()
self.registry._setup_models__(self.cr)
self.DMModel = self.env['data_merge.model']
self.DMRule = self.env['data_merge.rule']
self.DMGroup = self.env['data_merge.group']
self.DMRecord = self.env['data_merge.record']
self.DMTestModel = self.env['ir.model'].create({
'name': 'Test Model',
'model': 'x_dm_test_model',
'field_id': [
(0, 0, {'name': 'x_name', 'ttype': 'char', 'field_description': 'Name'}),
(0, 0, {'name': 'x_email', 'ttype': 'char', 'field_description': 'Email'}),
]
})
self.DMTestModelRef = self.env['ir.model'].create({
'name': 'Test Model Ref',
'model': 'x_dm_test_model_ref',
'field_id': [
(0, 0, {'name': 'x_name', 'ttype': 'char', 'field_description': 'Name'}),
(0, 0, {'name': 'x_test_id', 'ttype': 'many2one', 'field_description': 'Test Ref', 'relation': 'x_dm_test_model', 'index': True}),
]
})
self.DMTestModel2 = self.env['ir.model'].create({
'name': 'Test Model',
'model': 'x_dm_test_model2',
'field_id': [
(0, 0, {'name': 'x_name', 'ttype': 'char', 'field_description': 'Name'}),
(0, 0, {'name': 'x_email', 'ttype': 'char', 'field_description': 'Email'}),
]
})
self.DMTestModelCompanyDependent = self.env['ir.model'].create({
'name': 'Test Model Company Dependent',
'model': 'x_dm_test_model_cd',
'field_id': [
(0, 0, {'name': 'x_name', 'ttype': 'char', 'field_description': 'Name'}),
(0, 0, {'name': 'x_cd', 'ttype': 'char', 'field_description': 'CD'}),
]
})
self.MyModel = self.DMModel.create({
'name': 'test of test model',
'res_model_id': self.DMTestModel.id,
})
self.MyModel2 = self.DMModel.create({
'name': 'test of test model 2',
'res_model_id': self.DMTestModel2.id,
})
field = self.registry['x_dm_test_model_cd']._fields['x_cd']
field.company_dependent = True
self.registry.field_depends_context[field] = ('company',)
reset_cached_properties(field)
convert_column(self.env.cr, 'x_dm_test_model_cd', 'x_cd', 'jsonb')
def _create_record(self, model, **kwargs):
return self.env[model].create(kwargs)
def _create_rule(self, field_name, mode, model_name='x_dm_test_model'):
model = self.MyModel if model_name == 'x_dm_test_model' else self.MyModel2
if mode == 'accent' and not self.registry.has_unaccent:
raise unittest.SkipTest("Unaccent rules require unaccent to be enabled")
self.DMRule.create({
'model_id': model.id,
'field_id': self.env['ir.model.fields']._get(model_name, field_name).id,
'match_mode': mode
})