bt2: pass custom Python object to Python component's __init__()

[babeltrace.git] / tests / data / cli / convert / auto-source-discovery-grouping / bt_plugin_test.py

diff --git a/tests/data/cli/convert/auto-source-discovery-grouping/bt_plugin_test.py b/tests/data/cli/convert/auto-source-discovery-grouping/bt_plugin_test.py
index 4d6e86e2da8e20c97c2fbab845324ad4b6d54b54..3719ceb9fcfc3ad8b4eaea34a9bd268bf990e709 100644 (file)
--- a/tests/data/cli/convert/auto-source-discovery-grouping/bt_plugin_test.py
+++ b/tests/data/cli/convert/auto-source-discovery-grouping/bt_plugin_test.py
@@ -22,7 +22,7 @@ class TestSourceExt(Base, bt2._UserSourceComponent, message_iterator_class=TestI
     files are not grouped.
     """
 
-    def __init__(self, params):
+    def __init__(self, params, obj):
         self._print_params(params)
 
     @staticmethod
@@ -60,7 +60,7 @@ class TestSourceSomeDir(
     directory "some-dir" won't be found by TestSourceExt, because we won't
     recurse in "some-dir"."""
 
-    def __init__(self, params):
+    def __init__(self, params, obj):
         self._print_params(params)
 
     @staticmethod
@@ -79,7 +79,7 @@ class TestSourceSomeDir(
 class TestSourceABCDE(Base, bt2._UserSourceComponent, message_iterator_class=TestIter):
     """A source that recognizes the arbitrary string input "ABCDE"."""
 
-    def __init__(self, params):
+    def __init__(self, params, obj):
         self._print_params(params)
 
     @staticmethod