Fix: flt-utils.muxer: reference leak in muxer_msg_iter_add_upstream_msg_iter error...

[babeltrace.git] / tests / python-plugin-provider / bt_plugin_test_python_plugin_provider.py

diff --git a/tests/python-plugin-provider/bt_plugin_test_python_plugin_provider.py b/tests/python-plugin-provider/bt_plugin_test_python_plugin_provider.py
index 83f5e3dc76d4b3980d8ebd161d5e15ee4c9767aa..ed018c29ca813ece1937ef002da258c428c60faf 100644 (file)
--- a/tests/python-plugin-provider/bt_plugin_test_python_plugin_provider.py
+++ b/tests/python-plugin-provider/bt_plugin_test_python_plugin_provider.py
@@ -1,29 +1,36 @@
+# SPDX-License-Identifier: GPL-2.0-only
+#
+# Copyright (C) 2019 EfficiOS Inc.
+#
+
 import bt2
 
 
-class MyIter(bt2._UserNotificationIterator):
+class MyIter(bt2._UserMessageIterator):
     pass
 
 
 @bt2.plugin_component_class
-class MySource(bt2._UserSourceComponent,
-               notification_iterator_class=MyIter):
+class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter):
     pass
 
 
 @bt2.plugin_component_class
-class MyFilter(bt2._UserFilterComponent,
-               notification_iterator_class=MyIter):
+class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
     pass
 
 
 @bt2.plugin_component_class
 class MySink(bt2._UserSinkComponent):
-    def _consume(self):
+    def _user_consume(self):
         pass
 
 
-bt2.register_plugin(__name__, 'sparkling', author='Philippe Proulx',
-                    description='A delicious plugin.',
-                    version=(1, 2, 3, 'EXTRA'),
-                    license='MIT')
+bt2.register_plugin(
+    __name__,
+    'sparkling',
+    author='Philippe Proulx',
+    description='A delicious plugin.',
+    version=(1, 2, 3, 'EXTRA'),
+    license='MIT',
+)