lib: make discarded events/packets support and clock snapshots optional

[babeltrace.git] / tests / bindings / python / bt2 / test_event.py

diff --git a/tests/bindings/python/bt2/test_event.py b/tests/bindings/python/bt2/test_event.py
index 04978dfa99c292f6bba509b919a5269968c49ac8..d420b33f67fccc01c0e0a27018e4515f20093676 100644 (file)
--- a/tests/bindings/python/bt2/test_event.py
+++ b/tests/bindings/python/bt2/test_event.py
@@ -11,7 +11,7 @@ class EventTestCase(unittest.TestCase):
                                    with_ep=False):
 
         class MyIter(bt2._UserMessageIterator):
-            def __init__(self):
+            def __init__(self, self_output_port):
                 self._at = 0
 
             def __next__(self):
@@ -170,7 +170,8 @@ class EventTestCase(unittest.TestCase):
 
     def test_no_clock_value(self):
         msg = self._create_test_event_message(with_clockclass=False)
-        self.assertIsNone(msg.default_clock_snapshot)
+        with self.assertRaises(bt2.NonexistentClockSnapshot):
+            msg.default_clock_snapshot
 
     def test_stream(self):
         msg = self._create_test_event_message()