Skip to content

Commit 0e9f3b6

Browse files
committed
Added service name overrides to test writer
1 parent 1b32ece commit 0e9f3b6

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

dd-java-agent/testing/src/main/groovy/datadog/trace/agent/test/datastreams/RecordingDatastreamsPayloadWriter.groovy

+7
Original file line numberDiff line numberDiff line change
@@ -18,9 +18,12 @@ class RecordingDatastreamsPayloadWriter implements DatastreamsPayloadWriter {
1818
@SuppressWarnings('UnusedPrivateField')
1919
private final Set<String> backlogs = []
2020

21+
private final Set<String> serviceNameOverrides = []
22+
2123
@Override
2224
synchronized void writePayload(Collection<StatsBucket> data, String serviceNameOverride) {
2325
log.info("payload written - {}", data)
26+
serviceNameOverrides.add(serviceNameOverride)
2427
this.@payloads.addAll(data)
2528
data.each { this.@groups.addAll(it.groups) }
2629
for (StatsBucket bucket : data) {
@@ -32,6 +35,10 @@ class RecordingDatastreamsPayloadWriter implements DatastreamsPayloadWriter {
3235
}
3336
}
3437

38+
synchronized List<String> getServices() {
39+
Collections.unmodifiableList(new ArrayList<>(this.@serviceNameOverrides))
40+
}
41+
3542
synchronized List<StatsBucket> getPayloads() {
3643
Collections.unmodifiableList(new ArrayList<>(this.@payloads))
3744
}

0 commit comments

Comments
 (0)