component: allow propagation of custom registry

component/models/collection.py

@@ -93,4 +93,9 @@ def work_on(self, model_name, **kwargs):
 
         """
         self.ensure_one()
+        # Allow propagation of custom component registry via context
+        # TODO: maybe to be moved to `WorkContext.__init__`
+        components_registry = self.env.context.get("components_registry")
+        if components_registry:
+            kwargs["components_registry"] = components_registry
         yield WorkContext(model_name=model_name, collection=self, **kwargs)

component/tests/common.py

@@ -41,6 +41,10 @@ def setUpComponent(cls):
             # build the components of the current tested addon
             current_addon = _get_addon_name(cls.__module__)
             env["component.builder"].load_components(current_addon)
+            # share component registry everywhere
+            cls.env.context = dict(
+                cls.env.context, components_registry=cls._components_registry
+            )
 
     # pylint: disable=W8106
     def setUp(self):
@@ -201,6 +205,9 @@ def setUp(self):
         # of the components. Here, we'll add components later in
         # the components registry, but we don't mind for the tests.
         self.comp_registry.ready = True
+        self.env.context = dict(
+            self.env.context, components_registry=self.comp_registry
+        )
 
     def tearDown(self):
         super().tearDown()