babeltrace2-plugin-ctf(7): "theirs" -> "its" (single CTF trace)

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

diff --git a/tests/bindings/python/bt2/test_event_class.py b/tests/bindings/python/bt2/test_event_class.py
index 131907107ec2de8ccf785c77a5a56ca37a1e8448..33b8dc788967dc33fd76e5d10fc9153ac62d7757 100644 (file)
--- a/tests/bindings/python/bt2/test_event_class.py
+++ b/tests/bindings/python/bt2/test_event_class.py
@@ -34,7 +34,7 @@ def _create_const_event_class(tc, stream_class):
     )
 
     class MyIter(bt2._UserMessageIterator):
-        def __init__(self, self_port_output):
+        def __init__(self, config, self_port_output):
 
             trace = tc()
             stream = trace.create_stream(stream_class)
@@ -50,7 +50,7 @@ def _create_const_event_class(tc, stream_class):
             return self._msgs.pop(0)
 
     class MySrc(bt2._UserSourceComponent, message_iterator_class=MyIter):
-        def __init__(self, params, obj):
+        def __init__(self, config, params, obj):
             self._add_output_port('out', params)
 
     graph = bt2.Graph()
@@ -99,6 +99,8 @@ class EventClassTestCase(unittest.TestCase):
         with self.assertRaises(TypeError):
             sc.create_event_class(id='lel')
 
+        self.assertEqual(len(sc), 0)
+
     def test_create_specific_context_field_class(self):
         fc = self._tc.create_structure_field_class()
         ec = self._stream_class.create_event_class(specific_context_field_class=fc)
@@ -118,6 +120,8 @@ class EventClassTestCase(unittest.TestCase):
         with self.assertRaises(TypeError):
             self._stream_class.create_event_class(specific_context_field_class='lel')
 
+        self.assertEqual(len(self._stream_class), 0)
+
     def test_create_payload_field_class(self):
         fc = self._tc.create_structure_field_class()
         ec = self._stream_class.create_event_class(payload_field_class=fc)
@@ -137,6 +141,8 @@ class EventClassTestCase(unittest.TestCase):
         with self.assertRaises(TypeError):
             self._stream_class.create_event_class(payload_field_class='lel')
 
+        self.assertEqual(len(self._stream_class), 0)
+
     def test_create_name(self):
         ec = self._stream_class.create_event_class(name='viande à chien')
         self.assertEqual(ec.name, 'viande à chien')
@@ -145,6 +151,8 @@ class EventClassTestCase(unittest.TestCase):
         with self.assertRaises(TypeError):
             self._stream_class.create_event_class(name=2)
 
+        self.assertEqual(len(self._stream_class), 0)
+
     def test_emf_uri(self):
         ec = self._stream_class.create_event_class(emf_uri='salut')
         self.assertEqual(ec.emf_uri, 'salut')
@@ -153,6 +161,8 @@ class EventClassTestCase(unittest.TestCase):
         with self.assertRaises(TypeError):
             self._stream_class.create_event_class(emf_uri=23)
 
+        self.assertEqual(len(self._stream_class), 0)
+
     def test_create_log_level(self):
         ec = self._stream_class.create_event_class(
             log_level=bt2.EventClassLogLevel.EMERGENCY
@@ -163,6 +173,8 @@ class EventClassTestCase(unittest.TestCase):
         with self.assertRaises(ValueError):
             self._stream_class.create_event_class(log_level='zoom')
 
+        self.assertEqual(len(self._stream_class), 0)
+
     def test_create_user_attributes(self):
         ec = self._stream_class.create_event_class(user_attributes={'salut': 23})
         self.assertEqual(ec.user_attributes, {'salut': 23})
@@ -176,10 +188,14 @@ class EventClassTestCase(unittest.TestCase):
         with self.assertRaises(TypeError):
             self._stream_class.create_event_class(user_attributes=object())
 
+        self.assertEqual(len(self._stream_class), 0)
+
     def test_create_invalid_user_attributes_value_type(self):
         with self.assertRaises(TypeError):
             self._stream_class.create_event_class(user_attributes=23)
 
+        self.assertEqual(len(self._stream_class), 0)
+
     def test_stream_class(self):
         ec = self._stream_class.create_event_class()
         self.assertEqual(ec.stream_class.addr, self._stream_class.addr)
@@ -188,3 +204,7 @@ class EventClassTestCase(unittest.TestCase):
     def test_const_stream_class(self):
         ec_const = _create_const_event_class(self._tc, self._stream_class)
         self.assertIs(type(ec_const.stream_class), bt2_stream_class._StreamClassConst)
+
+
+if __name__ == '__main__':
+    unittest.main()