Fix: _EventConst.__getitem__(): check if event has a packet

[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 4288fd91f1471e54bab75be2f3325bc1e5df838b..e8effef34535ea1cb55953db817a0a2803a1d56b 100644 (file)
--- a/tests/bindings/python/bt2/test_event.py
+++ b/tests/bindings/python/bt2/test_event.py
@@ -95,7 +95,7 @@ class EventTestCase(unittest.TestCase):
                     cc = tc.create_structure_field_class()
                     cc += [
                         ('cpu_id', tc.create_signed_integer_field_class(8)),
-                        ('stuff', tc.create_real_field_class()),
+                        ('stuff', tc.create_double_precision_real_field_class()),
                     ]
 
                 # packet context (stream-class-defined)
@@ -105,7 +105,10 @@ class EventTestCase(unittest.TestCase):
                     pc = tc.create_structure_field_class()
                     pc += [
                         ('something', tc.create_unsigned_integer_field_class(8)),
-                        ('something_else', tc.create_real_field_class()),
+                        (
+                            'something_else',
+                            tc.create_double_precision_real_field_class(),
+                        ),
                     ]
 
                 stream_class = tc.create_stream_class(
@@ -328,6 +331,20 @@ class EventTestCase(unittest.TestCase):
         with self.assertRaises(KeyError):
             ev['yes']
 
+    def test_const_getitem_no_packet(self):
+        def event_fields_config(event):
+            event.payload_field['giraffe'] = 1
+            event.payload_field['gnu'] = 23
+            event.payload_field['mosquito'] = 42
+
+        msg = self._create_test_const_event_message(
+            event_fields_config=event_fields_config, with_ep=True,
+        )
+        ev = msg.event
+
+        with self.assertRaises(KeyError):
+            ev['yes']
+
     def test_getitem(self):
         msg = utils.get_event_message()
         ev = msg.event