def test_logs_warning_if_using_function_as_transport(
dummy_request,
dummy_response,
):
dummy_request.registry.settings = {
'zipkin.is_tracing': lambda _: False,
'zipkin.transport_handler': lambda x: None,
'request_context': 'rctxstorage',
}
handler = mock.Mock(return_value=dummy_response)
with pytest.deprecated_call():
tween.zipkin_tween(handler, None)(dummy_request)
def test_zipkin_tween_post_handler_hook(
mock_span,
dummy_request,
dummy_response,
is_tracing,
set_callback,
called,
):
"""
We should invoke the post processor callback regardless of trace id
or sampling
"""
mock_post_handler_hook = mock.Mock()
dummy_request.registry.settings = {
'zipkin.is_tracing': lambda _: is_tracing,
'zipkin.transport_handler': MockTransport(),
}
if set_callback:
dummy_request.registry.settings['zipkin.post_handler_hook'] = \
mock_post_handler_hook
handler = mock.Mock()
handler.return_value = dummy_response
assert tween.zipkin_tween(handler, None)(dummy_request) == dummy_response
assert handler.call_count == 1
assert mock_span.call_count == 1
assert mock_post_handler_hook.call_count == called
if set_callback:
mock_post_handler_hook.assert_called_once_with(
dummy_request,
dummy_response,
mock_span.return_value.__enter__.return_value,
)
def test_zipkin_tween_not_sampled(mock_span, dummy_request, dummy_response):
"""
If the request is not sampled, we shouldn't use the
py_zipkin context manager
"""
dummy_request.registry.settings = {
'zipkin.is_tracing': lambda _: False,
'zipkin.transport_handler': lambda _: None,
}
handler = mock.Mock()
handler.return_value = dummy_response
assert tween.zipkin_tween(handler, None)(dummy_request) == dummy_response
assert handler.call_count == 1
assert mock_span.call_count == 0
def test_zipkin_tween_sampling(
mock_span,
dummy_request,
dummy_response,
is_tracing,
):
"""
We should enter py_zipkin context manager and
generate a trace id regardless of whether we are sampling
"""
dummy_request.registry.settings = {
'zipkin.is_tracing': lambda _: is_tracing,
'zipkin.transport_handler': lambda _: None,
}
handler = mock.Mock()
handler.return_value = dummy_response
assert tween.zipkin_tween(handler, None)(dummy_request) == dummy_response
assert handler.call_count == 1
assert mock_span.call_count == 1
def test_zipkin_tween_sampling(
mock_span,
dummy_request,
dummy_response,
is_tracing,
):
"""
We should enter py_zipkin context manager and
generate a trace id regardless of whether we are sampling
"""
dummy_request.registry.settings = {
'zipkin.is_tracing': lambda _: is_tracing,
'zipkin.transport_handler': MockTransport(),
}
handler = mock.Mock()
handler.return_value = dummy_response
assert tween.zipkin_tween(handler, None)(dummy_request) == dummy_response
assert handler.call_count == 1
assert mock_span.call_count == 1
def test_zipkin_tween_context_stack(
mock_thread_local_stack,
dummy_request,
dummy_response,
):
dummy_request.registry.settings = {
'zipkin.is_tracing': lambda _: False,
'zipkin.transport_handler': lambda _: None,
'zipkin.request_context': 'rctxstorage.zipkin_context',
}
context_stack = mock.Mock(spec=py_zipkin.stack.Stack)
dummy_request.rctxstorage = DummyRequestContext(
zipkin_context=context_stack, )
handler = mock.Mock(return_value=dummy_response)
assert tween.zipkin_tween(handler, None)(dummy_request) == dummy_response
assert mock_thread_local_stack.call_count == 0
assert context_stack.push.call_count == 1
assert context_stack.pop.call_count == 1
def test_zipkin_tween_context_stack(
mock_thread_local_stack,
dummy_request,
dummy_response,
):
dummy_request.registry.settings = {
'zipkin.is_tracing': lambda _: False,
'zipkin.transport_handler': lambda _: None,
'zipkin.request_context': 'rctxstorage.zipkin_context',
}
context_stack = mock.Mock(spec=py_zipkin.stack.Stack)
dummy_request.rctxstorage = DummyRequestContext(
zipkin_context=context_stack,
)
handler = mock.Mock(return_value=dummy_response)
assert tween.zipkin_tween(handler, None)(dummy_request) == dummy_response
assert mock_thread_local_stack.call_count == 0
assert context_stack.push.call_count == 1
assert context_stack.pop.call_count == 1
def test_zipkin_tween_context_stack_none(
mock_thread_local_stack,
dummy_request,
dummy_response,
):
mock_thread_local_stack_instance = mock.Mock()
mock_thread_local_stack.return_value = mock_thread_local_stack_instance
dummy_request.registry.settings = {
'zipkin.is_tracing': lambda _: False,
'zipkin.transport_handler': lambda _: None,
'request_context': 'rctxstorage',
}
# explicitly delete attribute, will raise AttributeError on access
delattr(dummy_request, dummy_request.registry.settings['request_context'])
handler = mock.Mock(return_value=dummy_response)
assert tween.zipkin_tween(handler, None)(dummy_request) == dummy_response
assert mock_thread_local_stack.call_count == 1
assert mock_thread_local_stack_instance.push.call_count == 1
assert mock_thread_local_stack_instance.pop.call_count == 1
def test_zipkin_tween_context_stack(
dummy_request,
dummy_response,
):
old_context_stack = get_default_tracer()._context_stack
dummy_request.registry.settings = {
'zipkin.is_tracing': lambda _: False,
'zipkin.transport_handler': MockTransport(),
'zipkin.request_context': 'rctxstorage.zipkin_context',
}
context_stack = mock.Mock(spec=Stack)
dummy_request.rctxstorage = DummyRequestContext(
zipkin_context=context_stack, )
handler = mock.Mock(return_value=dummy_response)
response = tween.zipkin_tween(handler, None)(dummy_request)
get_default_tracer()._context_stack = old_context_stack
assert response == dummy_response
assert context_stack.push.call_count == 1
assert context_stack.pop.call_count == 1
def test_zipkin_tween_context_stack_none(
mock_thread_local_stack,
dummy_request,
dummy_response,
):
mock_thread_local_stack_instance = mock.Mock()
mock_thread_local_stack.return_value = mock_thread_local_stack_instance
dummy_request.registry.settings = {
'zipkin.is_tracing': lambda _: False,
'zipkin.transport_handler': lambda _: None,
'request_context': 'rctxstorage',
}
# explicitly delete attribute, will raise AttributeError on access
delattr(dummy_request, dummy_request.registry.settings['request_context'])
handler = mock.Mock(return_value=dummy_response)
assert tween.zipkin_tween(handler, None)(dummy_request) == dummy_response
assert mock_thread_local_stack.call_count == 1
assert mock_thread_local_stack_instance.push.call_count == 1
assert mock_thread_local_stack_instance.pop.call_count == 1