From a457acb2a57567e0dc2f9e4ccc6ece3f964c068a Mon Sep 17 00:00:00 2001
From: Stuart McCulloch <stuart.mcculloch@datadoghq.com>
Date: Fri, 20 Dec 2024 01:59:18 +0000
Subject: [PATCH] Cleanup test side-effects

---
 .../trace/core/propagation/XRayHttpInjectorTest.groovy     | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/dd-trace-core/src/test/groovy/datadog/trace/core/propagation/XRayHttpInjectorTest.groovy b/dd-trace-core/src/test/groovy/datadog/trace/core/propagation/XRayHttpInjectorTest.groovy
index 16226ef6362..e6229c93e1c 100644
--- a/dd-trace-core/src/test/groovy/datadog/trace/core/propagation/XRayHttpInjectorTest.groovy
+++ b/dd-trace-core/src/test/groovy/datadog/trace/core/propagation/XRayHttpInjectorTest.groovy
@@ -6,6 +6,7 @@ import datadog.trace.api.DDTraceId
 import datadog.trace.api.DynamicConfig
 import datadog.trace.api.time.TimeSource
 import datadog.trace.bootstrap.instrumentation.api.AgentTracer.NoopPathwayContext
+import datadog.trace.core.datastreams.DataStreamsMonitoring
 
 import static datadog.trace.api.sampling.PrioritySampling.*
 import static datadog.trace.api.sampling.SamplingMechanism.*
@@ -136,7 +137,11 @@ class XRayHttpInjectorTest extends DDCoreSpecification {
     setup:
     def writer = new ListWriter()
     def timeSource = Mock(TimeSource)
-    def tracer = tracerBuilder().writer(writer).timeSource(timeSource).build()
+    def tracer = tracerBuilder()
+      .dataStreamsMonitoring(Mock(DataStreamsMonitoring))
+      .writer(writer)
+      .timeSource(timeSource)
+      .build()
     final DDSpanContext mockedContext =
       new DDSpanContext(
       DDTraceId.from("1"),