bt2: make `bt2._OverflowError` inherit `bt2._Error`
[babeltrace.git] / tests / bindings / python / bt2 / test_port.py
index b346cbb6c272a7c7bbaa796edb3ddf6856c4999f..35d8797e8973036f66479b6e9653487ff3dce023 100644 (file)
@@ -1,10 +1,25 @@
-from bt2 import value
+#
+# Copyright (C) 2019 EfficiOS Inc.
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of the GNU General Public License
+# as published by the Free Software Foundation; only version 2
+# of the License.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
+#
+
 import unittest
-import copy
 import bt2
 
 
-@unittest.skip("this is broken")
 class PortTestCase(unittest.TestCase):
     @staticmethod
     def _create_comp(comp_cls, name=None):
@@ -20,8 +35,7 @@ class PortTestCase(unittest.TestCase):
             def __next__(self):
                 raise bt2.Stop
 
-        class MySource(bt2._UserSourceComponent,
-                       message_iterator_class=MyIter):
+        class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 port = comp_self._add_output_port('out')
                 self.assertEqual(port.name, 'out')
@@ -29,14 +43,12 @@ class PortTestCase(unittest.TestCase):
         comp = self._create_comp(MySource)
         self.assertEqual(len(comp.output_ports), 1)
 
-
     def test_flt_add_output_port(self):
         class MyIter(bt2._UserMessageIterator):
             def __next__(self):
                 raise bt2.Stop
 
-        class MyFilter(bt2._UserFilterComponent,
-                       message_iterator_class=MyIter):
+        class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 port = comp_self._add_output_port('out')
                 self.assertEqual(port.name, 'out')
@@ -49,8 +61,7 @@ class PortTestCase(unittest.TestCase):
             def __next__(self):
                 raise bt2.Stop
 
-        class MyFilter(bt2._UserFilterComponent,
-                       message_iterator_class=MyIter):
+        class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 port = comp_self._add_input_port('in')
                 self.assertEqual(port.name, 'in')
@@ -67,6 +78,9 @@ class PortTestCase(unittest.TestCase):
             def _consume(self):
                 pass
 
+            def _graph_is_configured(self):
+                pass
+
         comp = self._create_comp(MySink)
         self.assertEqual(len(comp.input_ports), 1)
 
@@ -75,8 +89,7 @@ class PortTestCase(unittest.TestCase):
             def __next__(self):
                 raise bt2.Stop
 
-        class MySource(bt2._UserSourceComponent,
-                       message_iterator_class=MyIter):
+        class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 port1 = comp_self._add_output_port('clear')
                 port2 = comp_self._add_output_port('print')
@@ -85,15 +98,14 @@ class PortTestCase(unittest.TestCase):
                 self.assertEqual(port2.addr, comp_self._output_ports['print'].addr)
                 self.assertEqual(port1.addr, comp_self._output_ports['clear'].addr)
 
-        comp = self._create_comp(MySource)
+        self._create_comp(MySource)
 
     def test_user_flt_output_ports_getitem(self):
         class MyIter(bt2._UserMessageIterator):
             def __next__(self):
                 raise bt2.Stop
 
-        class MyFilter(bt2._UserFilterComponent,
-                       message_iterator_class=MyIter):
+        class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 port1 = comp_self._add_output_port('clear')
                 port2 = comp_self._add_output_port('print')
@@ -102,15 +114,14 @@ class PortTestCase(unittest.TestCase):
                 self.assertEqual(port2.addr, comp_self._output_ports['print'].addr)
                 self.assertEqual(port1.addr, comp_self._output_ports['clear'].addr)
 
-        comp = self._create_comp(MyFilter)
+        self._create_comp(MyFilter)
 
     def test_user_flt_input_ports_getitem(self):
         class MyIter(bt2._UserMessageIterator):
             def __next__(self):
                 raise bt2.Stop
 
-        class MyFilter(bt2._UserFilterComponent,
-                       message_iterator_class=MyIter):
+        class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 port1 = comp_self._add_input_port('clear')
                 port2 = comp_self._add_input_port('print')
