Skip to content

Commit 8663672

Browse files
committed
port longrunning tests
1 parent b6bd66b commit 8663672

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

dd-trace-core/src/test/groovy/datadog/trace/core/LongRunningTracesTrackerTest.groovy

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@ package datadog.trace.core
22

33
import datadog.communication.ddagent.DDAgentFeaturesDiscovery
44
import datadog.communication.ddagent.SharedCommunicationObjects
5-
import datadog.communication.monitor.Monitoring
65
import datadog.trace.api.Config
76
import datadog.trace.api.DDTraceId
87
import 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

Comments
 (0)