+ def _user_consume(self):
+ # We should not reach this.
+ assert False
+
+ g = bt2.Graph()
+ src = g.add_component(MySource, 'src')
+ flt = g.add_component(MyFilter, 'flt')
+ snk = g.add_component(MySink, 'snk')
+ g.connect_ports(src.output_ports['out'], flt.input_ports['in'])
+ g.connect_ports(flt.output_ports['out'], snk.input_ports['in'])
+
+ with self.assertRaisesRegex(bt2._Error, 'ValueError: woops'):
+ g.run()
+
+
+def _setup_seek_test(
+ sink_cls,
+ user_seek_beginning=None,
+ user_can_seek_beginning=None,
+ user_seek_ns_from_origin=None,
+ user_can_seek_ns_from_origin=None,
+ can_seek_forward=False,
+):
+ class MySourceIter(bt2._UserMessageIterator):
+ def __init__(self, config, port):
+ tc, sc, ec = port.user_data
+ trace = tc()
+ stream = trace.create_stream(sc)
+ packet = stream.create_packet()
+
+ self._msgs = [
+ self._create_stream_beginning_message(stream),
+ self._create_packet_beginning_message(packet),
+ self._create_event_message(ec, packet),
+ self._create_event_message(ec, packet),
+ self._create_packet_end_message(packet),
+ self._create_stream_end_message(stream),
+ ]
+ self._at = 0
+ config.can_seek_forward = can_seek_forward
+
+ def __next__(self):
+ if self._at < len(self._msgs):
+ msg = self._msgs[self._at]
+ self._at += 1
+ return msg
+ else:
+ raise StopIteration
+
+ if user_seek_beginning is not None:
+ MySourceIter._user_seek_beginning = user_seek_beginning
+
+ if user_can_seek_beginning is not None:
+ MySourceIter._user_can_seek_beginning = user_can_seek_beginning
+
+ if user_seek_ns_from_origin is not None:
+ MySourceIter._user_seek_ns_from_origin = user_seek_ns_from_origin
+
+ if user_can_seek_ns_from_origin is not None:
+ MySourceIter._user_can_seek_ns_from_origin = user_can_seek_ns_from_origin
+
+ class MySource(bt2._UserSourceComponent, message_iterator_class=MySourceIter):
+ def __init__(self, config, params, obj):
+ tc = self._create_trace_class()
+ sc = tc.create_stream_class(supports_packets=True)
+ ec = sc.create_event_class()
+
+ self._add_output_port('out', (tc, sc, ec))
+
+ class MyFilterIter(bt2._UserMessageIterator):
+ def __init__(self, config, port):
+ self._upstream_iter = self._create_message_iterator(
+ self._component._input_ports['in']
+ )
+ config.can_seek_forward = self._upstream_iter.can_seek_forward
+
+ def __next__(self):
+ return next(self._upstream_iter)
+
+ def _user_can_seek_beginning(self):
+ return self._upstream_iter.can_seek_beginning()
+
+ def _user_seek_beginning(self):
+ self._upstream_iter.seek_beginning()
+
+ def _user_can_seek_ns_from_origin(self, ns_from_origin):
+ return self._upstream_iter.can_seek_ns_from_origin(ns_from_origin)
+
+ def _user_seek_ns_from_origin(self, ns_from_origin):
+ self._upstream_iter.seek_ns_from_origin(ns_from_origin)
+
+ class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyFilterIter):
+ def __init__(self, config, params, obj):
+ self._add_input_port('in')
+ self._add_output_port('out')
+
+ return _create_graph(MySource, sink_cls, flt_comp_cls=MyFilter)
+
+
+class UserMessageIteratorSeekBeginningTestCase(unittest.TestCase):
+ def test_can_seek_beginning_without_seek_beginning(self):
+ with self.assertRaisesRegex(
+ bt2._IncompleteUserClass,
+ "cannot create component class 'MySource': message iterator class implements _user_can_seek_beginning but not _user_seek_beginning",
+ ):
+ _setup_seek_test(SimpleSink, user_can_seek_beginning=lambda: None)