bt2: Add remaining trace-ir `*Const` classes and adapt tests

[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 473efca7923aa1468195cfa4ffdcc3983a0cb2d6..b510fc2b055af5aae7be998489622a8550257f92 100644 (file)
--- a/tests/bindings/python/bt2/test_stream_class.py
+++ b/tests/bindings/python/bt2/test_stream_class.py
@@ -18,6 +18,11 @@
 
 import unittest
 from utils import run_in_component_init
+from bt2 import stream_class as bt2_stream_class
+from bt2 import trace_class as bt2_trace_class
+from bt2 import clock_class as bt2_clock_class
+from bt2 import event_class as bt2_event_class
+from bt2 import field_class as bt2_field_class
 
 
 class StreamClassTestCase(unittest.TestCase):
@@ -33,6 +38,7 @@ class StreamClassTestCase(unittest.TestCase):
     def test_create_default(self):
         sc = self._tc.create_stream_class()
 
+        self.assertIs(type(sc), bt2_stream_class._StreamClass)
         self.assertIsNone(sc.name)
         self.assertIsNone(sc.packet_context_field_class)
         self.assertIsNone(sc.event_common_context_field_class)
@@ -46,6 +52,7 @@ class StreamClassTestCase(unittest.TestCase):
         self.assertFalse(sc.discarded_events_have_default_clock_snapshots)
         self.assertFalse(sc.supports_discarded_packets)
         self.assertFalse(sc.discarded_packets_have_default_clock_snapshots)
+        self.assertEqual(len(sc.user_attributes), 0)
 
     def test_create_name(self):
         sc = self._tc.create_stream_class(name='bozo')
@@ -61,6 +68,9 @@ class StreamClassTestCase(unittest.TestCase):
             packet_context_field_class=fc, supports_packets=True
         )
         self.assertEqual(sc.packet_context_field_class, fc)
+        self.assertIs(
+            type(sc.packet_context_field_class), bt2_field_class._StructureFieldClass
+        )
 
     def test_create_invalid_packet_context_field_class(self):
         with self.assertRaises(TypeError):
@@ -76,6 +86,10 @@ class StreamClassTestCase(unittest.TestCase):
         fc = self._tc.create_structure_field_class()
         sc = self._tc.create_stream_class(event_common_context_field_class=fc)
         self.assertEqual(sc.event_common_context_field_class, fc)
+        self.assertIs(
+            type(sc.event_common_context_field_class),
+            bt2_field_class._StructureFieldClass,
+        )
 
     def test_create_invalid_event_common_context_field_class(self):
         with self.assertRaises(TypeError):
@@ -84,11 +98,24 @@ class StreamClassTestCase(unittest.TestCase):
     def test_create_default_clock_class(self):
         sc = self._tc.create_stream_class(default_clock_class=self._cc)
         self.assertEqual(sc.default_clock_class.addr, self._cc.addr)
+        self.assertIs(type(sc.default_clock_class), bt2_clock_class._ClockClass)
 
     def test_create_invalid_default_clock_class(self):
         with self.assertRaises(TypeError):
             self._tc.create_stream_class(default_clock_class=12)
 
+    def test_create_user_attributes(self):
+        sc = self._tc.create_stream_class(user_attributes={'salut': 23})
+        self.assertEqual(sc.user_attributes, {'salut': 23})
+
+    def test_create_invalid_user_attributes(self):
+        with self.assertRaises(TypeError):
+            self._tc.create_stream_class(user_attributes=object())
+
+    def test_create_invalid_user_attributes_value_type(self):
+        with self.assertRaises(TypeError):
+            self._tc.create_stream_class(user_attributes=23)
+
     def test_automatic_stream_ids(self):
         sc = self._tc.create_stream_class(assigns_automatic_stream_id=True)
         self.assertTrue(sc.assigns_automatic_stream_id)
@@ -175,33 +202,33 @@ class StreamClassTestCase(unittest.TestCase):
 
     def test_supports_packets_raises_type_error(self):
         with self.assertRaises(TypeError):
