Add span data to the transactions trace context

sentry_sdk/tracing.py

@@ -1027,6 +1027,15 @@ def to_json(self):
 
         return rv
 
+    def get_trace_context(self):
+        # type: () -> Any
+        trace_context = super().get_trace_context()
+
+        if self._data:
+            trace_context["data"] = self._data
+
+        return trace_context
+
     def get_baggage(self):
         # type: () -> Baggage
         """Returns the :py:class:`~sentry_sdk.tracing_utils.Baggage`

tests/tracing/test_misc.py

@@ -11,6 +11,7 @@
 from sentry_sdk.tracing import Span, Transaction
 from sentry_sdk.tracing_utils import should_propagate_trace
 from sentry_sdk.utils import Dsn
+from tests.conftest import ApproxDict
 
 
 def test_span_trimming(sentry_init, capture_events):
@@ -60,6 +61,35 @@ def test_transaction_naming(sentry_init, capture_events):
     assert events[2]["transaction"] == "a"
 
 
+def test_transaction_data(sentry_init, capture_events):
+    sentry_init(traces_sample_rate=1.0)
+    events = capture_events()
+
+    with start_transaction(name="test-transaction"):
+        span_or_tx = sentry_sdk.get_current_span()
+        span_or_tx.set_data("foo", "bar")
+        with start_span(op="test-span") as span:
+            span.set_data("spanfoo", "spanbar")
+
+    (transaction,) = events
+    span = transaction["spans"][0]
+
+    transaction_data = transaction["contexts"]["trace"]["data"]
+    span_data = span["data"]
+
+    assert transaction_data == ApproxDict(
+        {
+            "foo": "bar",
+        }
+    )
+
+    assert span_data == ApproxDict(
+        {
+            "spanfoo": "spanbar",
+        }
+    )
+
+
 def test_start_transaction(sentry_init):
     sentry_init(traces_sample_rate=1.0)