server-ux/base_optional_quick_create/tests/test_quick_create.py

47 lines
1.7 KiB
Python
Raw Normal View History

2021-04-28 15:44:37 +00:00
# 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)