-            sc = self._tc.create_stream_class(
+            self._tc.create_stream_class(
                 default_clock_class=self._cc, supports_packets=23
             )
 
     def test_packets_have_begin_default_cs_raises_type_error(self):
         with self.assertRaises(TypeError):
-            sc = self._tc.create_stream_class(
+            self._tc.create_stream_class(
                 default_clock_class=self._cc,
                 packets_have_beginning_default_clock_snapshot=23,
             )
 
     def test_packets_have_end_default_cs_raises_type_error(self):
         with self.assertRaises(TypeError):
-            sc = self._tc.create_stream_class(
+            self._tc.create_stream_class(
                 default_clock_class=self._cc, packets_have_end_default_clock_snapshot=23
             )
 
     def test_does_not_support_packets_raises_with_begin_cs(self):
         with self.assertRaises(ValueError):
-            sc = self._tc.create_stream_class(
+            self._tc.create_stream_class(
                 default_clock_class=self._cc,
                 packets_have_beginning_default_clock_snapshot=True,
             )
 
     def test_does_not_support_packets_raises_with_end_cs(self):
         with self.assertRaises(ValueError):
-            sc = self._tc.create_stream_class(
+            self._tc.create_stream_class(
                 default_clock_class=self._cc,
                 packets_have_end_default_clock_snapshot=True,
             )
@@ -224,20 +251,20 @@ class StreamClassTestCase(unittest.TestCase):
 
     def test_supports_discarded_events_raises_type_error(self):
         with self.assertRaises(TypeError):
-            sc = self._tc.create_stream_class(
+            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(
+            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(
+            self._tc.create_stream_class(
                 default_clock_class=self._cc,
                 discarded_events_have_default_clock_snapshots=True,
             )
@@ -263,13 +290,13 @@ class StreamClassTestCase(unittest.TestCase):
 
     def test_supports_discarded_packets_raises_without_packet_support(self):
         with self.assertRaises(ValueError):
-            sc = self._tc.create_stream_class(
+            self._tc.create_stream_class(
                 default_clock_class=self._cc, supports_discarded_packets=True
             )
 
     def test_supports_discarded_packets_raises_type_error(self):
         with self.assertRaises(TypeError):
-            sc = self._tc.create_stream_class(
+            self._tc.create_stream_class(
                 default_clock_class=self._cc,
                 supports_discarded_packets=23,
                 supports_packets=True,
@@ -277,7 +304,7 @@ class StreamClassTestCase(unittest.TestCase):
 
     def test_discarded_packets_have_default_cs_raises_type_error(self):
         with self.assertRaises(TypeError):
-            sc = self._tc.create_stream_class(
+            self._tc.create_stream_class(
                 default_clock_class=self._cc,
                 discarded_packets_have_default_clock_snapshots=23,
                 supports_packets=True,
@@ -285,7 +312,7 @@ class StreamClassTestCase(unittest.TestCase):
 
     def test_does_not_support_discarded_packets_raises_with_cs(self):
         with self.assertRaises(ValueError):
-            sc = self._tc.create_stream_class(
+            self._tc.create_stream_class(
                 default_clock_class=self._cc,
                 discarded_packets_have_default_clock_snapshots=True,
                 supports_packets=True,
@@ -294,6 +321,7 @@ class StreamClassTestCase(unittest.TestCase):
     def test_trace_class(self):
         sc = self._tc.create_stream_class()
         self.assertEqual(sc.trace_class.addr, self._tc.addr)
+        self.assertIs(type(sc.trace_class), bt2_trace_class._TraceClass)
 
     def _create_stream_class_with_event_classes(self):
         sc = self._tc.create_stream_class(assigns_automatic_event_class_id=False)
@@ -305,7 +333,9 @@ class StreamClassTestCase(unittest.TestCase):
         sc, ec1, ec2 = self._create_stream_class_with_event_classes()
 
         self.assertEqual(sc[23].addr, ec1.addr)
+        self.assertEqual(type(sc[23]), bt2_event_class._EventClass)
         self.assertEqual(sc[17].addr, ec2.addr)
+        self.assertEqual(type(sc[17]), bt2_event_class._EventClass)
 
     def test_getitem_wrong_key_type(self):
         sc, _, _ = self._create_stream_class_with_event_classes()