bt2: check for _graph_is_configured method in user sink classes

[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..d5274ae49d3cef72591ea88526fedf1b2d2db83c 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,19 +1,37 @@
+#
+# Copyright (C) 2019 EfficiOS Inc.
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of the GNU General Public License
+# as published by the Free Software Foundation; only version 2
+# of the License.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
+#
+
 import bt2
 
 
-class MyIter(bt2._UserNotificationIterator):
+class MyIter(bt2._UserMessageIterator):
     pass
 
 
 @bt2.plugin_component_class
 class MySource(bt2._UserSourceComponent,
-               notification_iterator_class=MyIter):
+               message_iterator_class=MyIter):
     pass
 
 
 @bt2.plugin_component_class
 class MyFilter(bt2._UserFilterComponent,
-               notification_iterator_class=MyIter):
+               message_iterator_class=MyIter):
     pass
 
 
@@ -22,6 +40,9 @@ class MySink(bt2._UserSinkComponent):
     def _consume(self):
         pass
 
+    def _graph_is_configured(self):
+        pass
+
 
 bt2.register_plugin(__name__, 'sparkling', author='Philippe Proulx',
                     description='A delicious plugin.',