def test_create_with_count(self):
def create_stream_class(tc, cc):
return tc.create_stream_class(supports_discarded_events=True)
def msg_iter_next(msg_iter, stream):
return msg_iter._create_discarded_events_message(stream, count=242)
msg = utils.run_in_message_iterator_next(create_stream_class, msg_iter_next)
self.assertIs(type(msg), bt2._DiscardedEventsMessage)
self.assertEqual(msg.count, 242)
def test_create(self):
def create_stream_class(tc, cc):
return tc.create_stream_class(
supports_packets=True, supports_discarded_packets=True
)
def msg_iter_next(msg_iter, stream):
return msg_iter._create_discarded_packets_message(stream)
msg = utils.run_in_message_iterator_next(create_stream_class, msg_iter_next)
self.assertIs(type(msg), bt2._DiscardedPacketsMessage)
self.assertIs(msg.count, None)
def test_create_unsupported_raises(self):
def create_stream_class(tc, cc):
return tc.create_stream_class(supports_packets=True,)
def msg_iter_next(msg_iter, stream):
with self.assertRaisesRegex(
ValueError, 'stream class does not support discarded packets'
):
msg_iter._create_discarded_packets_message(stream)
return 123
res = utils.run_in_message_iterator_next(create_stream_class, msg_iter_next)
self.assertEqual(res, 123)
def test_create_with_count_zero_raises(self):
def create_stream_class(tc, cc):
return tc.create_stream_class(supports_discarded_events=True)
def msg_iter_next(msg_iter, stream):
with self.assertRaisesRegex(
ValueError, 'discarded event count is 0',
):
msg_iter._create_discarded_events_message(stream, count=0)
return 123
res = utils.run_in_message_iterator_next(create_stream_class, msg_iter_next)
self.assertEqual(res, 123)
def test_create_unsupported_clock_snapshots_raises(self):
def create_stream_class(tc, cc):
return tc.create_stream_class(supports_discarded_events=True)
def msg_iter_next(msg_iter, stream):
with self.assertRaisesRegex(
ValueError,
'discarded events have no default clock snapshots for this stream class',
):
msg_iter._create_discarded_events_message(
stream, beg_clock_snapshot=10, end_clock_snapshot=20
)
return 123
res = utils.run_in_message_iterator_next(create_stream_class, msg_iter_next)
self.assertEqual(res, 123)
def test_create_with_clock_snapshots(self):
def create_stream_class(tc, cc):
return tc.create_stream_class(
default_clock_class=cc,
supports_discarded_events=True,
discarded_events_have_default_clock_snapshots=True,
)
def msg_iter_next(msg_iter, stream):
return msg_iter._create_discarded_events_message(
stream, beg_clock_snapshot=10, end_clock_snapshot=20
)
msg = utils.run_in_message_iterator_next(create_stream_class, msg_iter_next)
self.assertIs(type(msg), bt2._DiscardedEventsMessage)
self.assertEqual(msg.beginning_default_clock_snapshot, 10)
self.assertEqual(msg.end_default_clock_snapshot, 20)
def test_create_missing_clock_snapshots_raises(self):
def create_stream_class(tc, cc):
return tc.create_stream_class(
default_clock_class=cc,
supports_packets=True,
supports_discarded_packets=True,
discarded_packets_have_default_clock_snapshots=True,
)
def msg_iter_next(msg_iter, stream):
with self.assertRaisesRegex(
ValueError,
'discarded packets have default clock snapshots for this stream class',
):
msg_iter._create_discarded_packets_message(stream)
return 123
res = utils.run_in_message_iterator_next(create_stream_class, msg_iter_next)
self.assertEqual(res, 123)
def test_create_clock_snapshots_end_gt_begin_raises(self):
def create_stream_class(tc, cc):
return tc.create_stream_class(
default_clock_class=cc,
supports_discarded_events=True,
discarded_events_have_default_clock_snapshots=True,
)
def msg_iter_next(msg_iter, stream):
with self.assertRaisesRegex(
ValueError,
r'beginning default clock snapshot value \(20\) is greater than end default clock snapshot value \(10\)',
):
msg_iter._create_discarded_events_message(
stream, beg_clock_snapshot=20, end_clock_snapshot=10
)
return 123
res = utils.run_in_message_iterator_next(create_stream_class, msg_iter_next)
self.assertEqual(res, 123)
