+# SPDX-License-Identifier: GPL-2.0-only
+#
+# Copyright (C) 2019 EfficiOS Inc.
+#
+
import bt2
import os
class TestIter(bt2._UserMessageIterator):
- def __init__(self, output_port):
+ def __init__(self, config, output_port):
params = output_port.user_data['params']
obj = output_port.user_data['obj']
@bt2.plugin_component_class
class TestSourceA(Base, bt2._UserSourceComponent, message_iterator_class=TestIter):
- def __init__(self, params, obj):
+ def __init__(self, config, params, obj):
super().__init__(params, obj)
@staticmethod
@bt2.plugin_component_class
class TestSourceB(Base, bt2._UserSourceComponent, message_iterator_class=TestIter):
- def __init__(self, params, obj):
+ def __init__(self, config, params, obj):
super().__init__(params, obj)
@staticmethod