def test_create_headers_for_new_span_returns_header_if_active_request(
gen_mock, get_mock):
get_mock.return_value = mock.Mock(
trace_id='1', span_id='3', is_sampled=True)
gen_mock.return_value = '2'
expected = {
'X-B3-TraceId': '1',
'X-B3-SpanId': '2',
'X-B3-ParentSpanId': '3',
'X-B3-Flags': '0',
'X-B3-Sampled': '1',
}
assert expected == zipkin.create_headers_for_new_span()
def sample_child_span(request):
return zipkin.create_headers_for_new_span()
def test_create_headers_for_new_span_raises_if_no_active_request(get_mock):
get_mock.return_value = None
assert {} == zipkin.create_headers_for_new_span()
def sample_child_span(request):
return zipkin.create_headers_for_new_span()
