Skip to content

Commit 9380529

Browse files
committed
fix tests
1 parent 45ed66b commit 9380529

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

dd-trace-core/src/test/groovy/datadog/trace/core/datastreams/DataStreamsWritingTest.groovy

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ class DataStreamsWritingTest extends DDCoreSpecification {
107107

108108
def "Write bucket to mock server with process tags enabled #processTagsEnabled"() {
109109
setup:
110-
injectSysConfig(EXPERIMENTAL_PROPAGATE_PROCESS_TAGS_ENABLED, processTagsEnabled)
110+
injectSysConfig(EXPERIMENTAL_PROPAGATE_PROCESS_TAGS_ENABLED, "$processTagsEnabled")
111111
ProcessTags.reset()
112112

113113
def conditions = new PollingConditions(timeout: 2)
@@ -157,23 +157,23 @@ class DataStreamsWritingTest extends DDCoreSpecification {
157157
assert requestBodies.size() == 1
158158
}
159159

160-
validateMessage(requestBodies[0])
160+
validateMessage(requestBodies[0], processTagsEnabled)
161161

162162
cleanup:
163163
injectSysConfig(EXPERIMENTAL_PROPAGATE_PROCESS_TAGS_ENABLED, "false")
164164
ProcessTags.reset()
165165

166166
where:
167-
processTagsEnabled << ["true", "false"]
167+
processTagsEnabled << [true, false]
168168
}
169169

170-
def validateMessage(byte[] message) {
170+
def validateMessage(byte[] message, boolean processTagsEnabled) {
171171
GzipSource gzipSource = new GzipSource(Okio.source(new ByteArrayInputStream(message)))
172172

173173
BufferedSource bufferedSource = Okio.buffer(gzipSource)
174174
MessageUnpacker unpacker = MessagePack.newDefaultUnpacker(bufferedSource.inputStream())
175175

176-
assert unpacker.unpackMapHeader() == 8
176+
assert unpacker.unpackMapHeader() == 8 + (processTagsEnabled ? 1 : 0)
177177
assert unpacker.unpackString() == "Env"
178178
assert unpacker.unpackString() == "test"
179179
assert unpacker.unpackString() == "Service"

0 commit comments

Comments
 (0)