X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=tests%2Fbindings%2Fpython%2Fbt2%2Ftest_port.py;fp=tests%2Fbindings%2Fpython%2Fbt2%2Ftest_port.py;h=c404d8fac8cd1cf9286383f7c717edce3fb1d648;hb=1606ab4cf99d9a9063a1c38bbddb3f5f49db8969;hp=53fdf9c79168a81d99cdcd4c65f602c9e62a7c01;hpb=e7279cae2af449410ef6ead35b0eac8d2528e0d6;p=babeltrace.git

diff --git a/tests/bindings/python/bt2/test_port.py b/tests/bindings/python/bt2/test_port.py
index 53fdf9c7..c404d8fa 100644
--- 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)