def test_extract_bad_values(trace_id, parent_span_id, sampling_priority,
dd_origin):
headers = dict()
wsgi_headers = dict()
if trace_id is not NOT_SET:
headers[HTTP_HEADER_TRACE_ID] = trace_id
wsgi_headers[get_wsgi_header(HTTP_HEADER_TRACE_ID)] = trace_id
if parent_span_id is not NOT_SET:
headers[HTTP_HEADER_PARENT_ID] = parent_span_id
wsgi_headers[get_wsgi_header(HTTP_HEADER_PARENT_ID)] = parent_span_id
if sampling_priority is not NOT_SET:
headers[HTTP_HEADER_SAMPLING_PRIORITY] = sampling_priority
wsgi_headers[get_wsgi_header(
HTTP_HEADER_SAMPLING_PRIORITY)] = sampling_priority
if dd_origin is not NOT_SET:
headers[HTTP_HEADER_ORIGIN] = dd_origin
wsgi_headers[get_wsgi_header(HTTP_HEADER_ORIGIN)] = dd_origin
# x-datadog-*headers
context = HTTPPropagator.extract(headers)
assert context.trace_id is None
assert context.span_id is None
assert context.sampling_priority is None
assert context.dd_origin is None
assert context._meta == {}
# HTTP_X_DATADOG_* headers
context = HTTPPropagator.extract(wsgi_headers)
assert context.trace_id is None
assert context.span_id is None
assert context.sampling_priority is None
assert context.dd_origin is None
assert context._meta == {}
def generate_headers(self):
headers = json.loads(self.headers)
if self.wsgi_style:
headers = {
utils.get_wsgi_header(header): value
for header, value in headers.items()
}
for i in range(self.extra_headers):
header = "x-test-header-{}".format(i)
if self.wsgi_style:
header = utils.get_wsgi_header(header)
headers[header] = str(i)
return headers
def test_get_wsgi_header(self):
assert get_wsgi_header(
"x-datadog-trace-id") == "HTTP_X_DATADOG_TRACE_ID"