Fix: bt2: add precond. check, for stream class supporting discarded msgs with clock...

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

diff --git a/tests/bindings/python/bt2/test_stream_class.py b/tests/bindings/python/bt2/test_stream_class.py
index 9010dcd4c7d68e6755f26a8e7488358b6def47c2..f68be98ddede8556dc687548c166c2456f5f318e 100644 (file)
--- a/tests/bindings/python/bt2/test_stream_class.py
+++ b/tests/bindings/python/bt2/test_stream_class.py
@@ -344,6 +344,20 @@ class StreamClassTestCase(unittest.TestCase):
 
         self.assertEqual(len(self._tc), 0)
 
+    def test_supports_discarded_events_with_clock_snapshots_without_default_clock_class_raises(
+        self,
+    ):
+        with self.assertRaisesRegex(
+            ValueError,
+            'cannot have no default clock class, but have default clock snapshots for discarded event messages',
+        ):
+            self._tc.create_stream_class(
+                supports_discarded_events=True,
+                discarded_events_have_default_clock_snapshots=True,
+            )
+
+        self.assertEqual(len(self._tc), 0)
+
     def test_supports_discarded_packets_without_cs(self):
         sc = self._tc.create_stream_class(
             default_clock_class=self._cc,
@@ -406,6 +420,21 @@ class StreamClassTestCase(unittest.TestCase):
 
         self.assertEqual(len(self._tc), 0)
 
+    def test_supports_discarded_packets_with_clock_snapshots_without_default_clock_class_raises(
+        self,
+    ):
+        with self.assertRaisesRegex(
+            ValueError,
+            'cannot have no default clock class, but have default clock snapshots for discarded packet messages',
+        ):
+            self._tc.create_stream_class(
+                supports_packets=True,
+                supports_discarded_packets=True,
+                discarded_packets_have_default_clock_snapshots=True,
+            )
+
+        self.assertEqual(len(self._tc), 0)
+
     def test_trace_class(self):
         sc = self._tc.create_stream_class()
         self.assertEqual(sc.trace_class.addr, self._tc.addr)