def test_create_sampled_zipkin_attr_creates_ZipkinAttr_object(mock, request): mock.return_value = 'bla' request.zipkin_trace_id = '12' request.headers = { 'X-B3-TraceId': '12', 'X-B3-SpanId': '23', 'X-B3-ParentSpanId': '34', 'X-B3-Flags': '45', } zipkin_attr = request_helper.ZipkinAttrs( trace_id='12', span_id='23', parent_span_id='34', flags='45', is_sampled='bla') assert zipkin_attr == request_helper.create_zipkin_attr(request)
def unsampled_zipkin_attr(zipkin_attributes): return request_helper.ZipkinAttrs(is_sampled=False, **zipkin_attributes)