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

[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 243563f5bd7e16eda466f67f7f19136a32a04ddc..9c52801b8500bc3112a679021440e44c09275d67 100644 (file)
--- a/tests/bindings/python/bt2/test_stream_class.py
+++ b/tests/bindings/python/bt2/test_stream_class.py
@@ -24,6 +24,10 @@ class StreamClassTestCase(unittest.TestCase):
         self.assertTrue(sc.assigns_automatic_stream_id)
         self.assertFalse(sc.packets_have_default_beginning_clock_snapshot)
         self.assertFalse(sc.packets_have_default_end_clock_snapshot)
+        self.assertFalse(sc.supports_discarded_events)
+        self.assertFalse(sc.discarded_events_have_default_clock_snapshots)
+        self.assertFalse(sc.supports_discarded_packets)
+        self.assertFalse(sc.discarded_packets_have_default_clock_snapshots)
 
     def test_create_name(self):
         sc = self._tc.create_stream_class(name='bozo')
@@ -131,6 +135,62 @@ class StreamClassTestCase(unittest.TestCase):
         with self.assertRaises(TypeError):
             sc = self._tc.create_stream_class(packets_have_default_end_clock_snapshot="something")
 
+    def test_supports_discarded_events_without_cs(self):
+        sc = self._tc.create_stream_class(default_clock_class=self._cc,
+                                          supports_discarded_events=True)
+        self.assertTrue(sc.supports_discarded_events)
+        self.assertFalse(sc.discarded_events_have_default_clock_snapshots)
+
+    def test_supports_discarded_events_with_cs(self):
+        sc = self._tc.create_stream_class(default_clock_class=self._cc,
+                                          supports_discarded_events=True,
+                                          discarded_events_have_default_clock_snapshots=True)
+        self.assertTrue(sc.supports_discarded_events)
+        self.assertTrue(sc.discarded_events_have_default_clock_snapshots)
+
+    def test_supports_discarded_events_raises_type_error(self):
+        with self.assertRaises(TypeError):
+            sc = self._tc.create_stream_class(default_clock_class=self._cc,
+                                              supports_discarded_events=23)
+
+    def test_discarded_events_have_default_cs_raises_type_error(self):
+        with self.assertRaises(TypeError):
+            sc = self._tc.create_stream_class(default_clock_class=self._cc,
+                                              discarded_events_have_default_clock_snapshots=23)
+
+    def test_does_not_support_discarded_events_raises_with_cs(self):
+        with self.assertRaises(ValueError):
+            sc = self._tc.create_stream_class(default_clock_class=self._cc,
+                                              discarded_events_have_default_clock_snapshots=True)
+
+    def test_supports_discarded_packets_without_cs(self):
+        sc = self._tc.create_stream_class(default_clock_class=self._cc,
+                                          supports_discarded_packets=True)
+        self.assertTrue(sc.supports_discarded_packets)
+        self.assertFalse(sc.discarded_packets_have_default_clock_snapshots)
+
+    def test_supports_discarded_packets_with_cs(self):
+        sc = self._tc.create_stream_class(default_clock_class=self._cc,
+                                          supports_discarded_packets=True,
+                                          discarded_packets_have_default_clock_snapshots=True)
+        self.assertTrue(sc.supports_discarded_packets)
+        self.assertTrue(sc.discarded_packets_have_default_clock_snapshots)
+
+    def test_supports_discarded_packets_raises_type_error(self):
+        with self.assertRaises(TypeError):
+            sc = self._tc.create_stream_class(default_clock_class=self._cc,
+                                              supports_discarded_packets=23)
+
+    def test_discarded_packets_have_default_cs_raises_type_error(self):
+        with self.assertRaises(TypeError):
+            sc = self._tc.create_stream_class(default_clock_class=self._cc,
+                                              discarded_packets_have_default_clock_snapshots=23)
+
+    def test_does_not_support_discarded_packets_raises_with_cs(self):
+        with self.assertRaises(ValueError):
+            sc = self._tc.create_stream_class(default_clock_class=self._cc,
+                                              discarded_packets_have_default_clock_snapshots=True)
+
     def test_trace_class(self):
         sc = self._tc.create_stream_class()
         self.assertEqual(sc.trace_class.addr, self._tc.addr)