Skip to content

Commit 5ddffa8

Browse files
committed
Fix test after upgrade of dd-trace-cpp
1 parent c8f152b commit 5ddffa8

File tree

1 file changed

+26
-12
lines changed

1 file changed

+26
-12
lines changed

test/cases/configuration/test_apm_tracing_enabled.py

Lines changed: 26 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)