X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=tests%2Fdata%2Fplugins%2Fflt.utils.muxer%2Fbt_plugin_muxer_test.py;h=c263d20396060017deb108d68e85d0bbee8bb406;hb=8d8b141db4c46135a35be19e4a1c192f6a36d67b;hp=f8135fc6fdaf6c32fa320d8c59e91cfda9da8c2e;hpb=2b950395ddd2aaee36d0be59ed303826bceb0193;p=babeltrace.git

diff --git a/tests/data/plugins/flt.utils.muxer/bt_plugin_muxer_test.py b/tests/data/plugins/flt.utils.muxer/bt_plugin_muxer_test.py
index f8135fc6..c263d203 100644
--- a/tests/data/plugins/flt.utils.muxer/bt_plugin_muxer_test.py
+++ b/tests/data/plugins/flt.utils.muxer/bt_plugin_muxer_test.py
@@ -2,7 +2,7 @@ import bt2
 
 
 class TheIteratorOfConfusion(bt2._UserMessageIterator):
-    def __init__(self, port):
+    def __init__(self, config, port):
         self._at = 0
         test_name = port.user_data[0]
         TEST_CASES[test_name].create_msgs(self, port.user_data[1:])
@@ -19,9 +19,7 @@ class TheIteratorOfConfusion(bt2._UserMessageIterator):
 class TheSourceOfConfusion(
     bt2._UserSourceComponent, message_iterator_class=TheIteratorOfConfusion
 ):
-    def __init__(self, params, obj):
-        tc = self._create_trace_class()
-
+    def __init__(self, config, params, obj):
         test_name = str(params['test-name'])
 
         TEST_CASES[test_name].source_setup(self, test_name)