@@ -6,6 +6,7 @@ import datadog.trace.api.DDTraceId
6
6
import datadog.trace.api.DynamicConfig
7
7
import datadog.trace.api.time.TimeSource
8
8
import datadog.trace.bootstrap.instrumentation.api.AgentTracer.NoopPathwayContext
9
+ import datadog.trace.core.datastreams.DataStreamsMonitoring
9
10
10
11
import static datadog.trace.api.sampling.PrioritySampling.*
11
12
import static datadog.trace.api.sampling.SamplingMechanism.*
@@ -25,7 +26,11 @@ class XRayHttpInjectorTest extends DDCoreSpecification {
25
26
setup :
26
27
def writer = new ListWriter ()
27
28
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()
29
34
final DDSpanContext mockedContext =
30
35
new DDSpanContext (
31
36
DDTraceId . from(" $traceId " ),
@@ -76,7 +81,11 @@ class XRayHttpInjectorTest extends DDCoreSpecification {
76
81
setup :
77
82
def writer = new ListWriter ()
78
83
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()
80
89
def headers = [
81
90
' X-Amzn-Trace-Id' : " Root=1-00000000-00000000${ traceId.padLeft(16, '0')} ;Parent=${ spanId.padLeft(16, '0')} "
82
91
]
@@ -136,7 +145,11 @@ class XRayHttpInjectorTest extends DDCoreSpecification {
136
145
setup :
137
146
def writer = new ListWriter ()
138
147
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()
140
153
final DDSpanContext mockedContext =
141
154
new DDSpanContext (
142
155
DDTraceId . from(" 1" ),
0 commit comments