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