tests: make test_sink_self_port_user_data actually test a sink

[babeltrace.git] / tests / bindings / python / bt2 / test_port.py

diff --git a/tests/bindings/python/bt2/test_port.py b/tests/bindings/python/bt2/test_port.py
index 53fdf9c79168a81d99cdcd4c65f602c9e62a7c01..c404d8fac8cd1cf9286383f7c717edce3fb1d648 100644 (file)
--- a/tests/bindings/python/bt2/test_port.py
+++ b/tests/bindings/python/bt2/test_port.py
@@ -815,11 +815,7 @@ class PortTestCase(unittest.TestCase):
         )
 
     def test_sink_self_port_user_data(self):
-        class MyIter(bt2._UserMessageIterator):
-            def __next__(self):
-                raise bt2.Stop
-
-        class MySink(bt2._UserFilterComponent, message_iterator_class=MyIter):
+        class MySink(bt2._UserSinkComponent):
             def __init__(comp_self, config, params, obj):
                 nonlocal user_datas
 
@@ -828,6 +824,9 @@ class PortTestCase(unittest.TestCase):
                 p = comp_self._add_input_port('port2', set())
                 user_datas.append(p.user_data)
 
+            def _user_consume(self):
+                pass
+
         user_datas = []
 
         self._create_comp(MySink)