@@ -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