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=edeefd08df4635289c838ec524f52b9b7476e0c4;hb=c7e5224bd55c9b9864f7e3a4e8d1a3df276ebc2b;hp=127fd2e8fd36c7e08239e88f107ea006ce9f9081;hpb=f0a42b33ac3951cd5cb2ee0f66ac04437a681621;p=babeltrace.git

diff --git a/tests/bindings/python/bt2/test_connection.py b/tests/bindings/python/bt2/test_connection.py
index 127fd2e8..edeefd08 100644
--- a/tests/bindings/python/bt2/test_connection.py
+++ b/tests/bindings/python/bt2/test_connection.py
@@ -18,6 +18,7 @@
 
 import unittest
 import bt2
+from bt2 import connection as bt2_connection
 
 
 class ConnectionTestCase(unittest.TestCase):
@@ -40,7 +41,8 @@ class ConnectionTestCase(unittest.TestCase):
         graph = bt2.Graph()
         src = graph.add_component(MySource, 'src')
         sink = graph.add_component(MySink, 'sink')
-        graph.connect_ports(src.output_ports['out'], sink.input_ports['in'])
+        conn = graph.connect_ports(src.output_ports['out'], sink.input_ports['in'])
+        self.assertIs(type(conn), bt2_connection._ConnectionConst)
 
     def test_downstream_port(self):
         class MyIter(bt2._UserMessageIterator):
@@ -63,6 +65,7 @@ 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.downstream_port.addr, sink.input_ports['in'].addr)
+        self.assertIs(type(conn), bt2_connection._ConnectionConst)
 
     def test_upstream_port(self):
         class MyIter(bt2._UserMessageIterator):