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.exceptions import UserError
from flectra.tests.common import TransactionCase


class TestQuickCreate(TransactionCase):
    def setUp(self, *args, **kwargs):
        super().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)

        # New Model

        # Setting the flag, patches the method
        self.env["ir.model"].create(
            {"name": "Test Model", "model": "x_.test.model", "avoid_quick_create": True}
        )
        with self.assertRaises(UserError):
            self.env["x_.test.model"].name_create("TEST Model")

        # Unsetting the flag, unpatches the method
        self.env["ir.model"].create(
            {
                "name": "Test Model",
                "model": "x_.test.model.quick",
                "avoid_quick_create": False,
            }
        )
        test_id = self.env["x_.test.model.quick"].name_create("TEST Model")
        self.assertEqual(bool(test_id), True)