bt2: Adapt test_port.py and make it pass

[babeltrace.git] / tests / bindings / python / bt2 / test_component.py

diff --git a/tests/bindings/python/bt2/test_component.py b/tests/bindings/python/bt2/test_component.py
index a0e05b918b92502e3a9d077635d7b764f800e5c4..30523ecd3bd71d17178ecc93734a40ef2987e587 100644 (file)
--- a/tests/bindings/python/bt2/test_component.py
+++ b/tests/bindings/python/bt2/test_component.py
@@ -12,7 +12,7 @@ class UserComponentTestCase(unittest.TestCase):
         if name is None:
             name = 'comp'
 
-        return graph.add_sink_component(comp_cls, name)
+        return graph.add_component(comp_cls, name)
 
     def test_name(self):
         class MySink(bt2._UserSinkComponent):
@@ -57,7 +57,7 @@ class UserComponentTestCase(unittest.TestCase):
                 finalized = True
 
         graph = bt2.Graph()
-        comp = graph.add_sink_component(MySink, 'lel')
+        comp = graph.add_component(MySink, 'lel')
 
         del graph
         del comp
@@ -72,7 +72,7 @@ class GenericComponentTestCase(unittest.TestCase):
         if name is None:
             name = 'comp'
 
-        return graph.add_sink_component(comp_cls, name)
+        return graph.add_component(comp_cls, name)
 
     def test_name(self):
         class MySink(bt2._UserSinkComponent):