# 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 })