X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=tests%2Fbindings%2Fpython%2Fbt2%2Ftest_port.py;h=7e489d32c9ca350a754360f6446ac66b3a27a1f6;hb=53487b4e9e5e77b3117be384f2ba3344b0b545ed;hp=4566c3a60d0a52e4dbd1b2110f5c25a21c00480c;hpb=2e00bc76cf37af167c45253cfc8f8d197222d6b8;p=babeltrace.git diff --git a/tests/bindings/python/bt2/test_port.py b/tests/bindings/python/bt2/test_port.py index 4566c3a6..7e489d32 100644 --- a/tests/bindings/python/bt2/test_port.py +++ b/tests/bindings/python/bt2/test_port.py @@ -1,3 +1,21 @@ +# +# 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 bt2 @@ -17,9 +35,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MySource(bt2._UserSourceComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): port = comp_self._add_output_port('out') self.assertEqual(port.name, 'out') @@ -31,9 +48,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MyFilter(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): port = comp_self._add_output_port('out') self.assertEqual(port.name, 'out') @@ -45,9 +61,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MyFilter(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): port = comp_self._add_input_port('in') self.assertEqual(port.name, 'in') @@ -56,11 +71,11 @@ class PortTestCase(unittest.TestCase): def test_sink_add_input_port(self): class MySink(bt2._UserSinkComponent): - def __init__(comp_self, params): + def __init__(comp_self, params, obj): port = comp_self._add_input_port('in') self.assertEqual(port.name, 'in') - def _consume(self): + def _user_consume(self): pass comp = self._create_comp(MySink) @@ -71,9 +86,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MySource(bt2._UserSourceComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): port1 = comp_self._add_output_port('clear') port2 = comp_self._add_output_port('print') port3 = comp_self._add_output_port('insert') @@ -88,9 +102,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MyFilter(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): port1 = comp_self._add_output_port('clear') port2 = comp_self._add_output_port('print') port3 = comp_self._add_output_port('insert') @@ -105,9 +118,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MyFilter(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): port1 = comp_self._add_input_port('clear') port2 = comp_self._add_input_port('print') port3 = comp_self._add_input_port('insert') @@ -119,7 +131,7 @@ class PortTestCase(unittest.TestCase): def test_user_sink_input_ports_getitem(self): class MySink(bt2._UserSinkComponent): - def __init__(comp_self, params): + def __init__(comp_self, params, obj): port1 = comp_self._add_input_port('clear') port2 = comp_self._add_input_port('print') port3 = comp_self._add_input_port('insert') @@ -127,7 +139,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) - def _consume(self): + def _user_consume(self): pass self._create_comp(MySink) @@ -137,9 +149,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MySource(bt2._UserSourceComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): comp_self._add_output_port('clear') comp_self._add_output_port('print') comp_self._add_output_port('insert') @@ -154,9 +165,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MyFilter(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): comp_self._add_output_port('clear') comp_self._add_output_port('print') comp_self._add_output_port('insert') @@ -171,9 +181,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MyFilter(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): comp_self._add_input_port('clear') comp_self._add_input_port('print') comp_self._add_input_port('insert') @@ -185,7 +194,7 @@ class PortTestCase(unittest.TestCase): def test_user_sink_input_ports_getitem_invalid_key(self): class MySink(bt2._UserSinkComponent): - def __init__(comp_self, params): + def __init__(comp_self, params, obj): comp_self._add_input_port('clear') comp_self._add_input_port('print') comp_self._add_input_port('insert') @@ -193,7 +202,7 @@ class PortTestCase(unittest.TestCase): with self.assertRaises(KeyError): comp_self._input_ports['hello'] - def _consume(self): + def _user_consume(self): pass self._create_comp(MySink) @@ -203,9 +212,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MySource(bt2._UserSourceComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): comp_self._add_output_port('clear') comp_self._add_output_port('print') comp_self._add_output_port('insert') @@ -218,9 +226,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MyFilter(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): comp_self._add_output_port('clear') comp_self._add_output_port('print') comp_self._add_output_port('insert') @@ -233,9 +240,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MyFilter(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): comp_self._add_input_port('clear') comp_self._add_input_port('print') comp_self._add_input_port('insert') @@ -245,13 +251,13 @@ class PortTestCase(unittest.TestCase): def test_user_sink_input_ports_len(self): class MySink(bt2._UserSinkComponent): - def __init__(comp_self, params): + def __init__(comp_self, params, obj): 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) - def _consume(self): + def _user_consume(self): pass self._create_comp(MySink) @@ -261,9 +267,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MySource(bt2._UserSourceComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): port1 = comp_self._add_output_port('clear') port2 = comp_self._add_output_port('print') port3 = comp_self._add_output_port('insert') @@ -286,9 +291,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MyFilter(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): port1 = comp_self._add_output_port('clear') port2 = comp_self._add_output_port('print') port3 = comp_self._add_output_port('insert') @@ -311,9 +315,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MyFilter(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): port1 = comp_self._add_input_port('clear') port2 = comp_self._add_input_port('print') port3 = comp_self._add_input_port('insert') @@ -333,7 +336,7 @@ class PortTestCase(unittest.TestCase): def test_user_sink_input_ports_iter(self): class MySink(bt2._UserSinkComponent): - def __init__(comp_self, params): + def __init__(comp_self, params, obj): port1 = comp_self._add_input_port('clear') port2 = comp_self._add_input_port('print') port3 = comp_self._add_input_port('insert') @@ -349,7 +352,7 @@ class PortTestCase(unittest.TestCase): self.assertEqual(ports[2][0], 'insert') self.assertEqual(ports[2][1].addr, port3.addr) - def _consume(self): + def _user_consume(self): pass self._create_comp(MySink) @@ -363,9 +366,8 @@ class PortTestCase(unittest.TestCase): port2 = None port3 = None - class MySource(bt2._UserSourceComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): nonlocal port1, port2, port3 port1 = comp_self._add_output_port('clear') port2 = comp_self._add_output_port('print') @@ -388,9 +390,8 @@ class PortTestCase(unittest.TestCase): port2 = None port3 = None - class MyFilter(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): nonlocal port1, port2, port3 port1 = comp_self._add_output_port('clear') port2 = comp_self._add_output_port('print') @@ -413,9 +414,8 @@ class PortTestCase(unittest.TestCase): port2 = None port3 = None - class MyFilter(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): nonlocal port1, port2, port3 port1 = comp_self._add_input_port('clear') port2 = comp_self._add_input_port('print') @@ -435,13 +435,13 @@ class PortTestCase(unittest.TestCase): port3 = None class MySink(bt2._UserSinkComponent): - def __init__(comp_self, params): + def __init__(comp_self, params, obj): nonlocal port1, port2, port3 port1 = comp_self._add_input_port('clear') port2 = comp_self._add_input_port('print') port3 = comp_self._add_input_port('insert') - def _consume(self): + def _user_consume(self): pass comp = self._create_comp(MySink) @@ -457,9 +457,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MySource(bt2._UserSourceComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): comp_self._add_output_port('clear') comp_self._add_output_port('print') comp_self._add_output_port('insert') @@ -474,9 +473,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MyFilter(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): comp_self._add_output_port('clear') comp_self._add_output_port('print') comp_self._add_output_port('insert') @@ -491,9 +489,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MyFilter(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): comp_self._add_input_port('clear') comp_self._add_input_port('print') comp_self._add_input_port('insert') @@ -505,7 +502,7 @@ class PortTestCase(unittest.TestCase): def test_gen_sink_input_ports_getitem_invalid_key(self): class MySink(bt2._UserSinkComponent): - def __init__(comp_self, params): + def __init__(comp_self, params, obj): comp_self._add_input_port('clear') comp_self._add_input_port('print') comp_self._add_input_port('insert') @@ -513,7 +510,7 @@ class PortTestCase(unittest.TestCase): with self.assertRaises(KeyError): comp_self._input_ports['hello'] - def _consume(self): + def _user_consume(self): pass comp = self._create_comp(MySink) @@ -526,9 +523,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MySource(bt2._UserSourceComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): comp_self._add_output_port('clear') comp_self._add_output_port('print') comp_self._add_output_port('insert') @@ -541,9 +537,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MyFilter(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): comp_self._add_output_port('clear') comp_self._add_output_port('print') comp_self._add_output_port('insert') @@ -556,9 +551,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MyFilter(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): comp_self._add_input_port('clear') comp_self._add_input_port('print') comp_self._add_input_port('insert') @@ -568,12 +562,12 @@ class PortTestCase(unittest.TestCase): def test_gen_sink_input_ports_len(self): class MySink(bt2._UserSinkComponent): - def __init__(comp_self, params): + def __init__(comp_self, params, obj): comp_self._add_input_port('clear') comp_self._add_input_port('print') comp_self._add_input_port('insert') - def _consume(self): + def _user_consume(self): pass comp = self._create_comp(MySink) @@ -588,9 +582,8 @@ class PortTestCase(unittest.TestCase): port2 = None port3 = None - class MySource(bt2._UserSourceComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): nonlocal port1, port2, port3 port1 = comp_self._add_output_port('clear') port2 = comp_self._add_output_port('print') @@ -621,9 +614,8 @@ class PortTestCase(unittest.TestCase): port2 = None port3 = None - class MyFilter(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): nonlocal port1, port2, port3 port1 = comp_self._add_output_port('clear') port2 = comp_self._add_output_port('print') @@ -654,9 +646,8 @@ class PortTestCase(unittest.TestCase): port2 = None port3 = None - class MyFilter(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): nonlocal port1, port2, port3 port1 = comp_self._add_input_port('clear') port2 = comp_self._add_input_port('print') @@ -684,13 +675,13 @@ class PortTestCase(unittest.TestCase): port3 = None class MySink(bt2._UserSinkComponent): - def __init__(comp_self, params): + def __init__(comp_self, params, obj): nonlocal port1, port2, port3 port1 = comp_self._add_input_port('clear') port2 = comp_self._add_input_port('print') port3 = comp_self._add_input_port('insert') - def _consume(self): + def _user_consume(self): pass comp = self._create_comp(MySink) @@ -711,10 +702,10 @@ class PortTestCase(unittest.TestCase): def test_name(self): class MySink(bt2._UserSinkComponent): - def __init__(comp_self, params): + def __init__(comp_self, params, obj): comp_self._add_input_port('clear') - def _consume(self): + def _user_consume(self): pass comp = self._create_comp(MySink) @@ -722,10 +713,10 @@ class PortTestCase(unittest.TestCase): def test_connection_none(self): class MySink(bt2._UserSinkComponent): - def __init__(comp_self, params): + def __init__(comp_self, params, obj): comp_self._add_input_port('clear') - def _consume(self): + def _user_consume(self): pass comp = self._create_comp(MySink) @@ -733,10 +724,10 @@ class PortTestCase(unittest.TestCase): def test_is_connected_false(self): class MySink(bt2._UserSinkComponent): - def __init__(comp_self, params): + def __init__(comp_self, params, obj): comp_self._add_input_port('clear') - def _consume(self): + def _user_consume(self): pass comp = self._create_comp(MySink) @@ -744,33 +735,33 @@ class PortTestCase(unittest.TestCase): def test_self_name(self): class MySink(bt2._UserSinkComponent): - def __init__(comp_self, params): + def __init__(comp_self, params, obj): port = comp_self._add_input_port('clear') self.assertEqual(port.name, 'clear') - def _consume(self): + def _user_consume(self): pass self._create_comp(MySink) def test_self_connection_none(self): class MySink(bt2._UserSinkComponent): - def __init__(comp_self, params): + def __init__(comp_self, params, obj): port = comp_self._add_input_port('clear') self.assertIsNone(port.connection) - def _consume(self): + def _user_consume(self): pass self._create_comp(MySink) def test_self_is_connected_false(self): class MySink(bt2._UserSinkComponent): - def __init__(comp_self, params): + def __init__(comp_self, params, obj): port = comp_self._add_input_port('clear') self.assertFalse(port.is_connected) - def _consume(self): + def _user_consume(self): pass self._create_comp(MySink) @@ -780,9 +771,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MySource(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MySource(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): nonlocal user_datas p = comp_self._add_output_port('port1') @@ -800,9 +790,8 @@ class PortTestCase(unittest.TestCase): def __next__(self): raise bt2.Stop - class MyFilter(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): nonlocal user_datas p = comp_self._add_output_port('port1') @@ -818,17 +807,17 @@ class PortTestCase(unittest.TestCase): user_datas = [] comp = self._create_comp(MyFilter) - self.assertEqual(user_datas, - [None, 'user data string', None, {'user data': 'dict'}]) + self.assertEqual( + user_datas, [None, 'user data string', None, {'user data': 'dict'}] + ) def test_sink_self_port_user_data(self): class MyIter(bt2._UserMessageIterator): def __next__(self): raise bt2.Stop - class MySink(bt2._UserFilterComponent, - message_iterator_class=MyIter): - def __init__(comp_self, params): + class MySink(bt2._UserFilterComponent, message_iterator_class=MyIter): + def __init__(comp_self, params, obj): nonlocal user_datas p = comp_self._add_input_port('port1')