@@ -119,7 +130,7 @@ class PortTestCase(unittest.TestCase):
                 self.assertEqual(port2.addr, comp_self._input_ports['print'].addr)
                 self.assertEqual(port1.addr, comp_self._input_ports['clear'].addr)
 
-        comp = self._create_comp(MyFilter)
+        self._create_comp(MyFilter)
 
     def test_user_sink_input_ports_getitem(self):
         class MySink(bt2._UserSinkComponent):
@@ -134,15 +145,17 @@ class PortTestCase(unittest.TestCase):
             def _consume(self):
                 pass
 
-        comp = self._create_comp(MySink)
+            def _graph_is_configured(self):
+                pass
+
+        self._create_comp(MySink)
 
     def test_user_src_output_ports_getitem_invalid_key(self):
         class MyIter(bt2._UserMessageIterator):
             def __next__(self):
                 raise bt2.Stop
 
-        class MySource(bt2._UserSourceComponent,
-                       message_iterator_class=MyIter):
+        class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 comp_self._add_output_port('clear')
                 comp_self._add_output_port('print')
@@ -151,15 +164,14 @@ class PortTestCase(unittest.TestCase):
                 with self.assertRaises(KeyError):
                     comp_self._output_ports['hello']
 
-        comp = self._create_comp(MySource)
+        self._create_comp(MySource)
 
     def test_user_flt_output_ports_getitem_invalid_key(self):
         class MyIter(bt2._UserMessageIterator):
             def __next__(self):
                 raise bt2.Stop
 
-        class MyFilter(bt2._UserFilterComponent,
-                       message_iterator_class=MyIter):
+        class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 comp_self._add_output_port('clear')
                 comp_self._add_output_port('print')
@@ -168,15 +180,14 @@ class PortTestCase(unittest.TestCase):
                 with self.assertRaises(KeyError):
                     comp_self._output_ports['hello']
 
-        comp = self._create_comp(MyFilter)
+        self._create_comp(MyFilter)
 
     def test_user_flt_input_ports_getitem_invalid_key(self):
         class MyIter(bt2._UserMessageIterator):
             def __next__(self):
                 raise bt2.Stop
 
-        class MyFilter(bt2._UserFilterComponent,
-                       message_iterator_class=MyIter):
+        class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 comp_self._add_input_port('clear')
                 comp_self._add_input_port('print')
@@ -185,7 +196,7 @@ class PortTestCase(unittest.TestCase):
                 with self.assertRaises(KeyError):
                     comp_self._input_ports['hello']
 
-        comp = self._create_comp(MyFilter)
+        self._create_comp(MyFilter)
 
     def test_user_sink_input_ports_getitem_invalid_key(self):
         class MySink(bt2._UserSinkComponent):
@@ -200,52 +211,52 @@ class PortTestCase(unittest.TestCase):
             def _consume(self):
                 pass
 
-        comp = self._create_comp(MySink)
+            def _graph_is_configured(self):
+                pass
+
+        self._create_comp(MySink)
 
     def test_user_src_output_ports_len(self):
         class MyIter(bt2._UserMessageIterator):
             def __next__(self):
                 raise bt2.Stop
 
-        class MySource(bt2._UserSourceComponent,
-                       message_iterator_class=MyIter):
+        class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 comp_self._add_output_port('clear')
                 comp_self._add_output_port('print')
                 comp_self._add_output_port('insert')
                 self.assertEqual(len(comp_self._output_ports), 3)
 
-        comp = self._create_comp(MySource)
+        self._create_comp(MySource)
 
     def test_user_flt_output_ports_len(self):
         class MyIter(bt2._UserMessageIterator):
             def __next__(self):
                 raise bt2.Stop
 
-        class MyFilter(bt2._UserFilterComponent,
-                       message_iterator_class=MyIter):
+        class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 comp_self._add_output_port('clear')
                 comp_self._add_output_port('print')
                 comp_self._add_output_port('insert')
                 self.assertEqual(len(comp_self._output_ports), 3)
 
