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()