lib: create input port msg iterator from self {msg iterator, sink comp.}

[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 1c183c9bf1861be1103c1e56b41fabd3840d7b0b..4f6aade1febc787f8544e736beac3edd5cb55647 100644 (file)
--- a/tests/bindings/python/bt2/test_graph.py
+++ b/tests/bindings/python/bt2/test_graph.py
@@ -224,7 +224,9 @@ class GraphTestCase(unittest.TestCase):
                 return next(self._msg_iter)
 
             def _graph_is_configured(self):
-                self._msg_iter = self._input_ports['in'].create_message_iterator()
+                self._msg_iter = self._create_input_port_message_iterator(
+                    self._input_ports['in']
+                )
 
         graph = bt2.Graph()
         up = graph.add_component(MySource, 'down')
@@ -280,7 +282,9 @@ class GraphTestCase(unittest.TestCase):
                 comp_self._at += 1
 
             def _graph_is_configured(self):
-                self._msg_iter = self._input_port.create_message_iterator()
+                self._msg_iter = self._create_input_port_message_iterator(
+                    self._input_port
+                )
 
         src = self._graph.add_component(MySource, 'src')
         sink = self._graph.add_component(MySink, 'sink')
@@ -329,7 +333,9 @@ class GraphTestCase(unittest.TestCase):
                 comp_self._at += 1
 
             def _graph_is_configured(self):
-                self._msg_iter = self._input_port.create_message_iterator()
+                self._msg_iter = self._create_input_port_message_iterator(
+                    self._input_port
+                )
 
         src = self._graph.add_component(MySource, 'src')
         sink = self._graph.add_component(MySink, 'sink')
@@ -386,7 +392,9 @@ class GraphTestCase(unittest.TestCase):
                 comp_self._at += 1
 
             def _graph_is_configured(self):
-                self._msg_iter = self._input_port.create_message_iterator()
+                self._msg_iter = self._create_input_port_message_iterator(
+                    self._input_port
+                )
 
         src = self._graph.add_component(MySource, 'src')
         sink = self._graph.add_component(MySink, 'sink')
@@ -394,7 +402,7 @@ class GraphTestCase(unittest.TestCase):
             src.output_ports['out'], sink.input_ports['in']
         )
 
-        with self.assertRaises(bt2.Error):
+        with self.assertRaises(bt2._Error):
             self._graph.run()
 
     def test_listeners(self):
@@ -510,7 +518,7 @@ class GraphTestCase(unittest.TestCase):
 
         graph = bt2.Graph()
 
-        with self.assertRaises(bt2.Error):
+        with self.assertRaises(bt2._Error):
             graph.add_component(MySink, 'comp')
 
     def test_raise_in_port_added_listener(self):
@@ -530,7 +538,7 @@ class GraphTestCase(unittest.TestCase):
         graph = bt2.Graph()
         graph.add_port_added_listener(port_added_listener)
 
-        with self.assertRaises(bt2.Error):
+        with self.assertRaises(bt2._Error):
             graph.add_component(MySink, 'comp')
 
     def test_raise_in_ports_connected_listener(self):
@@ -562,5 +570,5 @@ class GraphTestCase(unittest.TestCase):
         up = graph.add_component(MySource, 'down')
         down = graph.add_component(MySink, 'up')
 
-        with self.assertRaises(bt2.Error):
+        with self.assertRaises(bt2._Error):
             graph.connect_ports(up.output_ports['out'], down.input_ports['in'])