def test_create_http_headers_new_span_id(gen_mock): tracer = MockTracer() tracer.push_zipkin_attrs( ZipkinAttrs( trace_id="17133d482ba4f605", span_id="37133d482ba4f605", is_sampled=True, parent_span_id="27133d482ba4f605", flags=None, )) gen_mock.return_value = "47133d482ba4f605" # if new_span_id = True we generate a new span id assert request_helpers.create_http_headers(tracer=tracer, new_span_id=True) == { "X-B3-TraceId": "17133d482ba4f605", "X-B3-SpanId": "47133d482ba4f605", "X-B3-ParentSpanId": "37133d482ba4f605", "X-B3-Flags": "0", "X-B3-Sampled": "1", } # by default we keep the same span id as the current span assert request_helpers.create_http_headers(tracer=tracer) == { "X-B3-TraceId": "17133d482ba4f605", "X-B3-SpanId": "37133d482ba4f605", "X-B3-ParentSpanId": "27133d482ba4f605", "X-B3-Flags": "0", "X-B3-Sampled": "1", }
def test_create_headers_for_new_span_custom_tracer(gen_mock): tracer = MockTracer() tracer.push_zipkin_attrs( mock.Mock( trace_id='27133d482ba4f605', span_id='37133d482ba4f605', is_sampled=True, )) gen_mock.return_value = '17133d482ba4f605' expected = { 'X-B3-TraceId': '27133d482ba4f605', 'X-B3-SpanId': '17133d482ba4f605', 'X-B3-ParentSpanId': '37133d482ba4f605', 'X-B3-Flags': '0', 'X-B3-Sampled': '1', } assert expected == zipkin.create_http_headers_for_new_span(tracer=tracer, )
def test_create_http_headers_custom_tracer(): tracer = MockTracer() tracer.push_zipkin_attrs( ZipkinAttrs( trace_id="17133d482ba4f605", span_id="37133d482ba4f605", is_sampled=True, parent_span_id="27133d482ba4f605", flags=None, )) expected = { "X-B3-TraceId": "17133d482ba4f605", "X-B3-SpanId": "37133d482ba4f605", "X-B3-ParentSpanId": "27133d482ba4f605", "X-B3-Flags": "0", "X-B3-Sampled": "1", } assert expected == request_helpers.create_http_headers(tracer=tracer)