-        comp = self._create_comp(MyFilter)
+        self._create_comp(MyFilter)
 
     def test_user_flt_input_ports_len(self):
         class MyIter(bt2._UserMessageIterator):
             def __next__(self):
                 raise bt2.Stop
 
-        class MyFilter(bt2._UserFilterComponent,
-                       message_iterator_class=MyIter):
+        class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 comp_self._add_input_port('clear')
                 comp_self._add_input_port('print')
                 comp_self._add_input_port('insert')
                 self.assertEqual(len(comp_self._input_ports), 3)
 
-        comp = self._create_comp(MyFilter)
+        self._create_comp(MyFilter)
 
     def test_user_sink_input_ports_len(self):
         class MySink(bt2._UserSinkComponent):
@@ -258,15 +269,17 @@ class PortTestCase(unittest.TestCase):
             def _consume(self):
                 pass
 
-        comp = self._create_comp(MySink)
+            def _graph_is_configured(self):
+                pass
+
+        self._create_comp(MySink)
 
     def test_user_src_output_ports_iter(self):
         class MyIter(bt2._UserMessageIterator):
             def __next__(self):
                 raise bt2.Stop
 
-        class MySource(bt2._UserSourceComponent,
-                       message_iterator_class=MyIter):
+        class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 port1 = comp_self._add_output_port('clear')
                 port2 = comp_self._add_output_port('print')
@@ -283,15 +296,14 @@ class PortTestCase(unittest.TestCase):
                 self.assertEqual(ports[2][0], 'insert')
                 self.assertEqual(ports[2][1].addr, port3.addr)
 
-        comp = self._create_comp(MySource)
+        self._create_comp(MySource)
 
     def test_user_flt_output_ports_iter(self):
         class MyIter(bt2._UserMessageIterator):
             def __next__(self):
                 raise bt2.Stop
 
-        class MyFilter(bt2._UserFilterComponent,
-                       message_iterator_class=MyIter):
+        class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 port1 = comp_self._add_output_port('clear')
                 port2 = comp_self._add_output_port('print')
@@ -308,15 +320,14 @@ class PortTestCase(unittest.TestCase):
                 self.assertEqual(ports[2][0], 'insert')
                 self.assertEqual(ports[2][1].addr, port3.addr)
 
-        comp = self._create_comp(MyFilter)
+        self._create_comp(MyFilter)
 
     def test_user_flt_input_ports_iter(self):
         class MyIter(bt2._UserMessageIterator):
             def __next__(self):
                 raise bt2.Stop
 
-        class MyFilter(bt2._UserFilterComponent,
-                       message_iterator_class=MyIter):
+        class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 port1 = comp_self._add_input_port('clear')
                 port2 = comp_self._add_input_port('print')
@@ -333,7 +344,7 @@ class PortTestCase(unittest.TestCase):
                 self.assertEqual(ports[2][0], 'insert')
                 self.assertEqual(ports[2][1].addr, port3.addr)
 
-        comp = self._create_comp(MyFilter)
+        self._create_comp(MyFilter)
 
     def test_user_sink_input_ports_iter(self):
         class MySink(bt2._UserSinkComponent):
@@ -356,7 +367,10 @@ class PortTestCase(unittest.TestCase):
             def _consume(self):
                 pass
 
-        comp = self._create_comp(MySink)
+            def _graph_is_configured(self):
+                pass
+
+        self._create_comp(MySink)
 
     def test_gen_src_output_ports_getitem(self):
         class MyIter(bt2._UserMessageIterator):
@@ -367,8 +381,7 @@ class PortTestCase(unittest.TestCase):
         port2 = None
         port3 = None
 
-        class MySource(bt2._UserSourceComponent,
-                       message_iterator_class=MyIter):
+        class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 nonlocal port1, port2, port3
                 port1 = comp_self._add_output_port('clear')
@@ -392,8 +405,7 @@ class PortTestCase(unittest.TestCase):
         port2 = None
         port3 = None
 
-        class MyFilter(bt2._UserFilterComponent,
-                       message_iterator_class=MyIter):
+        class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 nonlocal port1, port2, port3
                 port1 = comp_self._add_output_port('clear')
