Skip to content

Commit 8e8639a

Browse files
committed
Cleanup test side-effects
1 parent c9f2377 commit 8e8639a

File tree

1 file changed

+16
-3
lines changed

1 file changed

+16
-3
lines changed

dd-trace-core/src/test/groovy/datadog/trace/core/propagation/XRayHttpInjectorTest.groovy

Lines changed: 16 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ import datadog.trace.api.DDTraceId
66
import datadog.trace.api.DynamicConfig
77
import datadog.trace.api.time.TimeSource
88
import datadog.trace.bootstrap.instrumentation.api.AgentTracer.NoopPathwayContext
9+
import datadog.trace.core.datastreams.DataStreamsMonitoring
910

1011
import static datadog.trace.api.sampling.PrioritySampling.*
1112
import 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

Comments
 (0)