bt2: Add `Const` suffix to `_*Port` classes and adapt tests

[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 7db49fb27011d36391a3a4ac24237cd5996f07f0..5c1a8271fef73a9dc1cc826bc6fc853d21c11e92 100644 (file)
--- a/tests/bindings/python/bt2/test_port.py
+++ b/tests/bindings/python/bt2/test_port.py
@@ -18,6 +18,7 @@
 
 import unittest
 import bt2
+from bt2 import port as bt2_port
 
 
 class PortTestCase(unittest.TestCase):
@@ -42,6 +43,7 @@ class PortTestCase(unittest.TestCase):
 
         comp = self._create_comp(MySource)
         self.assertEqual(len(comp.output_ports), 1)
+        self.assertIs(type(comp.output_ports['out']), bt2_port._OutputPortConst)
 
     def test_flt_add_output_port(self):
         class MyIter(bt2._UserMessageIterator):
@@ -68,6 +70,7 @@ class PortTestCase(unittest.TestCase):
 
         comp = self._create_comp(MyFilter)
         self.assertEqual(len(comp.input_ports), 1)
+        self.assertIs(type(comp.input_ports['in']), bt2_port._InputPortConst)
 
     def test_sink_add_input_port(self):
         class MySink(bt2._UserSinkComponent):