mirror of
https://gitlab.com/flectra-community/server-ux.git
synced 2024-12-25 05:41:48 +00:00
47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
|
# 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)
|