from bt2 import object as bt2_object
-class _Connection(bt2_object._SharedObject):
+class _ConnectionConst(bt2_object._SharedObject):
_get_ref = staticmethod(native_bt.connection_get_ref)
_put_ref = staticmethod(native_bt.connection_put_ref)
)
utils._handle_func_status(status, 'cannot connect component ports within graph')
assert conn_ptr
- return bt2_connection._Connection._create_from_ptr(conn_ptr)
+ return bt2_connection._ConnectionConst._create_from_ptr(conn_ptr)
def add_port_added_listener(self, listener):
if not callable(listener):
if conn_ptr is None:
return
- return bt2_connection._Connection._create_from_ptr_and_get_ref(conn_ptr)
+ return bt2_connection._ConnectionConst._create_from_ptr_and_get_ref(conn_ptr)
@property
def is_connected(self):
if conn_ptr is None:
return
- return bt2_connection._Connection._create_from_ptr_and_get_ref(conn_ptr)
+ return bt2_connection._ConnectionConst._create_from_ptr_and_get_ref(conn_ptr)
@property
def user_data(self):
import unittest
import bt2
+from bt2 import connection as bt2_connection
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):
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):