Initial commit: OCA Technical packages (595 packages)

odoo-bringout-oca-connector-component/component/tests/test_work_on.py

@@ -0,0 +1,73 @@
+# 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 TransactionComponentRegistryCase
+
+
+class TestWorkOn(TransactionComponentRegistryCase):
+    """Test on WorkContext
+
+    This model is mostly a container, so we check the access
+    to the attributes and properties.
+
+    """
+
+    def setUp(self):
+        super().setUp()
+        self._setup_registry(self)
+
+    def tearDown(self):
+        self._teardown_registry(self)
+        super().tearDown()
+
+    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)