server-ux/base_optional_quick_create/tests/test_quick_create.py

# Copyright 2018 Simone Rubino - Agile Business Group
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
from flectra.tests import common
from flectra.exceptions import UserError
from flectra.tests.common import TransactionCase


@common.at_install(False)
@common.post_install(True)
class TestQuickCreate(TransactionCase):

    def setUp(self, *args, **kwargs):
        super(TestQuickCreate, self).setUp()
        model_model = self.env['ir.model']
        self.partner_model = model_model.search([
            ('model', '=', 'res.partner')])

    def test_quick_create(self):
        partner_id = self.env['res.partner'].name_create('TEST partner')
        self.assertEqual(bool(partner_id), True)

        # Setting the flag, patches the method
        self.partner_model.avoid_quick_create = True
        with self.assertRaises(UserError):
            self.env['res.partner'].name_create('TEST partner')

        # Unsetting the flag, unpatches the method
        self.partner_model.avoid_quick_create = False
        partner_id = self.env['res.partner'].name_create('TEST partner')
        self.assertEqual(bool(partner_id), True)

    def not_for_flectra_test_create_model(self):
        model_id = self.env['ir.model'].create({'name': 'Test',
                                                'model': 'x_test_model'})
        self.assertEqual(bool(model_id), True)