bt2: Adapt test_message_iterator.py and make it pass

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

diff --git a/tests/bindings/python/bt2/test_graph.py b/tests/bindings/python/bt2/test_graph.py
index 8ed585a626aa0ff4d0753949d6ca99c03a78cc61..48e02873bb812b76d8c2bd414c4f1fb84b44bd8d 100644 (file)
--- a/tests/bindings/python/bt2/test_graph.py
+++ b/tests/bindings/python/bt2/test_graph.py
@@ -6,7 +6,7 @@ import bt2
 
 
 class _MyIter(bt2._UserMessageIterator):
-    def __init__(self):
+    def __init__(self, self_output_port):
         self._build_meta()
         self._at = 0
 
@@ -496,7 +496,8 @@ class GraphTestCase(unittest.TestCase):
             def _consume(self):
                 raise bt2.Stop
 
-        def ports_connected_listener(upstream_port, downstream_port):
+        def ports_connected_listener(upstream_component, upstream_port,
+                                     downstream_component, downstream_port):
             raise ValueError('oh noes!')
 
         graph = bt2.Graph()