.

Author: sentrivana

tests/tracing/test_sampling.py

@@ -6,6 +6,7 @@
 
 import sentry_sdk
 from sentry_sdk import start_span, capture_exception
+from sentry_sdk.tracing import BAGGAGE_HEADER_NAME
 from sentry_sdk.tracing_utils import Baggage
 from sentry_sdk.utils import logger
 
@@ -59,9 +60,11 @@ def test_uses_traces_sample_rate_correctly(
 ):
     sentry_init(traces_sample_rate=traces_sample_rate)
 
-    baggage = Baggage(sentry_items={"sample_rand": "0.500000"})
-    root_span = start_span(name="dogpark", baggage=baggage)
-    assert root_span.sampled is expected_decision
+    with sentry_sdk.continue_trace(
+        {BAGGAGE_HEADER_NAME: "sentry-sample_rand=0.500000"}
+    ):
+        with start_span(name="dogpark") as root_span:
+            assert root_span.sampled is expected_decision
 
 
 @pytest.mark.parametrize(