@@ -417,8 +429,7 @@ class PortTestCase(unittest.TestCase):
         port2 = None
         port3 = None
 
-        class MyFilter(bt2._UserFilterComponent,
-                       message_iterator_class=MyIter):
+        class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 nonlocal port1, port2, port3
                 port1 = comp_self._add_input_port('clear')
@@ -448,6 +459,9 @@ class PortTestCase(unittest.TestCase):
             def _consume(self):
                 pass
 
+            def _graph_is_configured(self):
+                pass
+
         comp = self._create_comp(MySink)
         self.assertEqual(port3.addr, comp.input_ports['insert'].addr)
         self.assertEqual(port2.addr, comp.input_ports['print'].addr)
@@ -461,8 +475,7 @@ class PortTestCase(unittest.TestCase):
             def __next__(self):
                 raise bt2.Stop
 
-        class MySource(bt2._UserSourceComponent,
-                       message_iterator_class=MyIter):
+        class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 comp_self._add_output_port('clear')
                 comp_self._add_output_port('print')
@@ -478,8 +491,7 @@ class PortTestCase(unittest.TestCase):
             def __next__(self):
                 raise bt2.Stop
 
-        class MyFilter(bt2._UserFilterComponent,
-                       message_iterator_class=MyIter):
+        class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 comp_self._add_output_port('clear')
                 comp_self._add_output_port('print')
@@ -495,8 +507,7 @@ class PortTestCase(unittest.TestCase):
             def __next__(self):
                 raise bt2.Stop
 
-        class MyFilter(bt2._UserFilterComponent,
-                       message_iterator_class=MyIter):
+        class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 comp_self._add_input_port('clear')
                 comp_self._add_input_port('print')
@@ -520,6 +531,9 @@ class PortTestCase(unittest.TestCase):
             def _consume(self):
                 pass
 
+            def _graph_is_configured(self):
+                pass
+
         comp = self._create_comp(MySink)
 
         with self.assertRaises(KeyError):
@@ -530,8 +544,7 @@ class PortTestCase(unittest.TestCase):
             def __next__(self):
                 raise bt2.Stop
 
-        class MySource(bt2._UserSourceComponent,
-                       message_iterator_class=MyIter):
+        class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 comp_self._add_output_port('clear')
                 comp_self._add_output_port('print')
@@ -545,8 +558,7 @@ class PortTestCase(unittest.TestCase):
             def __next__(self):
                 raise bt2.Stop
 
-        class MyFilter(bt2._UserFilterComponent,
-                       message_iterator_class=MyIter):
+        class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 comp_self._add_output_port('clear')
                 comp_self._add_output_port('print')
@@ -560,8 +572,7 @@ class PortTestCase(unittest.TestCase):
             def __next__(self):
                 raise bt2.Stop
 
-        class MyFilter(bt2._UserFilterComponent,
-                       message_iterator_class=MyIter):
+        class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 comp_self._add_input_port('clear')
                 comp_self._add_input_port('print')
@@ -580,6 +591,9 @@ class PortTestCase(unittest.TestCase):
             def _consume(self):
                 pass
 
+            def _graph_is_configured(self):
+                pass
+
         comp = self._create_comp(MySink)
         self.assertEqual(len(comp.input_ports), 3)
 
@@ -592,8 +606,7 @@ class PortTestCase(unittest.TestCase):
         port2 = None
         port3 = None
 
-        class MySource(bt2._UserSourceComponent,
-                       message_iterator_class=MyIter):
+        class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 nonlocal port1, port2, port3
                 port1 = comp_self._add_output_port('clear')
@@ -625,8 +638,7 @@ class PortTestCase(unittest.TestCase):
         port2 = None
         port3 = None
 
-        class MyFilter(bt2._UserFilterComponent,
-                       message_iterator_class=MyIter):
+        class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 nonlocal port1, port2, port3
                 port1 = comp_self._add_output_port('clear')
@@ -658,8 +670,7 @@ class PortTestCase(unittest.TestCase):
         port2 = None
         port3 = None
 
