X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=tests%2Fbindings%2Fpython%2Fbt2%2Ftest_connection.py;fp=tests%2Fbindings%2Fpython%2Fbt2%2Ftest_connection.py;h=77d12410eba45add697d563daf22fa4993d38907;hb=5813b3a3ffb8ff44d54b134e26d604af0b5e36db;hp=edeefd08df4635289c838ec524f52b9b7476e0c4;hpb=c7e5224bd55c9b9864f7e3a4e8d1a3df276ebc2b;p=babeltrace.git

diff --git a/tests/bindings/python/bt2/test_connection.py b/tests/bindings/python/bt2/test_connection.py
index edeefd08..77d12410 100644
--- a/tests/bindings/python/bt2/test_connection.py
+++ b/tests/bindings/python/bt2/test_connection.py
@@ -19,6 +19,7 @@
 import unittest
 import bt2
 from bt2 import connection as bt2_connection
+from bt2 import port as bt2_port
 
 
 class ConnectionTestCase(unittest.TestCase):
@@ -66,6 +67,7 @@ class ConnectionTestCase(unittest.TestCase):
         conn = graph.connect_ports(src.output_ports['out'], sink.input_ports['in'])
         self.assertEqual(conn.downstream_port.addr, sink.input_ports['in'].addr)
         self.assertIs(type(conn), bt2_connection._ConnectionConst)
+        self.assertIs(type(conn.downstream_port), bt2_port._InputPortConst)
 
     def test_upstream_port(self):
         class MyIter(bt2._UserMessageIterator):
@@ -88,3 +90,4 @@ class ConnectionTestCase(unittest.TestCase):
         sink = graph.add_component(MySink, 'sink')
         conn = graph.connect_ports(src.output_ports['out'], sink.input_ports['in'])
         self.assertEqual(conn.upstream_port.addr, src.output_ports['out'].addr)
+        self.assertIs(type(conn.upstream_port), bt2_port._OutputPortConst)