bt2: stream.py: add `trace` property to _Stream

Signed-off-by: Francis Deslauriers <francis.deslauriers@efficios.com>
Change-Id: I434ebf2dd9c5a29bb8ef15ef7058479eba5ff9f6
Reviewed-on: https://review.lttng.org/c/babeltrace/+/1807
Tested-by: jenkins <jenkins@lttng.org>
Reviewed-by: Philippe Proulx <eeppeliteloop@gmail.com>

diff --git a/src/bindings/python/bt2/bt2/stream.py b/src/bindings/python/bt2/bt2/stream.py
index 3a8821795caeb9c0aa5eeaf3cf75e89e9bc63044..308924e937384e003a074869c22a4abece677f52 100644 (file)
--- a/src/bindings/python/bt2/bt2/stream.py
+++ b/src/bindings/python/bt2/bt2/stream.py
@@ -24,6 +24,7 @@ from bt2 import native_bt, utils
 from bt2 import object as bt2_object
 from bt2 import packet as bt2_packet
 from bt2 import event as bt2_event
+from bt2 import trace as bt2_trace
 from bt2 import stream_class as bt2_stream_class
 import bt2
 
@@ -67,3 +68,9 @@ class _Stream(bt2_object._SharedObject):
             raise bt2._MemoryError('cannot create packet object')
 
         return bt2_packet._Packet._create_from_ptr(packet_ptr)
+
+    @property
+    def trace(self):
+        trace_ptr = native_bt.stream_borrow_trace(self._ptr)
+        assert trace_ptr is not None
+        return bt2_trace._Trace._create_from_ptr_and_get_ref(trace_ptr)

diff --git a/tests/bindings/python/bt2/test_stream.py b/tests/bindings/python/bt2/test_stream.py
index a49cc2be2e0d8316a765489a94fa3b99f6bfda92..7f80f8dfb67a97afe5df82ab227c0494746960b0 100644 (file)
--- a/tests/bindings/python/bt2/test_stream.py
+++ b/tests/bindings/python/bt2/test_stream.py
@@ -45,6 +45,10 @@ class StreamTestCase(unittest.TestCase):
         stream = self._tr.create_stream(self._sc)
         self.assertEqual(stream.cls, self._sc)
 
+    def test_trace(self):
+        stream = self._tr.create_stream(self._sc)
+        self.assertEqual(stream.trace.addr, self._tr.addr)
+
     def test_invalid_id(self):
         sc = self._tc.create_stream_class(assigns_automatic_stream_id=False)