Migrating compound extraction span link tests to weblog

tests/test_distributed.py

@@ -3,6 +3,7 @@
 # Copyright 2022 Datadog, Inc.
 
 from utils import weblog, interfaces, scenarios, features
+import json
 
 
 @scenarios.trace_propagation_style_w3c
@@ -25,3 +26,76 @@ def test_main(self):
         assert "x-datadog-sampling-priority" not in data["request_headers"]
         assert "x-datadog-tags" not in data["request_headers"]
         assert "x-datadog-trace-id" not in data["request_headers"]
+
+
+@scenarios.trace_propagation_style_w3c_datadog_b3
+@features.w3c_headers_injection_and_extraction
+class Test_Span_Links_From_Conflicting_Contexts:
+    """Verify headers containing conflicting trace context information are added as span links"""
+
+    def setup_span_links_from_conflicting_contexts(self):
+        extract_headers = {
+            "traceparent": "00-11111111111111110000000000000002-000000003ade68b1-01",
+            "tracestate": "dd=s:2;p:000000000000000a,foo=1",
+            "x-datadog-parent-id": "10",
+            "x-datadog-trace-id": "2",
+            "x-datadog-tags": "_dd.p.tid=2222222222222222",
+            "x-datadog-sampling-priority": "2",
+            "x-b3-traceid": "11111111111111110000000000000003",
+            "x-b3-spanid": "a2fb4a1d1a96d312",
+            "x-b3-sampled": "0",
+        }
+
+        self.req = weblog.get("/make_distant_call", params={"url": "http://weblog:7777"}, headers=extract_headers)
+
+    def test_span_links_from_conflicting_contexts(self):
+        trace = [span for _, _, span in interfaces.library.get_spans(self.req, full_trace=True)]
+        print(trace)
+        for span in trace:
+            links = retrieve_span_links(span)
+            if links != None:
+                assert len(links) == 2
+                print(links)
+                link1 = links[0]
+                assert link1["trace_id"] == 2
+                assert link1["span_id"] == 10
+                assert link1["attributes"] == {"reason": "terminated_context", "context_headers": "datadog"}
+                assert link1["trace_id_high"] == 2459565876494606882
+
+                link2 = links[1]
+                assert link2["trace_id"] == 3
+                assert link2["span_id"] == 11744061942159299346
+                assert link2["attributes"] == {"reason": "terminated_context", "context_headers": "b3multi"}
+                assert link2["trace_id_high"] == 1229782938247303441
+                return
+
+        # Return false if we get no span links
+        assert False
+
+
+def retrieve_span_links(span):
+    if span.get("span_links") is not None:
+        return span["span_links"]
+
+    if span["meta"].get("_dd.span_links") is not None:
+        # Convert span_links tags into msgpack v0.4 format
+        json_links = json.loads(span["meta"].get("_dd.span_links"))
+        links = []
+        for json_link in json_links:
+            link = {}
+            link["trace_id"] = int(json_link["trace_id"][-16:], base=16)
+            link["span_id"] = int(json_link["span_id"], base=16)
+            if len(json_link["trace_id"]) > 16:
+                link["trace_id_high"] = int(json_link["trace_id"][:16], base=16)
+            if "attributes" in json_link:
+                link["attributes"] = json_link.get("attributes")
+            if "tracestate" in json_link:
+                link["tracestate"] = json_link.get("tracestate")
+            elif "trace_state" in json_link:
+                link["tracestate"] = json_link.get("trace_state")
+            if "flags" in json_link:
+                link["flags"] = json_link.get("flags") | 1 << 31
+            else:
+                link["flags"] = 0
+            links.append(link)
+        return links

utils/_context/_scenarios/__init__.py

@@ -105,6 +105,15 @@ def all_endtoend_scenarios(test_object):
         doc="Test W3C trace style",
     )
 
+    trace_propagation_style_w3c_datadog_b3 = EndToEndScenario(
+        "TRACE_PROPAGATION_STYLE_W3C_DATADOG_B3",
+        weblog_env={
+            "DD_TRACE_PROPAGATION_STYLE_INJECT": "tracecontext,datadog,b3multi",
+            "DD_TRACE_PROPAGATION_STYLE_EXTRACT": "tracecontext,datadog,b3multi",
+        },
+        doc="Test compound propagation trace style",
+    )
+
     # Telemetry scenarios
     telemetry_dependency_loaded_test_for_dependency_collection_disabled = EndToEndScenario(
         "TELEMETRY_DEPENDENCY_LOADED_TEST_FOR_DEPENDENCY_COLLECTION_DISABLED",