@@ -2,7 +2,6 @@ package datadog.trace.core
22
33import datadog.communication.ddagent.DDAgentFeaturesDiscovery
44import datadog.communication.ddagent.SharedCommunicationObjects
5- import datadog.communication.monitor.Monitoring
65import datadog.trace.api.Config
76import datadog.trace.api.DDTraceId
87import datadog.trace.api.sampling.PrioritySampling
@@ -16,7 +15,7 @@ class LongRunningTracesTrackerTest extends DDSpecification {
1615 Config config = Mock (Config )
1716 int maxTrackedTraces = 10
1817 def sharedCommunicationObjects = Mock (SharedCommunicationObjects )
19- DDAgentFeaturesDiscovery features = new DDAgentFeaturesDiscovery ( null , Monitoring . DISABLED , null , false , false )
18+ DDAgentFeaturesDiscovery features = Mock ( DDAgentFeaturesDiscovery )
2019 LongRunningTracesTracker tracker
2120 def tracer = Mock (CoreTracer )
2221 def traceConfig = Stub (CoreTracer.ConfigSnapshot )
@@ -26,7 +25,7 @@ class LongRunningTracesTrackerTest extends DDSpecification {
2625
2726 def setup () {
2827 timeSource. set(0L )
29- features. supportsLongRunning = true
28+ features. supportsLongRunning() >> true
3029 tracer. captureTraceConfig() >> traceConfig
3130 tracer. getTimeWithNanoTicks(_) >> { Long x -> x }
3231 traceConfig. getServiceMapping() >> [:]
@@ -119,12 +118,12 @@ class LongRunningTracesTrackerTest extends DDSpecification {
119118 given :
120119 def trace = newTraceToTrack()
121120 tracker. add(trace)
122- features. supportsLongRunning = false
123121
124122 when :
125123 tracker. flushAndCompact(tracker. flushPeriodMilli - 1000 )
126124
127125 then :
126+ 1 * features. supportsLongRunning() >> false
128127 tracker. traceArray. size() == 0
129128 }
130129
0 commit comments