server-ux/base_optional_quick_create/tests/test_quick_create.py

36 lines
1.3 KiB
Python
Raw Normal View History

2021-03-23 19:38:33 +00:00
# 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 test_create_model(self):
model_id = self.env['ir.model'].create({'name': 'Test',
'model': 'x_test_model'})
self.assertEqual(bool(model_id), True)