test: Add streaming tests to test_http_headers (getsentry#6785)

sentrivana · web-flow · commit 2e0949760f60 · 2026-07-20T11:24:47.000+02:00
### Description Add span streaming variants to tests in `test_http_headers` #### Issues Part of getsentry#5395 #### Reminders - Please add tests to validate your changes, and lint your code using `uv run ruff`. - Add GH Issue ID _&_ Linear ID (if applicable) - PR title should use [conventional commit](https://develop.sentry.dev/engineering-practices/commit-messages/#type) style (`feat:`, `fix:`, `ref:`, `meta:`) - For external contributors: [CONTRIBUTING.md](https://github.com/getsentry/sentry-python/blob/master/CONTRIBUTING.md), [Sentry SDK development docs](https://develop.sentry.dev/sdk/), [Discord community](https://discord.gg/Ww9hbqr)
diff --git a/tests/tracing/test_http_headers.py b/tests/tracing/test_http_headers.py
@@ -2,6 +2,8 @@
 
 import pytest
 
+import sentry_sdk
+from sentry_sdk.traces import StreamedSpan
 from sentry_sdk.tracing import Transaction
 from sentry_sdk.tracing_utils import extract_sentrytrace_data
 
@@ -26,6 +28,35 @@ def test_to_traceparent(sampled):
         assert parts[2] == "1" if sampled is True else "0"  # sampled
 
 
+@pytest.mark.parametrize("traces_sample_rate", [1.0, 0.0, None])
+def test_to_traceparent_span_streaming(sentry_init, traces_sample_rate):
+    sentry_init(
+        traces_sample_rate=traces_sample_rate,
+        _experiments={
+            "trace_lifecycle": "stream",
+        },
+    )
+
+    with sentry_sdk.traces.start_span(name="/interactions/other-dogs/new-dog") as span:
+        traceparent = sentry_sdk.get_traceparent()
+
+        parts = traceparent.split("-")
+        if traces_sample_rate is None:
+            propagation_context = (
+                sentry_sdk.get_current_scope().get_active_propagation_context()
+            )
+            assert parts[0] == propagation_context.trace_id  # trace_id
+            assert parts[1] == propagation_context.span_id  # parent_span_id
+            assert len(parts) == 2
+        else:
+            assert parts[0] == span.trace_id  # trace_id
+            assert parts[1] == span.span_id  # parent_span_id
+            if traces_sample_rate == 1.0:
+                assert parts[2] == "1"
+            else:
+                assert parts[2] == "0"
+
+
 @pytest.mark.parametrize("sampling_decision", [True, False])
 def test_sentrytrace_extraction(sampling_decision):
     sentrytrace_header = "12312012123120121231201212312012-0415201309082013-{}".format(
@@ -75,3 +106,24 @@ def test_iter_headers(monkeypatch):
     assert (
         headers["sentry-trace"] == "12312012123120121231201212312012-0415201309082013-0"
     )
+
+
+def test_iter_headers_span_streaming(sentry_init, monkeypatch):
+    sentry_init(
+        traces_sample_rate=0.0,
+        _experiments={
+            "trace_lifecycle": "stream",
+        },
+    )
+    monkeypatch.setattr(
+        StreamedSpan,
+        "_to_traceparent",
+        mock.Mock(return_value="12312012123120121231201212312012-0415201309082013-0"),
+    )
+
+    with sentry_sdk.traces.start_span(name="/interactions/other-dogs/new-dog") as span:
+        headers = dict(span._iter_headers())
+        assert (
+            headers["sentry-trace"]
+            == "12312012123120121231201212312012-0415201309082013-0"
+        )
