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

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)