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 7e489d32c9ca350a754360f6446ac66b3a27a1f6..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):
@@ -782,7 +785,7 @@ class PortTestCase(unittest.TestCase):
 
         user_datas = []
 
-        comp = self._create_comp(MySource)
+        self._create_comp(MySource)
         self.assertEqual(user_datas, [None, 2])
 
     def test_filter_self_port_user_data(self):
@@ -806,7 +809,7 @@ class PortTestCase(unittest.TestCase):
 
         user_datas = []
 
-        comp = self._create_comp(MyFilter)
+        self._create_comp(MyFilter)
         self.assertEqual(
             user_datas, [None, 'user data string', None, {'user data': 'dict'}]
         )
@@ -827,5 +830,5 @@ class PortTestCase(unittest.TestCase):
 
         user_datas = []
 
-        comp = self._create_comp(MySink)
+        self._create_comp(MySink)
         self.assertEqual(user_datas, [None, set()])