66 lines
2.8 KiB
Python
66 lines
2.8 KiB
Python
|
# Copyright 2017 Camptocamp SA
|
||
|
|
# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl.html)
|
||
|
|
|
||
|
|
from odoo.addons.component.core import ComponentRegistry, WorkContext
|
||
|
|
|
||
|
|
from .common import SavepointComponentRegistryCase
|
||
|
|
|
||
|
|
|
||
|
|
class TestWorkOn(SavepointComponentRegistryCase):
|
||
|
|
"""Test on WorkContext
|
||
|
|
|
||
|
|
This model is mostly a container, so we check the access
|
||
|
|
to the attributes and properties.
|
||
|
|
|
||
|
|
"""
|
||
|
|
|
||
|
|
def test_collection_work_on(self):
|
||
|
|
"""Create a new instance and test attributes access"""
|
||
|
|
collection_record = self.collection.new()
|
||
|
|
with collection_record.work_on("res.partner") as work:
|
||
|
|
self.assertEqual(collection_record, work.collection)
|
||
|
|
self.assertEqual("collection.base", work.collection._name)
|
||
|
|
self.assertEqual("res.partner", work.model_name)
|
||
|
|
self.assertEqual(self.env["res.partner"], work.model)
|
||
|
|
self.assertEqual(self.env, work.env)
|
||
|
|
|
||
|
|
def test_collection_work_on_registry_via_context(self):
|
||
|
|
"""Test propagation of registry via context"""
|
||
|
|
registry = ComponentRegistry()
|
||
|
|
collection_record = self.collection.with_context(
|
||
|
|
components_registry=registry
|
||
|
|
).new()
|
||
|
|
with collection_record.work_on("res.partner") as work:
|
||
|
|
self.assertEqual(collection_record, work.collection)
|
||
|
|
self.assertEqual("collection.base", work.collection._name)
|
||
|
|
self.assertEqual("res.partner", work.model_name)
|
||
|
|
self.assertEqual(self.env["res.partner"], work.model)
|
||
|
|
self.assertEqual(work.env, collection_record.env)
|
||
|
|
self.assertEqual(work.components_registry, registry)
|
||
|
|
|
||
|
|
def test_propagate_work_on(self):
|
||
|
|
"""Check custom attributes and their propagation"""
|
||
|
|
registry = ComponentRegistry()
|
||
|
|
work = WorkContext(
|
||
|
|
model_name="res.partner",
|
||
|
|
collection=self.collection,
|
||
|
|
# we can customize the lookup registry, but used mostly for tests
|
||
|
|
components_registry=registry,
|
||
|
|
# we can pass our own keyword args that will set as attributes
|
||
|
|
test_keyword="value",
|
||
|
|
)
|
||
|
|
self.assertIs(registry, work.components_registry)
|
||
|
|
# check that our custom keyword is set as attribute
|
||
|
|
self.assertEqual("value", work.test_keyword)
|
||
|
|
|
||
|
|
# when we want to work on another model, work_on() create
|
||
|
|
# another instance and propagate the attributes to it
|
||
|
|
work2 = work.work_on("res.users")
|
||
|
|
self.assertNotEqual(work, work2)
|
||
|
|
self.assertEqual(self.env, work2.env)
|
||
|
|
self.assertEqual(self.collection, work2.collection)
|
||
|
|
self.assertEqual("res.users", work2.model_name)
|
||
|
|
self.assertIs(registry, work2.components_registry)
|
||
|
|
# test_keyword has been propagated to the new WorkContext instance
|
||
|
|
self.assertEqual("value", work2.test_keyword)
|