@@ -277,17 +277,26 @@ def test_distributed_tracing_waf(self):
277277 expected_headers = {
278278 "x-datadog-sampling-priority" : "2" ,
279279 "x-datadog-origin" : "rum" ,
280- "x-datadog-tags" : "_dd.p.dm=-5, _dd.p.ts=02" ,
280+ "x-datadog-tags" : [ "_dd.p.dm=-5" , " _dd.p.ts=02"] ,
281281 }
282282
283283 for header_name , expected_value in expected_headers .items ():
284284 self .assertIn (header_name , forwarded_headers ,
285285 f"Missing header: { header_name } " )
286- self .assertEqual (
287- forwarded_headers [header_name ],
288- expected_value ,
289- f"Header { header_name } mismatch: expected { expected_value } , got { forwarded_headers [header_name ]} " ,
290- )
286+ gotten_value = forwarded_headers [header_name ]
287+ if isinstance (expected_value , list ):
288+ for ev in expected_value :
289+ self .assertIn (
290+ ev ,
291+ gotten_value ,
292+ f"Header { header_name } missing expected value: { ev } " ,
293+ )
294+ else :
295+ self .assertEqual (
296+ gotten_value ,
297+ expected_value ,
298+ f"Header { header_name } mismatch: expected { expected_value } , got { forwarded_headers [header_name ]} " ,
299+ )
291300
292301 # Check traceparent format (trace ID should match, span ID should be different)
293302 self .assertIn ("traceparent" , forwarded_headers ,
@@ -327,12 +336,17 @@ def test_distributed_tracing_waf(self):
327336 tracestate = forwarded_headers ["tracestate" ]
328337 # The tracestate should contain the new span ID (same as in traceparent)
329338 expected_span_id = traceparent_parts [2 ]
330- expected_tracestate = f"dd=s:2;p:{ expected_span_id } ;o:rum;t.dm:-5;t.ts:02"
331- self .assertEqual (
332- tracestate ,
333- expected_tracestate ,
334- f"tracestate mismatch: expected { expected_tracestate } , got { tracestate } " ,
335- )
339+ expected_tracestate = [
340+ "dd=s:2" , f"p:{ expected_span_id } " , "o:rum" , "t.dm:-5" ,
341+ "t.ts:02"
342+ ]
343+ split_tracestate = tracestate .split (";" )
344+ for expected_part in expected_tracestate :
345+ self .assertIn (
346+ expected_part ,
347+ split_tracestate ,
348+ f"tracestate missing expected part: { expected_part } . Got { tracestate } " ,
349+ )
336350
337351 span_count = 0
338352
0 commit comments