@@ -6,6 +6,7 @@ import datadog.trace.api.DDTraceId
66import datadog.trace.api.DynamicConfig
77import datadog.trace.api.time.TimeSource
88import datadog.trace.bootstrap.instrumentation.api.AgentTracer.NoopPathwayContext
9+ import datadog.trace.core.datastreams.DataStreamsMonitoring
910
1011import static datadog.trace.api.sampling.PrioritySampling.*
1112import static datadog.trace.api.sampling.SamplingMechanism.*
@@ -25,7 +26,11 @@ class XRayHttpInjectorTest extends DDCoreSpecification {
2526 setup :
2627 def writer = new ListWriter ()
2728 def timeSource = Mock (TimeSource )
28- def tracer = tracerBuilder(). writer(writer). timeSource(timeSource). build()
29+ def tracer = tracerBuilder()
30+ .dataStreamsMonitoring(Mock (DataStreamsMonitoring ))
31+ .writer(writer)
32+ .timeSource(timeSource)
33+ .build()
2934 final DDSpanContext mockedContext =
3035 new DDSpanContext (
3136 DDTraceId . from(" $traceId " ),
@@ -76,7 +81,11 @@ class XRayHttpInjectorTest extends DDCoreSpecification {
7681 setup :
7782 def writer = new ListWriter ()
7883 def timeSource = Mock (TimeSource )
79- def tracer = tracerBuilder(). writer(writer). timeSource(timeSource). build()
84+ def tracer = tracerBuilder()
85+ .dataStreamsMonitoring(Mock (DataStreamsMonitoring ))
86+ .writer(writer)
87+ .timeSource(timeSource)
88+ .build()
8089 def headers = [
8190 ' X-Amzn-Trace-Id' : " Root=1-00000000-00000000${ traceId.padLeft(16, '0')} ;Parent=${ spanId.padLeft(16, '0')} "
8291 ]
@@ -136,7 +145,11 @@ class XRayHttpInjectorTest extends DDCoreSpecification {
136145 setup :
137146 def writer = new ListWriter ()
138147 def timeSource = Mock (TimeSource )
139- def tracer = tracerBuilder(). writer(writer). timeSource(timeSource). build()
148+ def tracer = tracerBuilder()
149+ .dataStreamsMonitoring(Mock (DataStreamsMonitoring ))
150+ .writer(writer)
151+ .timeSource(timeSource)
152+ .build()
140153 final DDSpanContext mockedContext =
141154 new DDSpanContext (
142155 DDTraceId . from(" 1" ),
0 commit comments