-        class MyFilter(bt2._UserFilterComponent,
-                       message_iterator_class=MyIter):
+        class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
                 nonlocal port1, port2, port3
                 port1 = comp_self._add_input_port('clear')
@@ -697,6 +708,9 @@ class PortTestCase(unittest.TestCase):
             def _consume(self):
                 pass
 
+            def _graph_is_configured(self):
+                pass
+
         comp = self._create_comp(MySink)
         ports = []
 
@@ -721,19 +735,11 @@ class PortTestCase(unittest.TestCase):
             def _consume(self):
                 pass
 
-        comp = self._create_comp(MySink)
-        self.assertEqual(comp.input_ports['clear'].name, 'clear')
-
-    def test_component(self):
-        class MySink(bt2._UserSinkComponent):
-            def __init__(comp_self, params):
-                comp_self._add_input_port('clear')
-
-            def _consume(self):
+            def _graph_is_configured(self):
                 pass
 
         comp = self._create_comp(MySink)
-        self.assertEqual(comp.input_ports['clear'].component.addr, comp.addr)
+        self.assertEqual(comp.input_ports['clear'].name, 'clear')
 
     def test_connection_none(self):
         class MySink(bt2._UserSinkComponent):
@@ -743,6 +749,9 @@ class PortTestCase(unittest.TestCase):
             def _consume(self):
                 pass
 
+            def _graph_is_configured(self):
+                pass
+
         comp = self._create_comp(MySink)
         self.assertIsNone(comp.input_ports['clear'].connection)
 
@@ -754,136 +763,114 @@ class PortTestCase(unittest.TestCase):
             def _consume(self):
                 pass
 
+            def _graph_is_configured(self):
+                pass
+
         comp = self._create_comp(MySink)
         self.assertFalse(comp.input_ports['clear'].is_connected)
 
-    def test_eq(self):
+    def test_self_name(self):
         class MySink(bt2._UserSinkComponent):
             def __init__(comp_self, params):
-                comp_self._add_input_port('clear')
+                port = comp_self._add_input_port('clear')
+                self.assertEqual(port.name, 'clear')
 
             def _consume(self):
                 pass
 
-        comp = self._create_comp(MySink)
-        self.assertEqual(comp.input_ports['clear'],
-                         comp.input_ports['clear'])
-
-    def test_eq_invalid(self):
-        class MySink(bt2._UserSinkComponent):
-            def __init__(comp_self, params):
-                comp_self._add_input_port('clear')
-
-            def _consume(self):
+            def _graph_is_configured(self):
                 pass
 
-        comp = self._create_comp(MySink)
-        self.assertNotEqual(comp.input_ports['clear'], 23)
+        self._create_comp(MySink)
 
-    def test_disconnect_no_connection(self):
+    def test_self_connection_none(self):
         class MySink(bt2._UserSinkComponent):
             def __init__(comp_self, params):
                 port = comp_self._add_input_port('clear')
+                self.assertIsNone(port.connection)
 
             def _consume(self):
                 pass
 
-        comp = self._create_comp(MySink)
-        comp.input_ports['clear'].disconnect()
-
-    def test_priv_name(self):
-        class MySink(bt2._UserSinkComponent):
-            def __init__(comp_self, params):
-                port = comp_self._add_input_port('clear')
-                self.assertEqual(port.name, 'clear')
-
-            def _consume(self):
+            def _graph_is_configured(self):
                 pass
 
-        comp = self._create_comp(MySink)
+        self._create_comp(MySink)
 
-    def test_priv_component(self):
+    def test_self_is_connected_false(self):
         class MySink(bt2._UserSinkComponent):
             def __init__(comp_self, params):
                 port = comp_self._add_input_port('clear')
-                self.assertEqual(port.component, comp_self)
+                self.assertFalse(port.is_connected)
 
             def _consume(self):
                 pass
 
