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

[babeltrace.git] / tests / data / plugins / flt.utils.trimmer / bt_plugin_trimmer_test.py

diff --git a/tests/data/plugins/flt.utils.trimmer/bt_plugin_trimmer_test.py b/tests/data/plugins/flt.utils.trimmer/bt_plugin_trimmer_test.py
index 0fdb78f9f56ff4f94b5cdfff10c69ec67200b83f..c7d5985dffd001143305621121f53ed73a91d2c9 100644 (file)
--- a/tests/data/plugins/flt.utils.trimmer/bt_plugin_trimmer_test.py
+++ b/tests/data/plugins/flt.utils.trimmer/bt_plugin_trimmer_test.py
@@ -33,7 +33,7 @@ class TheIteratorOfAllEvil(bt2._UserMessageIterator):
         ]
         self._at = 0
 
-    def _seek_beginning(self):
+    def _user_seek_beginning(self):
         self._at = 0
 
     def __next__(self):
@@ -49,7 +49,7 @@ class TheIteratorOfAllEvil(bt2._UserMessageIterator):
 class TheSourceOfAllEvil(
     bt2._UserSourceComponent, message_iterator_class=TheIteratorOfAllEvil
 ):
-    def __init__(self, params):
+    def __init__(self, params, obj):
         tc = self._create_trace_class()
 
         # Use a clock class with an offset, so we can test with --begin or --end