Add custom sampling context testcases

Author: sentrivana

tests/tracing/test_sampling.py

@@ -211,6 +211,23 @@ def dummy_traces_sampler(sampling_context):
             pass
 
 
+def test_passes_custom_attributes_in_sampling_context(sentry_init):
+    def dummy_traces_sampler(sampling_context):
+        assert sampling_context["dog.name"] == "Lily"
+        assert sampling_context["dog.activity"] == "fetch"
+        return 1.0
+
+    sentry_init(traces_sample_rate=1.0, traces_sampler=dummy_traces_sampler)
+
+    with sentry_sdk.continue_trace(
+        {"sentry-trace": "12312012123120121231201212312012-1121201211212012-1"}
+    ):
+        with sentry_sdk.start_span(
+            name="dogpark", attributes={"dog.name": "Lily", "dog.activity": "fetch"}
+        ):
+            pass
+
+
 def test_passes_attributes_from_start_span_to_traces_sampler(
     sentry_init, DictionaryContaining  # noqa: N803
 ):
@@ -342,3 +359,20 @@ def dummy_profiles_sampler(sampling_context):
     with sentry_sdk.continue_trace({"sentry-trace": sentry_trace}):
         with sentry_sdk.start_span(name="dogpark", op="op"):
             pass
+
+
+def test_passes_custom_attributes_in_profiles_sampling_context(sentry_init):
+    def profiles_sampler(sampling_context):
+        assert sampling_context["dog.name"] == "Lily"
+        assert sampling_context["dog.activity"] == "fetch"
+        return 1.0
+
+    sentry_init(traces_sample_rate=1.0, profiles_sampler=profiles_sampler)
+
+    with sentry_sdk.continue_trace(
+        {"sentry-trace": "12312012123120121231201212312012-1121201211212012-1"}
+    ):
+        with sentry_sdk.start_span(
+            name="dogpark", attributes={"dog.name": "Lily", "dog.activity": "fetch"}
+        ):
+            pass