-        comp = self._create_comp(MySink)
-
-    def test_priv_connection_none(self):
-        class MySink(bt2._UserSinkComponent):
-            def __init__(comp_self, params):
-                port = comp_self._add_input_port('clear')
-                self.assertIsNone(port.connection)
-
-            def _consume(self):
+            def _graph_is_configured(self):
                 pass
 
-        comp = self._create_comp(MySink)
+        self._create_comp(MySink)
 
-    def test_priv_is_connected_false(self):
-        class MySink(bt2._UserSinkComponent):
-            def __init__(comp_self, params):
-                port = comp_self._add_input_port('clear')
-                self.assertFalse(port.is_connected)
+    def test_source_self_port_user_data(self):
+        class MyIter(bt2._UserMessageIterator):
+            def __next__(self):
+                raise bt2.Stop
 
-            def _consume(self):
-                pass
+        class MySource(bt2._UserFilterComponent, message_iterator_class=MyIter):
+            def __init__(comp_self, params):
+                nonlocal user_datas
 
-        comp = self._create_comp(MySink)
+                p = comp_self._add_output_port('port1')
+                user_datas.append(p.user_data)
+                p = comp_self._add_output_port('port2', 2)
+                user_datas.append(p.user_data)
 
-    def test_priv_eq(self):
-        class MySink(bt2._UserSinkComponent):
-            def __init__(comp_self, params):
-                port = comp_self._add_input_port('clear')
-                self.assertEqual(port, port)
+        user_datas = []
 
-            def _consume(self):
-                pass
+        comp = self._create_comp(MySource)
+        self.assertEqual(user_datas, [None, 2])
 
-        comp = self._create_comp(MySink)
+    def test_filter_self_port_user_data(self):
+        class MyIter(bt2._UserMessageIterator):
+            def __next__(self):
+                raise bt2.Stop
 
-    def test_priv_eq_invalid(self):
-        class MySink(bt2._UserSinkComponent):
+        class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
-                port = comp_self._add_input_port('clear')
-                self.assertNotEqual(port, 23)
+                nonlocal user_datas
 
-            def _consume(self):
-                pass
+                p = comp_self._add_output_port('port1')
+                user_datas.append(p.user_data)
+                p = comp_self._add_output_port('port2', 'user data string')
+                user_datas.append(p.user_data)
 
-        comp = self._create_comp(MySink)
+                p = comp_self._add_input_port('port3')
+                user_datas.append(p.user_data)
+                p = comp_self._add_input_port('port4', user_data={'user data': 'dict'})
+                user_datas.append(p.user_data)
 
-    def test_priv_disconnect_no_connection(self):
-        class MySink(bt2._UserSinkComponent):
-            def __init__(comp_self, params):
-                port = comp_self._add_input_port('clear')
-                port.disconnect()
+        user_datas = []
 
-            def _consume(self):
-                pass
+        comp = self._create_comp(MyFilter)
+        self.assertEqual(
+            user_datas, [None, 'user data string', None, {'user data': 'dict'}]
+        )
 
-        comp = self._create_comp(MySink)
+    def test_sink_self_port_user_data(self):
+        class MyIter(bt2._UserMessageIterator):
+            def __next__(self):
+                raise bt2.Stop
 
-    def test_priv_remove_from_component(self):
-        class MySink(bt2._UserSinkComponent):
+        class MySink(bt2._UserFilterComponent, message_iterator_class=MyIter):
             def __init__(comp_self, params):
-                port = comp_self._add_input_port('clear')
-                self.assertEqual(len(comp_self._input_ports), 1)
+                nonlocal user_datas
 
-                try:
-                    port.remove_from_component()
-                except:
-                    import traceback
-                    traceback.print_exc()
+                p = comp_self._add_input_port('port1')
+                user_datas.append(p.user_data)
+                p = comp_self._add_input_port('port2', set())
+                user_datas.append(p.user_data)
 
-                self.assertEqual(len(comp_self._input_ports), 0)
-                self.assertIsNone(port.component)
-
-            def _consume(self):
-                pass
+        user_datas = []
 
         comp = self._create_comp(MySink)
+        self.assertEqual(user_datas, [None, set()])
This page took 0.031464 seconds and 4 git commands to generate.