Add test for public current_span() accessors

Author: TheJokr

foundations/tests/tracing.rs

@@ -58,3 +58,37 @@ fn test_passive_sampler(mut ctx: TestTelemetryContext) {
 
     assert_eq!(ctx.traces(Default::default()).len(), 0);
 }
+
+#[with_test_telemetry(test)]
+fn test_span_accessors(mut ctx: TestTelemetryContext) {
+    // Initially, no trace (and no span) is present
+    assert!(!tracing::span_is_sampled());
+    assert_eq!(tracing::trace_id(), None);
+
+    {
+        // Start a new trace
+        let _root_scope = tracing::start_trace("my first span", Default::default());
+        assert!(tracing::span_is_sampled());
+
+        let trace_id = tracing::trace_id().expect("root scope should set trace ID");
+        let trace_state =
+            tracing::state_for_trace_stitching().expect("root scope should set stitching state");
+        assert_eq!(trace_state.trace_id().to_string(), trace_id);
+
+        // Enter a child span, which should have the same trace ID
+        let _child_scope = tracing::span("my child span");
+        assert!(tracing::span_is_sampled());
+        assert_eq!(tracing::trace_id(), Some(trace_id));
+    }
+
+    // Change to the passive sampler, which won't sample new root traces
+    ctx.set_tracing_settings(TracingSettings {
+        sampling_strategy: SamplingStrategy::Passive,
+        ..Default::default()
+    });
+    let _ctx_scope = ctx.scope();
+
+    let _unsampled_scope = tracing::start_trace("my unsampled span", Default::default());
+    assert!(!tracing::span_is_sampled());
+    assert_eq!(tracing::trace_id(), None);
+}