Commit | Line | Data |
---|---|---|
32d2d479 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 | ||
f6a5e476 | 19 | import unittest |
f6a5e476 PP |
20 | import bt2 |
21 | ||
22 | ||
871a292a | 23 | class _MyIter(bt2._UserMessageIterator): |
a4dcfa96 | 24 | def __init__(self, self_output_port): |
871a292a SM |
25 | self._build_meta() |
26 | self._at = 0 | |
27 | ||
28 | def _build_meta(self): | |
29 | self._tc = self._component._create_trace_class() | |
30 | self._t = self._tc() | |
37a93d41 | 31 | self._sc = self._tc.create_stream_class(supports_packets=True) |
871a292a SM |
32 | self._ec = self._sc.create_event_class(name='salut') |
33 | self._my_int_ft = self._tc.create_signed_integer_field_class(32) | |
34 | payload_ft = self._tc.create_structure_field_class() | |
61d96b89 | 35 | payload_ft += [('my_int', self._my_int_ft)] |
871a292a SM |
36 | self._ec.payload_field_type = payload_ft |
37 | self._stream = self._t.create_stream(self._sc) | |
38 | self._packet = self._stream.create_packet() | |
39 | ||
40 | def _create_event(self, value): | |
41 | ev = self._ec() | |
42 | ev.payload_field['my_int'] = value | |
43 | ev.packet = self._packet | |
44 | return ev | |
45 | ||
46 | ||
f6a5e476 PP |
47 | class GraphTestCase(unittest.TestCase): |
48 | def setUp(self): | |
49 | self._graph = bt2.Graph() | |
50 | ||
51 | def tearDown(self): | |
52 | del self._graph | |
53 | ||
54 | def test_create_empty(self): | |
55 | graph = bt2.Graph() | |
56 | ||
57 | def test_add_component_user_cls(self): | |
58 | class MySink(bt2._UserSinkComponent): | |
819d0ae7 | 59 | def _user_consume(self): |
8a08af82 | 60 | pass |
f6a5e476 PP |
61 | |
62 | comp = self._graph.add_component(MySink, 'salut') | |
63 | self.assertEqual(comp.name, 'salut') | |
64 | ||
65 | def test_add_component_gen_cls(self): | |
66 | class MySink(bt2._UserSinkComponent): | |
819d0ae7 | 67 | def _user_consume(self): |
8a08af82 | 68 | pass |
f6a5e476 PP |
69 | |
70 | comp = self._graph.add_component(MySink, 'salut') | |
871a292a | 71 | assert comp |
c88be1c8 | 72 | comp2 = self._graph.add_component(comp.cls, 'salut2') |
f6a5e476 PP |
73 | self.assertEqual(comp2.name, 'salut2') |
74 | ||
75 | def test_add_component_params(self): | |
76 | comp_params = None | |
77 | ||
78 | class MySink(bt2._UserSinkComponent): | |
b20382e2 | 79 | def __init__(self, params, obj): |
f6a5e476 PP |
80 | nonlocal comp_params |
81 | comp_params = params | |
82 | ||
819d0ae7 | 83 | def _user_consume(self): |
8a08af82 | 84 | pass |
f6a5e476 PP |
85 | |
86 | params = {'hello': 23, 'path': '/path/to/stuff'} | |
87 | comp = self._graph.add_component(MySink, 'salut', params) | |
88 | self.assertEqual(params, comp_params) | |
89 | del comp_params | |
90 | ||
b20382e2 PP |
91 | def test_add_component_obj_python_comp_cls(self): |
92 | comp_obj = None | |
93 | ||
94 | class MySink(bt2._UserSinkComponent): | |
95 | def __init__(self, params, obj): | |
96 | nonlocal comp_obj | |
97 | comp_obj = obj | |
98 | ||
99 | def _user_consume(self): | |
100 | pass | |
101 | ||
102 | obj = object() | |
103 | comp = self._graph.add_component(MySink, 'salut', obj=obj) | |
104 | self.assertIs(comp_obj, obj) | |
105 | del comp_obj | |
106 | ||
107 | def test_add_component_obj_none_python_comp_cls(self): | |
108 | comp_obj = None | |
109 | ||
110 | class MySink(bt2._UserSinkComponent): | |
111 | def __init__(self, params, obj): | |
112 | nonlocal comp_obj | |
113 | comp_obj = obj | |
114 | ||
115 | def _user_consume(self): | |
116 | pass | |
117 | ||
118 | comp = self._graph.add_component(MySink, 'salut') | |
119 | self.assertIsNone(comp_obj) | |
120 | del comp_obj | |
121 | ||
122 | def test_add_component_obj_non_python_comp_cls(self): | |
123 | comp_obj = None | |
124 | ||
125 | plugin = bt2.find_plugin('text', find_in_user_dir=False, find_in_sys_dir=False) | |
126 | assert plugin is not None | |
127 | cc = plugin.source_component_classes['dmesg'] | |
128 | assert cc is not None | |
129 | ||
130 | with self.assertRaises(ValueError): | |
131 | comp = self._graph.add_component(cc, 'salut', obj=57) | |
132 | ||
f6a5e476 PP |
133 | def test_add_component_invalid_cls_type(self): |
134 | with self.assertRaises(TypeError): | |
135 | self._graph.add_component(int, 'salut') | |
136 | ||
78c432bb PP |
137 | def test_add_component_invalid_logging_level_type(self): |
138 | class MySink(bt2._UserSinkComponent): | |
819d0ae7 | 139 | def _user_consume(self): |
8a08af82 | 140 | pass |
78c432bb PP |
141 | |
142 | with self.assertRaises(TypeError): | |
143 | self._graph.add_component(MySink, 'salut', logging_level='yo') | |
144 | ||
145 | def test_add_component_invalid_logging_level_value(self): | |
146 | class MySink(bt2._UserSinkComponent): | |
819d0ae7 | 147 | def _user_consume(self): |
8a08af82 | 148 | pass |
78c432bb PP |
149 | |
150 | with self.assertRaises(ValueError): | |
151 | self._graph.add_component(MySink, 'salut', logging_level=12345) | |
152 | ||
153 | def test_add_component_logging_level(self): | |
154 | class MySink(bt2._UserSinkComponent): | |
819d0ae7 | 155 | def _user_consume(self): |
8a08af82 | 156 | pass |
78c432bb | 157 | |
61d96b89 FD |
158 | comp = self._graph.add_component( |
159 | MySink, 'salut', logging_level=bt2.LoggingLevel.DEBUG | |
160 | ) | |
78c432bb PP |
161 | self.assertEqual(comp.logging_level, bt2.LoggingLevel.DEBUG) |
162 | ||
f6a5e476 | 163 | def test_connect_ports(self): |
fa4c33e3 | 164 | class MyIter(bt2._UserMessageIterator): |
f6a5e476 PP |
165 | def __next__(self): |
166 | raise bt2.Stop | |
167 | ||
61d96b89 | 168 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
b20382e2 | 169 | def __init__(self, params, obj): |
f6a5e476 PP |
170 | self._add_output_port('out') |
171 | ||
172 | class MySink(bt2._UserSinkComponent): | |
b20382e2 | 173 | def __init__(self, params, obj): |
f6a5e476 PP |
174 | self._add_input_port('in') |
175 | ||
819d0ae7 | 176 | def _user_consume(self): |
f6a5e476 PP |
177 | raise bt2.Stop |
178 | ||
179 | src = self._graph.add_component(MySource, 'src') | |
180 | sink = self._graph.add_component(MySink, 'sink') | |
871a292a | 181 | |
61d96b89 FD |
182 | conn = self._graph.connect_ports( |
183 | src.output_ports['out'], sink.input_ports['in'] | |
184 | ) | |
f6a5e476 PP |
185 | self.assertTrue(src.output_ports['out'].is_connected) |
186 | self.assertTrue(sink.input_ports['in'].is_connected) | |
3e2cc5dc FD |
187 | self.assertEqual(src.output_ports['out'].connection.addr, conn.addr) |
188 | self.assertEqual(sink.input_ports['in'].connection.addr, conn.addr) | |
f6a5e476 PP |
189 | |
190 | def test_connect_ports_invalid_direction(self): | |
fa4c33e3 | 191 | class MyIter(bt2._UserMessageIterator): |
f6a5e476 PP |
192 | def __next__(self): |
193 | raise bt2.Stop | |
194 | ||
61d96b89 | 195 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
b20382e2 | 196 | def __init__(self, params, obj): |
f6a5e476 PP |
197 | self._add_output_port('out') |
198 | ||
199 | class MySink(bt2._UserSinkComponent): | |
b20382e2 | 200 | def __init__(self, params, obj): |
f6a5e476 PP |
201 | self._add_input_port('in') |
202 | ||
819d0ae7 | 203 | def _user_consume(self): |
f6a5e476 PP |
204 | raise bt2.Stop |
205 | ||
206 | src = self._graph.add_component(MySource, 'src') | |
207 | sink = self._graph.add_component(MySink, 'sink') | |
208 | ||
209 | with self.assertRaises(TypeError): | |
61d96b89 FD |
210 | conn = self._graph.connect_ports( |
211 | sink.input_ports['in'], src.output_ports['out'] | |
212 | ) | |
f6a5e476 | 213 | |
d73bb381 PP |
214 | def test_add_interrupter(self): |
215 | class MyIter(bt2._UserMessageIterator): | |
216 | def __next__(self): | |
217 | raise TypeError | |
218 | ||
219 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): | |
b20382e2 | 220 | def __init__(self, params, obj): |
d73bb381 PP |
221 | self._add_output_port('out') |
222 | ||
223 | class MySink(bt2._UserSinkComponent): | |
b20382e2 | 224 | def __init__(self, params, obj): |
d73bb381 PP |
225 | self._add_input_port('in') |
226 | ||
819d0ae7 | 227 | def _user_consume(self): |
d73bb381 PP |
228 | next(self._msg_iter) |
229 | ||
819d0ae7 | 230 | def _user_graph_is_configured(self): |
d73bb381 PP |
231 | self._msg_iter = self._create_input_port_message_iterator( |
232 | self._input_ports['in'] | |
233 | ) | |
234 | ||
235 | # add two interrupters, set one of them | |
236 | interrupter1 = bt2.Interrupter() | |
237 | interrupter2 = bt2.Interrupter() | |
238 | self._graph.add_interrupter(interrupter1) | |
239 | src = self._graph.add_component(MySource, 'src') | |
240 | sink = self._graph.add_component(MySink, 'sink') | |
241 | self._graph.connect_ports(src.output_ports['out'], sink.input_ports['in']) | |
242 | self._graph.add_interrupter(interrupter2) | |
243 | ||
244 | with self.assertRaises(bt2._Error): | |
245 | self._graph.run() | |
246 | ||
247 | interrupter2.set() | |
248 | ||
249 | with self.assertRaises(bt2.TryAgain): | |
250 | self._graph.run() | |
251 | ||
252 | interrupter2.reset() | |
253 | ||
254 | with self.assertRaises(bt2._Error): | |
255 | self._graph.run() | |
256 | ||
257 | # Test that Graph.run() raises bt2.Interrupted if the graph gets | |
258 | # interrupted during execution. | |
259 | def test_interrupt_while_running(self): | |
871a292a | 260 | class MyIter(_MyIter): |
9ef22b36 | 261 | def __next__(self): |
871a292a | 262 | return self._create_stream_beginning_message(self._stream) |
9ef22b36 | 263 | |
61d96b89 | 264 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
b20382e2 | 265 | def __init__(self, params, obj): |
9ef22b36 PP |
266 | self._add_output_port('out') |
267 | ||
268 | class MySink(bt2._UserSinkComponent): | |
b20382e2 | 269 | def __init__(self, params, obj): |
9ef22b36 PP |
270 | self._add_input_port('in') |
271 | ||
819d0ae7 | 272 | def _user_consume(self): |
d73bb381 | 273 | # Pretend that somebody asynchronously interrupted the graph. |
871a292a | 274 | nonlocal graph |
d73bb381 | 275 | graph.interrupt() |
871a292a | 276 | return next(self._msg_iter) |
9ef22b36 | 277 | |
819d0ae7 | 278 | def _user_graph_is_configured(self): |
692f1a01 PP |
279 | self._msg_iter = self._create_input_port_message_iterator( |
280 | self._input_ports['in'] | |
281 | ) | |
9ef22b36 | 282 | |
d73bb381 PP |
283 | graph = self._graph |
284 | up = self._graph.add_component(MySource, 'down') | |
285 | down = self._graph.add_component(MySink, 'up') | |
286 | self._graph.connect_ports(up.output_ports['out'], down.input_ports['in']) | |
287 | ||
288 | with self.assertRaises(bt2.TryAgain): | |
289 | self._graph.run() | |
f6a5e476 PP |
290 | |
291 | def test_run(self): | |
871a292a | 292 | class MyIter(_MyIter): |
f6a5e476 | 293 | def __next__(self): |
871a292a SM |
294 | if self._at == 9: |
295 | raise StopIteration | |
296 | ||
297 | if self._at == 0: | |
298 | msg = self._create_stream_beginning_message(self._stream) | |
299 | elif self._at == 1: | |
300 | msg = self._create_packet_beginning_message(self._packet) | |
301 | elif self._at == 7: | |
302 | msg = self._create_packet_end_message(self._packet) | |
303 | elif self._at == 8: | |
304 | msg = self._create_stream_end_message(self._stream) | |
305 | else: | |
306 | msg = self._create_event_message(self._ec, self._packet) | |
f6a5e476 | 307 | |
f6a5e476 | 308 | self._at += 1 |
fa4c33e3 | 309 | return msg |
f6a5e476 | 310 | |
61d96b89 | 311 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
b20382e2 | 312 | def __init__(self, params, obj): |
f6a5e476 PP |
313 | self._add_output_port('out') |
314 | ||
315 | class MySink(bt2._UserSinkComponent): | |
b20382e2 | 316 | def __init__(self, params, obj): |
871a292a | 317 | self._input_port = self._add_input_port('in') |
f6a5e476 PP |
318 | self._at = 0 |
319 | ||
819d0ae7 | 320 | def _user_consume(comp_self): |
fa4c33e3 | 321 | msg = next(comp_self._msg_iter) |
f6a5e476 PP |
322 | |
323 | if comp_self._at == 0: | |
c946c9de | 324 | self.assertIsInstance(msg, bt2._StreamBeginningMessage) |
f6a5e476 | 325 | elif comp_self._at == 1: |
c946c9de | 326 | self.assertIsInstance(msg, bt2._PacketBeginningMessage) |
f6a5e476 | 327 | elif comp_self._at >= 2 and comp_self._at <= 6: |
c946c9de | 328 | self.assertIsInstance(msg, bt2._EventMessage) |
c88be1c8 | 329 | self.assertEqual(msg.event.cls.name, 'salut') |
f6a5e476 | 330 | elif comp_self._at == 7: |
c946c9de | 331 | self.assertIsInstance(msg, bt2._PacketEndMessage) |
f6a5e476 | 332 | elif comp_self._at == 8: |
c946c9de | 333 | self.assertIsInstance(msg, bt2._StreamEndMessage) |
f6a5e476 PP |
334 | |
335 | comp_self._at += 1 | |
336 | ||
819d0ae7 | 337 | def _user_graph_is_configured(self): |
692f1a01 PP |
338 | self._msg_iter = self._create_input_port_message_iterator( |
339 | self._input_port | |
340 | ) | |
f6a5e476 PP |
341 | |
342 | src = self._graph.add_component(MySource, 'src') | |
343 | sink = self._graph.add_component(MySink, 'sink') | |
61d96b89 FD |
344 | conn = self._graph.connect_ports( |
345 | src.output_ports['out'], sink.input_ports['in'] | |
346 | ) | |
f6a5e476 PP |
347 | self._graph.run() |
348 | ||
22b4de6a PP |
349 | def test_run_once(self): |
350 | class MyIter(_MyIter): | |
351 | pass | |
352 | ||
353 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): | |
354 | def __init__(self, params, obj): | |
355 | self._add_output_port('out') | |
356 | ||
357 | class MySink(bt2._UserSinkComponent): | |
358 | def __init__(self, params, obj): | |
359 | self._input_port = self._add_input_port('in') | |
360 | ||
361 | def _user_consume(comp_self): | |
362 | nonlocal run_count | |
363 | run_count += 1 | |
364 | raise bt2.TryAgain | |
365 | ||
366 | run_count = 0 | |
367 | src = self._graph.add_component(MySource, 'src') | |
368 | sink = self._graph.add_component(MySink, 'sink') | |
369 | conn = self._graph.connect_ports( | |
370 | src.output_ports['out'], sink.input_ports['in'] | |
371 | ) | |
372 | ||
373 | with self.assertRaises(bt2.TryAgain): | |
374 | self._graph.run_once() | |
375 | ||
376 | self.assertEqual(run_count, 1) | |
377 | ||
378 | def test_run_once_stops(self): | |
379 | class MyIter(_MyIter): | |
380 | pass | |
381 | ||
382 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): | |
383 | def __init__(self, params, obj): | |
384 | self._add_output_port('out') | |
385 | ||
386 | class MySink(bt2._UserSinkComponent): | |
387 | def __init__(self, params, obj): | |
388 | self._input_port = self._add_input_port('in') | |
389 | ||
390 | def _user_consume(comp_self): | |
391 | raise bt2.Stop | |
392 | ||
393 | src = self._graph.add_component(MySource, 'src') | |
394 | sink = self._graph.add_component(MySink, 'sink') | |
395 | conn = self._graph.connect_ports( | |
396 | src.output_ports['out'], sink.input_ports['in'] | |
397 | ) | |
398 | ||
399 | with self.assertRaises(bt2.Stop): | |
400 | self._graph.run_once() | |
401 | ||
f6a5e476 | 402 | def test_run_again(self): |
871a292a | 403 | class MyIter(_MyIter): |
f6a5e476 | 404 | def __next__(self): |
871a292a | 405 | if self._at == 3: |
f6a5e476 PP |
406 | raise bt2.TryAgain |
407 | ||
871a292a SM |
408 | if self._at == 0: |
409 | msg = self._create_stream_beginning_message(self._stream) | |
410 | elif self._at == 1: | |
411 | msg = self._create_packet_beginning_message(self._packet) | |
412 | elif self._at == 2: | |
413 | msg = self._create_event_message(self._ec, self._packet) | |
414 | ||
f6a5e476 | 415 | self._at += 1 |
fa4c33e3 | 416 | return msg |
f6a5e476 | 417 | |
61d96b89 | 418 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
b20382e2 | 419 | def __init__(self, params, obj): |
f6a5e476 PP |
420 | self._add_output_port('out') |
421 | ||
422 | class MySink(bt2._UserSinkComponent): | |
b20382e2 | 423 | def __init__(self, params, obj): |
871a292a | 424 | self._input_port = self._add_input_port('in') |
f6a5e476 PP |
425 | self._at = 0 |
426 | ||
819d0ae7 | 427 | def _user_consume(comp_self): |
871a292a | 428 | msg = next(comp_self._msg_iter) |
f6a5e476 | 429 | if comp_self._at == 0: |
c946c9de | 430 | self.assertIsInstance(msg, bt2._StreamBeginningMessage) |
f6a5e476 | 431 | elif comp_self._at == 1: |
c946c9de | 432 | self.assertIsInstance(msg, bt2._PacketBeginningMessage) |
871a292a | 433 | elif comp_self._at == 2: |
c946c9de | 434 | self.assertIsInstance(msg, bt2._EventMessage) |
f6a5e476 | 435 | raise bt2.TryAgain |
871a292a SM |
436 | else: |
437 | pass | |
f6a5e476 PP |
438 | |
439 | comp_self._at += 1 | |
440 | ||
819d0ae7 | 441 | def _user_graph_is_configured(self): |
692f1a01 PP |
442 | self._msg_iter = self._create_input_port_message_iterator( |
443 | self._input_port | |
444 | ) | |
f6a5e476 PP |
445 | |
446 | src = self._graph.add_component(MySource, 'src') | |
447 | sink = self._graph.add_component(MySink, 'sink') | |
61d96b89 FD |
448 | conn = self._graph.connect_ports( |
449 | src.output_ports['out'], sink.input_ports['in'] | |
450 | ) | |
f6a5e476 PP |
451 | |
452 | with self.assertRaises(bt2.TryAgain): | |
453 | self._graph.run() | |
454 | ||
f6a5e476 | 455 | def test_run_error(self): |
871a292a | 456 | raised_in_sink = False |
f6a5e476 | 457 | |
871a292a | 458 | class MyIter(_MyIter): |
f6a5e476 | 459 | def __next__(self): |
871a292a SM |
460 | # If this gets called after the sink raised an exception, it is |
461 | # an error. | |
462 | nonlocal raised_in_sink | |
463 | assert raised_in_sink is False | |
464 | ||
465 | if self._at == 0: | |
466 | msg = self._create_stream_beginning_message(self._stream) | |
467 | elif self._at == 1: | |
468 | msg = self._create_packet_beginning_message(self._packet) | |
469 | elif self._at == 2 or self._at == 3: | |
470 | msg = self._create_event_message(self._ec, self._packet) | |
471 | else: | |
f6a5e476 | 472 | raise bt2.TryAgain |
f6a5e476 | 473 | self._at += 1 |
fa4c33e3 | 474 | return msg |
f6a5e476 | 475 | |
61d96b89 | 476 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
b20382e2 | 477 | def __init__(self, params, obj): |
f6a5e476 PP |
478 | self._add_output_port('out') |
479 | ||
480 | class MySink(bt2._UserSinkComponent): | |
b20382e2 | 481 | def __init__(self, params, obj): |
871a292a | 482 | self._input_port = self._add_input_port('in') |
f6a5e476 PP |
483 | self._at = 0 |
484 | ||
819d0ae7 | 485 | def _user_consume(comp_self): |
871a292a | 486 | msg = next(comp_self._msg_iter) |
f6a5e476 | 487 | if comp_self._at == 0: |
c946c9de | 488 | self.assertIsInstance(msg, bt2._StreamBeginningMessage) |
f6a5e476 | 489 | elif comp_self._at == 1: |
c946c9de | 490 | self.assertIsInstance(msg, bt2._PacketBeginningMessage) |
871a292a | 491 | elif comp_self._at == 2: |
c946c9de | 492 | self.assertIsInstance(msg, bt2._EventMessage) |
871a292a SM |
493 | elif comp_self._at == 3: |
494 | nonlocal raised_in_sink | |
495 | raised_in_sink = True | |
f6a5e476 PP |
496 | raise RuntimeError('error!') |
497 | ||
498 | comp_self._at += 1 | |
499 | ||
819d0ae7 | 500 | def _user_graph_is_configured(self): |
692f1a01 PP |
501 | self._msg_iter = self._create_input_port_message_iterator( |
502 | self._input_port | |
503 | ) | |
f6a5e476 PP |
504 | |
505 | src = self._graph.add_component(MySource, 'src') | |
506 | sink = self._graph.add_component(MySink, 'sink') | |
61d96b89 FD |
507 | conn = self._graph.connect_ports( |
508 | src.output_ports['out'], sink.input_ports['in'] | |
509 | ) | |
f6a5e476 | 510 | |
614743a5 | 511 | with self.assertRaises(bt2._Error): |
f6a5e476 PP |
512 | self._graph.run() |
513 | ||
871a292a | 514 | def test_listeners(self): |
fa4c33e3 | 515 | class MyIter(bt2._UserMessageIterator): |
871a292a SM |
516 | def __next__(self): |
517 | raise bt2.Stop | |
9ef22b36 | 518 | |
61d96b89 | 519 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
b20382e2 | 520 | def __init__(self, params, obj): |
9ef22b36 | 521 | self._add_output_port('out') |
871a292a | 522 | self._add_output_port('zero') |
9ef22b36 PP |
523 | |
524 | class MySink(bt2._UserSinkComponent): | |
b20382e2 | 525 | def __init__(self, params, obj): |
9ef22b36 | 526 | self._add_input_port('in') |
9ef22b36 | 527 | |
819d0ae7 | 528 | def _user_consume(self): |
871a292a | 529 | raise bt2.Stop |
9ef22b36 | 530 | |
819d0ae7 | 531 | def _user_port_connected(self, port, other_port): |
871a292a | 532 | self._add_input_port('taste') |
9ef22b36 | 533 | |
871a292a SM |
534 | def port_added_listener(component, port): |
535 | nonlocal calls | |
536 | calls.append((port_added_listener, component, port)) | |
9ef22b36 | 537 | |
61d96b89 FD |
538 | def ports_connected_listener( |
539 | upstream_component, upstream_port, downstream_component, downstream_port | |
540 | ): | |
871a292a | 541 | nonlocal calls |
61d96b89 FD |
542 | calls.append( |
543 | ( | |
544 | ports_connected_listener, | |
545 | upstream_component, | |
546 | upstream_port, | |
547 | downstream_component, | |
548 | downstream_port, | |
549 | ) | |
550 | ) | |
871a292a SM |
551 | |
552 | calls = [] | |
553 | self._graph.add_port_added_listener(port_added_listener) | |
554 | self._graph.add_ports_connected_listener(ports_connected_listener) | |
9ef22b36 PP |
555 | src = self._graph.add_component(MySource, 'src') |
556 | sink = self._graph.add_component(MySink, 'sink') | |
61d96b89 | 557 | self._graph.connect_ports(src.output_ports['out'], sink.input_ports['in']) |
9ef22b36 | 558 | |
871a292a SM |
559 | self.assertEqual(len(calls), 5) |
560 | ||
561 | self.assertIs(calls[0][0], port_added_listener) | |
562 | self.assertEqual(calls[0][1].name, 'src') | |
563 | self.assertEqual(calls[0][2].name, 'out') | |
564 | ||
565 | self.assertIs(calls[1][0], port_added_listener) | |
566 | self.assertEqual(calls[1][1].name, 'src') | |
567 | self.assertEqual(calls[1][2].name, 'zero') | |
568 | ||
569 | self.assertIs(calls[2][0], port_added_listener) | |
570 | self.assertEqual(calls[2][1].name, 'sink') | |
571 | self.assertEqual(calls[2][2].name, 'in') | |
572 | ||
573 | self.assertIs(calls[3][0], port_added_listener) | |
574 | self.assertEqual(calls[3][1].name, 'sink') | |
575 | self.assertEqual(calls[3][2].name, 'taste') | |
576 | ||
577 | self.assertIs(calls[4][0], ports_connected_listener) | |
578 | self.assertEqual(calls[4][1].name, 'src') | |
579 | self.assertEqual(calls[4][2].name, 'out') | |
580 | self.assertEqual(calls[4][3].name, 'sink') | |
581 | self.assertEqual(calls[4][4].name, 'in') | |
582 | ||
583 | def test_invalid_listeners(self): | |
fa4c33e3 | 584 | class MyIter(bt2._UserMessageIterator): |
f6a5e476 PP |
585 | def __next__(self): |
586 | raise bt2.Stop | |
587 | ||
61d96b89 | 588 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
b20382e2 | 589 | def __init__(self, params, obj): |
f6a5e476 PP |
590 | self._add_output_port('out') |
591 | self._add_output_port('zero') | |
592 | ||
f6a5e476 | 593 | class MySink(bt2._UserSinkComponent): |
b20382e2 | 594 | def __init__(self, params, obj): |
f6a5e476 PP |
595 | self._add_input_port('in') |
596 | ||
819d0ae7 | 597 | def _user_consume(self): |
f6a5e476 PP |
598 | raise bt2.Stop |
599 | ||
819d0ae7 | 600 | def _user_port_connected(self, port, other_port): |
f6a5e476 PP |
601 | self._add_input_port('taste') |
602 | ||
871a292a SM |
603 | with self.assertRaises(TypeError): |
604 | self._graph.add_port_added_listener(1234) | |
605 | with self.assertRaises(TypeError): | |
606 | self._graph.add_ports_connected_listener(1234) | |
f6a5e476 | 607 | |
871a292a SM |
608 | def test_raise_in_component_init(self): |
609 | class MySink(bt2._UserSinkComponent): | |
b20382e2 | 610 | def __init__(self, params, obj): |
871a292a | 611 | raise ValueError('oops!') |
f6a5e476 | 612 | |
819d0ae7 | 613 | def _user_consume(self): |
871a292a SM |
614 | raise bt2.Stop |
615 | ||
616 | graph = bt2.Graph() | |
617 | ||
614743a5 | 618 | with self.assertRaises(bt2._Error): |
871a292a SM |
619 | graph.add_component(MySink, 'comp') |
620 | ||
621 | def test_raise_in_port_added_listener(self): | |
622 | class MySink(bt2._UserSinkComponent): | |
b20382e2 | 623 | def __init__(self, params, obj): |
871a292a SM |
624 | self._add_input_port('in') |
625 | ||
819d0ae7 | 626 | def _user_consume(self): |
871a292a SM |
627 | raise bt2.Stop |
628 | ||
629 | def port_added_listener(component, port): | |
630 | raise ValueError('oh noes!') | |
631 | ||
632 | graph = bt2.Graph() | |
633 | graph.add_port_added_listener(port_added_listener) | |
634 | ||
614743a5 | 635 | with self.assertRaises(bt2._Error): |
871a292a SM |
636 | graph.add_component(MySink, 'comp') |
637 | ||
638 | def test_raise_in_ports_connected_listener(self): | |
639 | class MyIter(bt2._UserMessageIterator): | |
640 | def __next__(self): | |
641 | raise bt2.Stop | |
642 | ||
61d96b89 | 643 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
b20382e2 | 644 | def __init__(self, params, obj): |
871a292a SM |
645 | self._add_output_port('out') |
646 | ||
647 | class MySink(bt2._UserSinkComponent): | |
b20382e2 | 648 | def __init__(self, params, obj): |
871a292a SM |
649 | self._add_input_port('in') |
650 | ||
819d0ae7 | 651 | def _user_consume(self): |
871a292a | 652 | raise bt2.Stop |
f6a5e476 | 653 | |
61d96b89 FD |
654 | def ports_connected_listener( |
655 | upstream_component, upstream_port, downstream_component, downstream_port | |
656 | ): | |
871a292a | 657 | raise ValueError('oh noes!') |
f6a5e476 | 658 | |
871a292a SM |
659 | graph = bt2.Graph() |
660 | graph.add_ports_connected_listener(ports_connected_listener) | |
661 | up = graph.add_component(MySource, 'down') | |
662 | down = graph.add_component(MySink, 'up') | |
f6a5e476 | 663 | |
614743a5 | 664 | with self.assertRaises(bt2._Error): |
871a292a | 665 | graph.connect_ports(up.output_ports['out'], down.input_ports['in']) |