server-ux/base_optional_quick_create/tests/test_quick_create.py
2022-06-29 21:02:59 +02:00

36 lines
1.4 KiB
Python

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