projects / babeltrace.git / blame
| Commit | Line | Data |
|---|---|---|
| d2d857a8 MJ |
1 | # |
| 2 | # Copyright (C) 2019 EfficiOS Inc. | |
| 3 | # | |
| 4 | # This program is free software; you can redistribute it and/or | |
| 5 | # modify it under the terms of the GNU General Public License | |
| 6 | # as published by the Free Software Foundation; only version 2 | |
| 7 | # of the License. | |
| 8 | # | |
| 9 | # This program is distributed in the hope that it will be useful, | |
| 10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
| 12 | # GNU General Public License for more details. | |
| 13 | # | |
| 14 | # You should have received a copy of the GNU General Public License | |
| 15 | # along with this program; if not, write to the Free Software | |
| 16 | # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. | |
| 17 | # | |
| 18 | ||
| 811644b8 | 19 | import unittest |
| 811644b8 | 20 | import bt2 |
| 5813b3a3 | 21 | from bt2 import port as bt2_port |
| 811644b8 PP |
22 | |
| 23 | ||
| 24 | class PortTestCase(unittest.TestCase): | |
| 25 | @staticmethod | |
| 26 | def _create_comp(comp_cls, name=None): | |
| 27 | graph = bt2.Graph() | |
| 28 | ||
| 29 | if name is None: | |
| 30 | name = 'comp' | |
| 31 | ||
| 32 | return graph.add_component(comp_cls, name) | |
| 33 | ||
| 34 | def test_src_add_output_port(self): | |
| 5602ef81 | 35 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
36 | def __next__(self): |
| 37 | raise bt2.Stop | |
| 38 | ||
| cfbd7cf3 | 39 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
| 59225a3e | 40 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
41 | port = comp_self._add_output_port('out') |
| 42 | self.assertEqual(port.name, 'out') | |
| 43 | ||
| 44 | comp = self._create_comp(MySource) | |
| 45 | self.assertEqual(len(comp.output_ports), 1) | |
| 5813b3a3 | 46 | self.assertIs(type(comp.output_ports['out']), bt2_port._OutputPortConst) |
| 811644b8 | 47 | |
| 811644b8 | 48 | def test_flt_add_output_port(self): |
| 5602ef81 | 49 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
50 | def __next__(self): |
| 51 | raise bt2.Stop | |
| 52 | ||
| cfbd7cf3 | 53 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 54 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
55 | port = comp_self._add_output_port('out') |
| 56 | self.assertEqual(port.name, 'out') | |
| 57 | ||
| 58 | comp = self._create_comp(MyFilter) | |
| 59 | self.assertEqual(len(comp.output_ports), 1) | |
| 60 | ||
| 61 | def test_flt_add_input_port(self): | |
| 5602ef81 | 62 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
63 | def __next__(self): |
| 64 | raise bt2.Stop | |
| 65 | ||
| cfbd7cf3 | 66 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 67 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
68 | port = comp_self._add_input_port('in') |
| 69 | self.assertEqual(port.name, 'in') | |
| 70 | ||
| 71 | comp = self._create_comp(MyFilter) | |
| 72 | self.assertEqual(len(comp.input_ports), 1) | |
| 5813b3a3 | 73 | self.assertIs(type(comp.input_ports['in']), bt2_port._InputPortConst) |
| 811644b8 PP |
74 | |
| 75 | def test_sink_add_input_port(self): | |
| 76 | class MySink(bt2._UserSinkComponent): | |
| 59225a3e | 77 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
78 | port = comp_self._add_input_port('in') |
| 79 | self.assertEqual(port.name, 'in') | |
| 80 | ||
| 6a91742b | 81 | def _user_consume(self): |
| a01b452b SM |
82 | pass |
| 83 | ||
| 811644b8 PP |
84 | comp = self._create_comp(MySink) |
| 85 | self.assertEqual(len(comp.input_ports), 1) | |
| 86 | ||
| 87 | def test_user_src_output_ports_getitem(self): | |
| 5602ef81 | 88 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
89 | def __next__(self): |
| 90 | raise bt2.Stop | |
| 91 | ||
| cfbd7cf3 | 92 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
| 59225a3e | 93 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
94 | port1 = comp_self._add_output_port('clear') |
| 95 | port2 = comp_self._add_output_port('print') | |
| 96 | port3 = comp_self._add_output_port('insert') | |
| 97 | self.assertEqual(port3.addr, comp_self._output_ports['insert'].addr) | |
| 98 | self.assertEqual(port2.addr, comp_self._output_ports['print'].addr) | |
| 99 | self.assertEqual(port1.addr, comp_self._output_ports['clear'].addr) | |
| 100 | ||
| 894a8df5 | 101 | self._create_comp(MySource) |
| 811644b8 PP |
102 | |
| 103 | def test_user_flt_output_ports_getitem(self): | |
| 5602ef81 | 104 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
105 | def __next__(self): |
| 106 | raise bt2.Stop | |
| 107 | ||
| cfbd7cf3 | 108 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 109 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
110 | port1 = comp_self._add_output_port('clear') |
| 111 | port2 = comp_self._add_output_port('print') | |
| 112 | port3 = comp_self._add_output_port('insert') | |
| 113 | self.assertEqual(port3.addr, comp_self._output_ports['insert'].addr) | |
| 114 | self.assertEqual(port2.addr, comp_self._output_ports['print'].addr) | |
| 115 | self.assertEqual(port1.addr, comp_self._output_ports['clear'].addr) | |
| 116 | ||
| 894a8df5 | 117 | self._create_comp(MyFilter) |
| 811644b8 PP |
118 | |
| 119 | def test_user_flt_input_ports_getitem(self): | |
| 5602ef81 | 120 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
121 | def __next__(self): |
| 122 | raise bt2.Stop | |
| 123 | ||
| cfbd7cf3 | 124 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 125 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
126 | port1 = comp_self._add_input_port('clear') |
| 127 | port2 = comp_self._add_input_port('print') | |
| 128 | port3 = comp_self._add_input_port('insert') | |
| 129 | self.assertEqual(port3.addr, comp_self._input_ports['insert'].addr) | |
| 130 | self.assertEqual(port2.addr, comp_self._input_ports['print'].addr) | |
| 131 | self.assertEqual(port1.addr, comp_self._input_ports['clear'].addr) | |
| 132 | ||
| 894a8df5 | 133 | self._create_comp(MyFilter) |
| 811644b8 PP |
134 | |
| 135 | def test_user_sink_input_ports_getitem(self): | |
| 136 | class MySink(bt2._UserSinkComponent): | |
| 59225a3e | 137 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
138 | port1 = comp_self._add_input_port('clear') |
| 139 | port2 = comp_self._add_input_port('print') | |
| 140 | port3 = comp_self._add_input_port('insert') | |
| 141 | self.assertEqual(port3.addr, comp_self._input_ports['insert'].addr) | |
| 142 | self.assertEqual(port2.addr, comp_self._input_ports['print'].addr) | |
| 143 | self.assertEqual(port1.addr, comp_self._input_ports['clear'].addr) | |
| 144 | ||
| 6a91742b | 145 | def _user_consume(self): |
| a01b452b SM |
146 | pass |
| 147 | ||
| 894a8df5 | 148 | self._create_comp(MySink) |
| 811644b8 PP |
149 | |
| 150 | def test_user_src_output_ports_getitem_invalid_key(self): | |
| 5602ef81 | 151 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
152 | def __next__(self): |
| 153 | raise bt2.Stop | |
| 154 | ||
| cfbd7cf3 | 155 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
| 59225a3e | 156 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
157 | comp_self._add_output_port('clear') |
| 158 | comp_self._add_output_port('print') | |
| 159 | comp_self._add_output_port('insert') | |
| 160 | ||
| 161 | with self.assertRaises(KeyError): | |
| 162 | comp_self._output_ports['hello'] | |
| 163 | ||
| 894a8df5 | 164 | self._create_comp(MySource) |
| 811644b8 PP |
165 | |
| 166 | def test_user_flt_output_ports_getitem_invalid_key(self): | |
| 5602ef81 | 167 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
168 | def __next__(self): |
| 169 | raise bt2.Stop | |
| 170 | ||
| cfbd7cf3 | 171 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 172 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
173 | comp_self._add_output_port('clear') |
| 174 | comp_self._add_output_port('print') | |
| 175 | comp_self._add_output_port('insert') | |
| 176 | ||
| 177 | with self.assertRaises(KeyError): | |
| 178 | comp_self._output_ports['hello'] | |
| 179 | ||
| 894a8df5 | 180 | self._create_comp(MyFilter) |
| 811644b8 PP |
181 | |
| 182 | def test_user_flt_input_ports_getitem_invalid_key(self): | |
| 5602ef81 | 183 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
184 | def __next__(self): |
| 185 | raise bt2.Stop | |
| 186 | ||
| cfbd7cf3 | 187 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 188 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
189 | comp_self._add_input_port('clear') |
| 190 | comp_self._add_input_port('print') | |
| 191 | comp_self._add_input_port('insert') | |
| 192 | ||
| 193 | with self.assertRaises(KeyError): | |
| 194 | comp_self._input_ports['hello'] | |
| 195 | ||
| 894a8df5 | 196 | self._create_comp(MyFilter) |
| 811644b8 PP |
197 | |
| 198 | def test_user_sink_input_ports_getitem_invalid_key(self): | |
| 199 | class MySink(bt2._UserSinkComponent): | |
| 59225a3e | 200 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
201 | comp_self._add_input_port('clear') |
| 202 | comp_self._add_input_port('print') | |
| 203 | comp_self._add_input_port('insert') | |
| 204 | ||
| 205 | with self.assertRaises(KeyError): | |
| 206 | comp_self._input_ports['hello'] | |
| 207 | ||
| 6a91742b | 208 | def _user_consume(self): |
| a01b452b SM |
209 | pass |
| 210 | ||
| 894a8df5 | 211 | self._create_comp(MySink) |
| 811644b8 PP |
212 | |
| 213 | def test_user_src_output_ports_len(self): | |
| 5602ef81 | 214 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
215 | def __next__(self): |
| 216 | raise bt2.Stop | |
| 217 | ||
| cfbd7cf3 | 218 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
| 59225a3e | 219 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
220 | comp_self._add_output_port('clear') |
| 221 | comp_self._add_output_port('print') | |
| 222 | comp_self._add_output_port('insert') | |
| 223 | self.assertEqual(len(comp_self._output_ports), 3) | |
| 224 | ||
| 894a8df5 | 225 | self._create_comp(MySource) |
| 811644b8 PP |
226 | |
| 227 | def test_user_flt_output_ports_len(self): | |
| 5602ef81 | 228 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
229 | def __next__(self): |
| 230 | raise bt2.Stop | |
| 231 | ||
| cfbd7cf3 | 232 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 233 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
234 | comp_self._add_output_port('clear') |
| 235 | comp_self._add_output_port('print') | |
| 236 | comp_self._add_output_port('insert') | |
| 237 | self.assertEqual(len(comp_self._output_ports), 3) | |
| 238 | ||
| 894a8df5 | 239 | self._create_comp(MyFilter) |
| 811644b8 PP |
240 | |
| 241 | def test_user_flt_input_ports_len(self): | |
| 5602ef81 | 242 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
243 | def __next__(self): |
| 244 | raise bt2.Stop | |
| 245 | ||
| cfbd7cf3 | 246 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 247 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
248 | comp_self._add_input_port('clear') |
| 249 | comp_self._add_input_port('print') | |
| 250 | comp_self._add_input_port('insert') | |
| 251 | self.assertEqual(len(comp_self._input_ports), 3) | |
| 252 | ||
| 894a8df5 | 253 | self._create_comp(MyFilter) |
| 811644b8 PP |
254 | |
| 255 | def test_user_sink_input_ports_len(self): | |
| 256 | class MySink(bt2._UserSinkComponent): | |
| 59225a3e | 257 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
258 | comp_self._add_input_port('clear') |
| 259 | comp_self._add_input_port('print') | |
| 260 | comp_self._add_input_port('insert') | |
| 261 | self.assertEqual(len(comp_self._input_ports), 3) | |
| 262 | ||
| 6a91742b | 263 | def _user_consume(self): |
| a01b452b SM |
264 | pass |
| 265 | ||
| 894a8df5 | 266 | self._create_comp(MySink) |
| 811644b8 PP |
267 | |
| 268 | def test_user_src_output_ports_iter(self): | |
| 5602ef81 | 269 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
270 | def __next__(self): |
| 271 | raise bt2.Stop | |
| 272 | ||
| cfbd7cf3 | 273 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
| 59225a3e | 274 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
275 | port1 = comp_self._add_output_port('clear') |
| 276 | port2 = comp_self._add_output_port('print') | |
| 277 | port3 = comp_self._add_output_port('insert') | |
| 278 | ports = [] | |
| 279 | ||
| 280 | for port_name, port in comp_self._output_ports.items(): | |
| 281 | ports.append((port_name, port)) | |
| 282 | ||
| 283 | self.assertEqual(ports[0][0], 'clear') | |
| 284 | self.assertEqual(ports[0][1].addr, port1.addr) | |
| 285 | self.assertEqual(ports[1][0], 'print') | |
| 286 | self.assertEqual(ports[1][1].addr, port2.addr) | |
| 287 | self.assertEqual(ports[2][0], 'insert') | |
| 288 | self.assertEqual(ports[2][1].addr, port3.addr) | |
| 289 | ||
| 894a8df5 | 290 | self._create_comp(MySource) |
| 811644b8 PP |
291 | |
| 292 | def test_user_flt_output_ports_iter(self): | |
| 5602ef81 | 293 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
294 | def __next__(self): |
| 295 | raise bt2.Stop | |
| 296 | ||
| cfbd7cf3 | 297 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 298 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
299 | port1 = comp_self._add_output_port('clear') |
| 300 | port2 = comp_self._add_output_port('print') | |
| 301 | port3 = comp_self._add_output_port('insert') | |
| 302 | ports = [] | |
| 303 | ||
| 304 | for port_name, port in comp_self._output_ports.items(): | |
| 305 | ports.append((port_name, port)) | |
| 306 | ||
| 307 | self.assertEqual(ports[0][0], 'clear') | |
| 308 | self.assertEqual(ports[0][1].addr, port1.addr) | |
| 309 | self.assertEqual(ports[1][0], 'print') | |
| 310 | self.assertEqual(ports[1][1].addr, port2.addr) | |
| 311 | self.assertEqual(ports[2][0], 'insert') | |
| 312 | self.assertEqual(ports[2][1].addr, port3.addr) | |
| 313 | ||
| 894a8df5 | 314 | self._create_comp(MyFilter) |
| 811644b8 PP |
315 | |
| 316 | def test_user_flt_input_ports_iter(self): | |
| 5602ef81 | 317 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
318 | def __next__(self): |
| 319 | raise bt2.Stop | |
| 320 | ||
| cfbd7cf3 | 321 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 322 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
323 | port1 = comp_self._add_input_port('clear') |
| 324 | port2 = comp_self._add_input_port('print') | |
| 325 | port3 = comp_self._add_input_port('insert') | |
| 326 | ports = [] | |
| 327 | ||
| 328 | for port_name, port in comp_self._input_ports.items(): | |
| 329 | ports.append((port_name, port)) | |
| 330 | ||
| 331 | self.assertEqual(ports[0][0], 'clear') | |
| 332 | self.assertEqual(ports[0][1].addr, port1.addr) | |
| 333 | self.assertEqual(ports[1][0], 'print') | |
| 334 | self.assertEqual(ports[1][1].addr, port2.addr) | |
| 335 | self.assertEqual(ports[2][0], 'insert') | |
| 336 | self.assertEqual(ports[2][1].addr, port3.addr) | |
| 337 | ||
| 894a8df5 | 338 | self._create_comp(MyFilter) |
| 811644b8 PP |
339 | |
| 340 | def test_user_sink_input_ports_iter(self): | |
| 341 | class MySink(bt2._UserSinkComponent): | |
| 59225a3e | 342 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
343 | port1 = comp_self._add_input_port('clear') |
| 344 | port2 = comp_self._add_input_port('print') | |
| 345 | port3 = comp_self._add_input_port('insert') | |
| 346 | ports = [] | |
| 347 | ||
| 348 | for port_name, port in comp_self._input_ports.items(): | |
| 349 | ports.append((port_name, port)) | |
| 350 | ||
| 351 | self.assertEqual(ports[0][0], 'clear') | |
| 352 | self.assertEqual(ports[0][1].addr, port1.addr) | |
| 353 | self.assertEqual(ports[1][0], 'print') | |
| 354 | self.assertEqual(ports[1][1].addr, port2.addr) | |
| 355 | self.assertEqual(ports[2][0], 'insert') | |
| 356 | self.assertEqual(ports[2][1].addr, port3.addr) | |
| 357 | ||
| 6a91742b | 358 | def _user_consume(self): |
| a01b452b SM |
359 | pass |
| 360 | ||
| 894a8df5 | 361 | self._create_comp(MySink) |
| 811644b8 PP |
362 | |
| 363 | def test_gen_src_output_ports_getitem(self): | |
| 5602ef81 | 364 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
365 | def __next__(self): |
| 366 | raise bt2.Stop | |
| 367 | ||
| 368 | port1 = None | |
| 369 | port2 = None | |
| 370 | port3 = None | |
| 371 | ||
| cfbd7cf3 | 372 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
| 59225a3e | 373 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
374 | nonlocal port1, port2, port3 |
| 375 | port1 = comp_self._add_output_port('clear') | |
| 376 | port2 = comp_self._add_output_port('print') | |
| 377 | port3 = comp_self._add_output_port('insert') | |
| 378 | ||
| 379 | comp = self._create_comp(MySource) | |
| 380 | self.assertEqual(port3.addr, comp.output_ports['insert'].addr) | |
| 381 | self.assertEqual(port2.addr, comp.output_ports['print'].addr) | |
| 382 | self.assertEqual(port1.addr, comp.output_ports['clear'].addr) | |
| 383 | del port1 | |
| 384 | del port2 | |
| 385 | del port3 | |
| 386 | ||
| 387 | def test_gen_flt_output_ports_getitem(self): | |
| 5602ef81 | 388 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
389 | def __next__(self): |
| 390 | raise bt2.Stop | |
| 391 | ||
| 392 | port1 = None | |
| 393 | port2 = None | |
| 394 | port3 = None | |
| 395 | ||
| cfbd7cf3 | 396 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 397 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
398 | nonlocal port1, port2, port3 |
| 399 | port1 = comp_self._add_output_port('clear') | |
| 400 | port2 = comp_self._add_output_port('print') | |
| 401 | port3 = comp_self._add_output_port('insert') | |
| 402 | ||
| 403 | comp = self._create_comp(MyFilter) | |
| 404 | self.assertEqual(port3.addr, comp.output_ports['insert'].addr) | |
| 405 | self.assertEqual(port2.addr, comp.output_ports['print'].addr) | |
| 406 | self.assertEqual(port1.addr, comp.output_ports['clear'].addr) | |
| 407 | del port1 | |
| 408 | del port2 | |
| 409 | del port3 | |
| 410 | ||
| 411 | def test_gen_flt_input_ports_getitem(self): | |
| 5602ef81 | 412 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
413 | def __next__(self): |
| 414 | raise bt2.Stop | |
| 415 | ||
| 416 | port1 = None | |
| 417 | port2 = None | |
| 418 | port3 = None | |
| 419 | ||
| cfbd7cf3 | 420 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 421 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
422 | nonlocal port1, port2, port3 |
| 423 | port1 = comp_self._add_input_port('clear') | |
| 424 | port2 = comp_self._add_input_port('print') | |
| 425 | port3 = comp_self._add_input_port('insert') | |
| 426 | ||
| 427 | comp = self._create_comp(MyFilter) | |
| 428 | self.assertEqual(port3.addr, comp.input_ports['insert'].addr) | |
| 429 | self.assertEqual(port2.addr, comp.input_ports['print'].addr) | |
| 430 | self.assertEqual(port1.addr, comp.input_ports['clear'].addr) | |
| 431 | del port1 | |
| 432 | del port2 | |
| 433 | del port3 | |
| 434 | ||
| 435 | def test_gen_sink_input_ports_getitem(self): | |
| 436 | port1 = None | |
| 437 | port2 = None | |
| 438 | port3 = None | |
| 439 | ||
| 440 | class MySink(bt2._UserSinkComponent): | |
| 59225a3e | 441 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
442 | nonlocal port1, port2, port3 |
| 443 | port1 = comp_self._add_input_port('clear') | |
| 444 | port2 = comp_self._add_input_port('print') | |
| 445 | port3 = comp_self._add_input_port('insert') | |
| 446 | ||
| 6a91742b | 447 | def _user_consume(self): |
| a01b452b SM |
448 | pass |
| 449 | ||
| 811644b8 PP |
450 | comp = self._create_comp(MySink) |
| 451 | self.assertEqual(port3.addr, comp.input_ports['insert'].addr) | |
| 452 | self.assertEqual(port2.addr, comp.input_ports['print'].addr) | |
| 453 | self.assertEqual(port1.addr, comp.input_ports['clear'].addr) | |
| 454 | del port1 | |
| 455 | del port2 | |
| 456 | del port3 | |
| 457 | ||
| 458 | def test_gen_src_output_ports_getitem_invalid_key(self): | |
| 5602ef81 | 459 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
460 | def __next__(self): |
| 461 | raise bt2.Stop | |
| 462 | ||
| cfbd7cf3 | 463 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
| 59225a3e | 464 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
465 | comp_self._add_output_port('clear') |
| 466 | comp_self._add_output_port('print') | |
| 467 | comp_self._add_output_port('insert') | |
| 468 | ||
| 469 | comp = self._create_comp(MySource) | |
| 470 | ||
| 471 | with self.assertRaises(KeyError): | |
| 472 | comp.output_ports['hello'] | |
| 473 | ||
| 474 | def test_gen_flt_output_ports_getitem_invalid_key(self): | |
| 5602ef81 | 475 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
476 | def __next__(self): |
| 477 | raise bt2.Stop | |
| 478 | ||
| cfbd7cf3 | 479 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 480 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
481 | comp_self._add_output_port('clear') |
| 482 | comp_self._add_output_port('print') | |
| 483 | comp_self._add_output_port('insert') | |
| 484 | ||
| 485 | comp = self._create_comp(MyFilter) | |
| 486 | ||
| 487 | with self.assertRaises(KeyError): | |
| 488 | comp.output_ports['hello'] | |
| 489 | ||
| 490 | def test_gen_flt_input_ports_getitem_invalid_key(self): | |
| 5602ef81 | 491 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
492 | def __next__(self): |
| 493 | raise bt2.Stop | |
| 494 | ||
| cfbd7cf3 | 495 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 496 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
497 | comp_self._add_input_port('clear') |
| 498 | comp_self._add_input_port('print') | |
| 499 | comp_self._add_input_port('insert') | |
| 500 | ||
| 501 | comp = self._create_comp(MyFilter) | |
| 502 | ||
| 503 | with self.assertRaises(KeyError): | |
| 504 | comp.input_ports['hello'] | |
| 505 | ||
| 506 | def test_gen_sink_input_ports_getitem_invalid_key(self): | |
| 507 | class MySink(bt2._UserSinkComponent): | |
| 59225a3e | 508 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
509 | comp_self._add_input_port('clear') |
| 510 | comp_self._add_input_port('print') | |
| 511 | comp_self._add_input_port('insert') | |
| 512 | ||
| 513 | with self.assertRaises(KeyError): | |
| 514 | comp_self._input_ports['hello'] | |
| 515 | ||
| 6a91742b | 516 | def _user_consume(self): |
| a01b452b SM |
517 | pass |
| 518 | ||
| 811644b8 PP |
519 | comp = self._create_comp(MySink) |
| 520 | ||
| 521 | with self.assertRaises(KeyError): | |
| 522 | comp.input_ports['hello'] | |
| 523 | ||
| 524 | def test_gen_src_output_ports_len(self): | |
| 5602ef81 | 525 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
526 | def __next__(self): |
| 527 | raise bt2.Stop | |
| 528 | ||
| cfbd7cf3 | 529 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
| 59225a3e | 530 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
531 | comp_self._add_output_port('clear') |
| 532 | comp_self._add_output_port('print') | |
| 533 | comp_self._add_output_port('insert') | |
| 534 | ||
| 535 | comp = self._create_comp(MySource) | |
| 536 | self.assertEqual(len(comp.output_ports), 3) | |
| 537 | ||
| 538 | def test_gen_flt_output_ports_len(self): | |
| 5602ef81 | 539 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
540 | def __next__(self): |
| 541 | raise bt2.Stop | |
| 542 | ||
| cfbd7cf3 | 543 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 544 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
545 | comp_self._add_output_port('clear') |
| 546 | comp_self._add_output_port('print') | |
| 547 | comp_self._add_output_port('insert') | |
| 548 | ||
| 549 | comp = self._create_comp(MyFilter) | |
| 550 | self.assertEqual(len(comp.output_ports), 3) | |
| 551 | ||
| 552 | def test_gen_flt_input_ports_len(self): | |
| 5602ef81 | 553 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
554 | def __next__(self): |
| 555 | raise bt2.Stop | |
| 556 | ||
| cfbd7cf3 | 557 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 558 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
559 | comp_self._add_input_port('clear') |
| 560 | comp_self._add_input_port('print') | |
| 561 | comp_self._add_input_port('insert') | |
| 562 | ||
| 563 | comp = self._create_comp(MyFilter) | |
| 564 | self.assertEqual(len(comp.input_ports), 3) | |
| 565 | ||
| 566 | def test_gen_sink_input_ports_len(self): | |
| 567 | class MySink(bt2._UserSinkComponent): | |
| 59225a3e | 568 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
569 | comp_self._add_input_port('clear') |
| 570 | comp_self._add_input_port('print') | |
| 571 | comp_self._add_input_port('insert') | |
| 572 | ||
| 6a91742b | 573 | def _user_consume(self): |
| a01b452b SM |
574 | pass |
| 575 | ||
| 811644b8 PP |
576 | comp = self._create_comp(MySink) |
| 577 | self.assertEqual(len(comp.input_ports), 3) | |
| 578 | ||
| 579 | def test_gen_src_output_ports_iter(self): | |
| 5602ef81 | 580 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
581 | def __next__(self): |
| 582 | raise bt2.Stop | |
| 583 | ||
| 584 | port1 = None | |
| 585 | port2 = None | |
| 586 | port3 = None | |
| 587 | ||
| cfbd7cf3 | 588 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
| 59225a3e | 589 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
590 | nonlocal port1, port2, port3 |
| 591 | port1 = comp_self._add_output_port('clear') | |
| 592 | port2 = comp_self._add_output_port('print') | |
| 593 | port3 = comp_self._add_output_port('insert') | |
| 594 | ||
| 595 | comp = self._create_comp(MySource) | |
| 596 | ports = [] | |
| 597 | ||
| 598 | for port_name, port in comp.output_ports.items(): | |
| 599 | ports.append((port_name, port)) | |
| 600 | ||
| 601 | self.assertEqual(ports[0][0], 'clear') | |
| 602 | self.assertEqual(ports[0][1].addr, port1.addr) | |
| 603 | self.assertEqual(ports[1][0], 'print') | |
| 604 | self.assertEqual(ports[1][1].addr, port2.addr) | |
| 605 | self.assertEqual(ports[2][0], 'insert') | |
| 606 | self.assertEqual(ports[2][1].addr, port3.addr) | |
| 607 | del port1 | |
| 608 | del port2 | |
| 609 | del port3 | |
| 610 | ||
| 611 | def test_gen_flt_output_ports_iter(self): | |
| 5602ef81 | 612 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
613 | def __next__(self): |
| 614 | raise bt2.Stop | |
| 615 | ||
| 616 | port1 = None | |
| 617 | port2 = None | |
| 618 | port3 = None | |
| 619 | ||
| cfbd7cf3 | 620 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 621 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
622 | nonlocal port1, port2, port3 |
| 623 | port1 = comp_self._add_output_port('clear') | |
| 624 | port2 = comp_self._add_output_port('print') | |
| 625 | port3 = comp_self._add_output_port('insert') | |
| 626 | ||
| 627 | comp = self._create_comp(MyFilter) | |
| 628 | ports = [] | |
| 629 | ||
| 630 | for port_name, port in comp.output_ports.items(): | |
| 631 | ports.append((port_name, port)) | |
| 632 | ||
| 633 | self.assertEqual(ports[0][0], 'clear') | |
| 634 | self.assertEqual(ports[0][1].addr, port1.addr) | |
| 635 | self.assertEqual(ports[1][0], 'print') | |
| 636 | self.assertEqual(ports[1][1].addr, port2.addr) | |
| 637 | self.assertEqual(ports[2][0], 'insert') | |
| 638 | self.assertEqual(ports[2][1].addr, port3.addr) | |
| 639 | del port1 | |
| 640 | del port2 | |
| 641 | del port3 | |
| 642 | ||
| 643 | def test_gen_flt_input_ports_iter(self): | |
| 5602ef81 | 644 | class MyIter(bt2._UserMessageIterator): |
| 811644b8 PP |
645 | def __next__(self): |
| 646 | raise bt2.Stop | |
| 647 | ||
| 648 | port1 = None | |
| 649 | port2 = None | |
| 650 | port3 = None | |
| 651 | ||
| cfbd7cf3 | 652 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 653 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
654 | nonlocal port1, port2, port3 |
| 655 | port1 = comp_self._add_input_port('clear') | |
| 656 | port2 = comp_self._add_input_port('print') | |
| 657 | port3 = comp_self._add_input_port('insert') | |
| 658 | ||
| 659 | comp = self._create_comp(MyFilter) | |
| 660 | ports = [] | |
| 661 | ||
| 662 | for port_name, port in comp.input_ports.items(): | |
| 663 | ports.append((port_name, port)) | |
| 664 | ||
| 665 | self.assertEqual(ports[0][0], 'clear') | |
| 666 | self.assertEqual(ports[0][1].addr, port1.addr) | |
| 667 | self.assertEqual(ports[1][0], 'print') | |
| 668 | self.assertEqual(ports[1][1].addr, port2.addr) | |
| 669 | self.assertEqual(ports[2][0], 'insert') | |
| 670 | self.assertEqual(ports[2][1].addr, port3.addr) | |
| 671 | del port1 | |
| 672 | del port2 | |
| 673 | del port3 | |
| 674 | ||
| 675 | def test_gen_sink_input_ports_iter(self): | |
| 676 | port1 = None | |
| 677 | port2 = None | |
| 678 | port3 = None | |
| 679 | ||
| 680 | class MySink(bt2._UserSinkComponent): | |
| 59225a3e | 681 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
682 | nonlocal port1, port2, port3 |
| 683 | port1 = comp_self._add_input_port('clear') | |
| 684 | port2 = comp_self._add_input_port('print') | |
| 685 | port3 = comp_self._add_input_port('insert') | |
| 686 | ||
| 6a91742b | 687 | def _user_consume(self): |
| a01b452b SM |
688 | pass |
| 689 | ||
| 811644b8 PP |
690 | comp = self._create_comp(MySink) |
| 691 | ports = [] | |
| 692 | ||
| 693 | for port_name, port in comp.input_ports.items(): | |
| 694 | ports.append((port_name, port)) | |
| 695 | ||
| 696 | self.assertEqual(ports[0][0], 'clear') | |
| 697 | self.assertEqual(ports[0][1].addr, port1.addr) | |
| 698 | self.assertEqual(ports[1][0], 'print') | |
| 699 | self.assertEqual(ports[1][1].addr, port2.addr) | |
| 700 | self.assertEqual(ports[2][0], 'insert') | |
| 701 | self.assertEqual(ports[2][1].addr, port3.addr) | |
| 702 | del port1 | |
| 703 | del port2 | |
| 704 | del port3 | |
| 705 | ||
| 706 | def test_name(self): | |
| 707 | class MySink(bt2._UserSinkComponent): | |
| 59225a3e | 708 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
709 | comp_self._add_input_port('clear') |
| 710 | ||
| 6a91742b | 711 | def _user_consume(self): |
| a01b452b SM |
712 | pass |
| 713 | ||
| 811644b8 PP |
714 | comp = self._create_comp(MySink) |
| 715 | self.assertEqual(comp.input_ports['clear'].name, 'clear') | |
| 716 | ||
| 811644b8 PP |
717 | def test_connection_none(self): |
| 718 | class MySink(bt2._UserSinkComponent): | |
| 59225a3e | 719 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
720 | comp_self._add_input_port('clear') |
| 721 | ||
| 6a91742b | 722 | def _user_consume(self): |
| a01b452b SM |
723 | pass |
| 724 | ||
| 811644b8 PP |
725 | comp = self._create_comp(MySink) |
| 726 | self.assertIsNone(comp.input_ports['clear'].connection) | |
| 727 | ||
| 728 | def test_is_connected_false(self): | |
| 729 | class MySink(bt2._UserSinkComponent): | |
| 59225a3e | 730 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
731 | comp_self._add_input_port('clear') |
| 732 | ||
| 6a91742b | 733 | def _user_consume(self): |
| a01b452b SM |
734 | pass |
| 735 | ||
| 811644b8 PP |
736 | comp = self._create_comp(MySink) |
| 737 | self.assertFalse(comp.input_ports['clear'].is_connected) | |
| 738 | ||
| 894a8df5 | 739 | def test_self_name(self): |
| 811644b8 | 740 | class MySink(bt2._UserSinkComponent): |
| 59225a3e | 741 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
742 | port = comp_self._add_input_port('clear') |
| 743 | self.assertEqual(port.name, 'clear') | |
| 744 | ||
| 6a91742b | 745 | def _user_consume(self): |
| a01b452b SM |
746 | pass |
| 747 | ||
| 894a8df5 | 748 | self._create_comp(MySink) |
| 811644b8 | 749 | |
| 894a8df5 | 750 | def test_self_connection_none(self): |
| 811644b8 | 751 | class MySink(bt2._UserSinkComponent): |
| 59225a3e | 752 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
753 | port = comp_self._add_input_port('clear') |
| 754 | self.assertIsNone(port.connection) | |
| 755 | ||
| 6a91742b | 756 | def _user_consume(self): |
| a01b452b SM |
757 | pass |
| 758 | ||
| 894a8df5 | 759 | self._create_comp(MySink) |
| 811644b8 | 760 | |
| 894a8df5 | 761 | def test_self_is_connected_false(self): |
| 811644b8 | 762 | class MySink(bt2._UserSinkComponent): |
| 59225a3e | 763 | def __init__(comp_self, config, params, obj): |
| 811644b8 PP |
764 | port = comp_self._add_input_port('clear') |
| 765 | self.assertFalse(port.is_connected) | |
| 766 | ||
| 6a91742b | 767 | def _user_consume(self): |
| a01b452b SM |
768 | pass |
| 769 | ||
| 894a8df5 | 770 | self._create_comp(MySink) |
| 2e00bc76 SM |
771 | |
| 772 | def test_source_self_port_user_data(self): | |
| 773 | class MyIter(bt2._UserMessageIterator): | |
| 774 | def __next__(self): | |
| 775 | raise bt2.Stop | |
| 776 | ||
| cfbd7cf3 | 777 | class MySource(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 778 | def __init__(comp_self, config, params, obj): |
| 2e00bc76 SM |
779 | nonlocal user_datas |
| 780 | ||
| 781 | p = comp_self._add_output_port('port1') | |
| 782 | user_datas.append(p.user_data) | |
| 783 | p = comp_self._add_output_port('port2', 2) | |
| 784 | user_datas.append(p.user_data) | |
| 785 | ||
| 786 | user_datas = [] | |
| 787 | ||
| 082db648 | 788 | self._create_comp(MySource) |
| 2e00bc76 SM |
789 | self.assertEqual(user_datas, [None, 2]) |
| 790 | ||
| 791 | def test_filter_self_port_user_data(self): | |
| 792 | class MyIter(bt2._UserMessageIterator): | |
| 793 | def __next__(self): | |
| 794 | raise bt2.Stop | |
| 795 | ||
| cfbd7cf3 | 796 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 797 | def __init__(comp_self, config, params, obj): |
| 2e00bc76 SM |
798 | nonlocal user_datas |
| 799 | ||
| 800 | p = comp_self._add_output_port('port1') | |
| 801 | user_datas.append(p.user_data) | |
| 802 | p = comp_self._add_output_port('port2', 'user data string') | |
| 803 | user_datas.append(p.user_data) | |
| 804 | ||
| 805 | p = comp_self._add_input_port('port3') | |
| 806 | user_datas.append(p.user_data) | |
| 807 | p = comp_self._add_input_port('port4', user_data={'user data': 'dict'}) | |
| 808 | user_datas.append(p.user_data) | |
| 809 | ||
| 810 | user_datas = [] | |
| 811 | ||
| 082db648 | 812 | self._create_comp(MyFilter) |
| cfbd7cf3 FD |
813 | self.assertEqual( |
| 814 | user_datas, [None, 'user data string', None, {'user data': 'dict'}] | |
| 815 | ) | |
| 2e00bc76 SM |
816 | |
| 817 | def test_sink_self_port_user_data(self): | |
| 818 | class MyIter(bt2._UserMessageIterator): | |
| 819 | def __next__(self): | |
| 820 | raise bt2.Stop | |
| 821 | ||
| cfbd7cf3 | 822 | class MySink(bt2._UserFilterComponent, message_iterator_class=MyIter): |
| 59225a3e | 823 | def __init__(comp_self, config, params, obj): |
| 2e00bc76 SM |
824 | nonlocal user_datas |
| 825 | ||
| 826 | p = comp_self._add_input_port('port1') | |
| 827 | user_datas.append(p.user_data) | |
| 828 | p = comp_self._add_input_port('port2', set()) | |
| 829 | user_datas.append(p.user_data) | |
| 830 | ||
| 831 | user_datas = [] | |
| 832 | ||
| 082db648 | 833 | self._create_comp(MySink) |
| 2e00bc76 | 834 | self.assertEqual(user_datas, [None, set()]) |
| d14ddbba SM |
835 | |
| 836 | ||
| 837 | if __name__ == '__main__': | |
| 838 | unittest.main() |