def __call__(self, request):
new_request = WSGIRequestWrapper.from_wsgi_environ(request.environ)
info = {"path": request.path, "method": request.method}
with before_request(new_request) as span:
span.log_kv(info)
with RequestContextManager(span) as ctxt:
response = request.get_response(self.application)
return self.process_response(response)
def test_caller():
environ = {
'HTTP_Custom-Caller-Header': 'Zapp',
}
from opentracing_instrumentation import config
with mock.patch.object(config.CONFIG, 'caller_name_headers',
['XXX', 'Custom-Caller-Header']):
request = WSGIRequestWrapper.from_wsgi_environ(environ)
assert request.caller_name == 'Zapp'
environ['HTTP_XXX'] = 'DOOP'
with mock.patch.object(config.CONFIG, 'caller_name_headers',
['XXX', 'Custom-Caller-Header']):
request = WSGIRequestWrapper.from_wsgi_environ(environ)
# header XXX is earlier in the list ==> higher priority
assert request.caller_name == 'DOOP'
def test_caller():
environ = {
'HTTP_Custom-Caller-Header': 'Zapp',
}
from opentracing_instrumentation import config
with mock.patch.object(config.CONFIG, 'caller_name_headers',
['XXX', 'Custom-Caller-Header']):
request = WSGIRequestWrapper.from_wsgi_environ(environ)
assert request.caller_name == 'Zapp'
environ['HTTP_XXX'] = 'DOOP'
with mock.patch.object(config.CONFIG, 'caller_name_headers',
['XXX', 'Custom-Caller-Header']):
request = WSGIRequestWrapper.from_wsgi_environ(environ)
# header XXX is earlier in the list ==> higher priority
assert request.caller_name == 'DOOP'
def test_creates_instance():
wsgi_environ = {
'SERVER_NAME': 'localhost',
'SERVER_PORT': '8888',
'REQUEST_METHOD': 'GET',
'PATH_INFO': '/Farnsworth',
'HTTP_X_FOO': 'bar',
'REMOTE_ADDR': 'localhost',
'REMOTE_PORT': '80',
'wsgi.url_scheme': 'http'
}
request = WSGIRequestWrapper.from_wsgi_environ(wsgi_environ)
assert request.server_port == '8888'
assert request.method == 'GET'
assert request.headers.get('x-foo') == 'bar'
assert request.remote_ip == 'localhost'
assert request.remote_port == '80'
assert request.full_url == 'http://localhost:8888/Farnsworth'
assert request.caller_name is None
wsgi_environ['SERVER_PORT'] = 8888 # int is also acceptable
request = WSGIRequestWrapper.from_wsgi_environ(wsgi_environ)
assert request.server_port == 8888
def test_creates_instance():
wsgi_environ = {
'SERVER_NAME': 'localhost',
'SERVER_PORT': '8888',
'REQUEST_METHOD': 'GET',
'PATH_INFO': '/Farnsworth',
'HTTP_X_FOO': 'bar',
'REMOTE_ADDR': 'localhost',
'REMOTE_PORT': '80',
'wsgi.url_scheme': 'http'
}
request = WSGIRequestWrapper.from_wsgi_environ(wsgi_environ)
assert request.server_port == '8888'
assert request.method == 'GET'
assert request.headers.get('x-foo') == 'bar'
assert request.remote_ip == 'localhost'
assert request.remote_port == '80'
assert request.full_url == 'http://localhost:8888/Farnsworth'
assert request.caller_name is None
wsgi_environ['SERVER_PORT'] = 8888 # int is also acceptable
request = WSGIRequestWrapper.from_wsgi_environ(wsgi_environ)
assert request.server_port == 8888
def test_url():
environ = {
'wsgi.url_scheme': 'http',
'HTTP_HOST': 'bender.com'
}
request = WSGIRequestWrapper.from_wsgi_environ(environ)
assert request.full_url == 'http://bender.com'
environ = {
'wsgi.url_scheme': 'http',
'SERVER_NAME': 'bender.com',
'SERVER_PORT': '80'
}
request = WSGIRequestWrapper.from_wsgi_environ(environ)
assert request.full_url == 'http://bender.com'
environ['SERVER_PORT'] = '8888'
request = WSGIRequestWrapper.from_wsgi_environ(environ)
assert request.full_url == 'http://bender.com:8888'
environ['wsgi.url_scheme'] = 'https'
request = WSGIRequestWrapper.from_wsgi_environ(environ)
assert request.full_url == 'https://bender.com:8888'
environ['SERVER_PORT'] = '443'
request = WSGIRequestWrapper.from_wsgi_environ(environ)
assert request.full_url == 'https://bender.com'
environ['SCRIPT_NAME'] = '/Farnsworth'
request = WSGIRequestWrapper.from_wsgi_environ(environ)
assert request.full_url == 'https://bender.com/Farnsworth'
environ['PATH_INFO'] = '/PlanetExpress'
request = WSGIRequestWrapper.from_wsgi_environ(environ)
assert request.full_url == 'https://bender.com/Farnsworth/PlanetExpress'
environ['QUERY_STRING'] = 'Bender=antiquing'
request = WSGIRequestWrapper.from_wsgi_environ(environ)
assert request.full_url == \
'https://bender.com/Farnsworth/PlanetExpress?Bender=antiquing'
def test_url():
environ = {
'wsgi.url_scheme': 'http',
'HTTP_HOST': 'bender.com'
}
request = WSGIRequestWrapper.from_wsgi_environ(environ)
assert request.full_url == 'http://bender.com'
environ = {
'wsgi.url_scheme': 'http',
'SERVER_NAME': 'bender.com',
'SERVER_PORT': '80'
}
request = WSGIRequestWrapper.from_wsgi_environ(environ)
assert request.full_url == 'http://bender.com'
environ['SERVER_PORT'] = '8888'
request = WSGIRequestWrapper.from_wsgi_environ(environ)
assert request.full_url == 'http://bender.com:8888'
environ['wsgi.url_scheme'] = 'https'
request = WSGIRequestWrapper.from_wsgi_environ(environ)
assert request.full_url == 'https://bender.com:8888'
environ['SERVER_PORT'] = '443'
request = WSGIRequestWrapper.from_wsgi_environ(environ)
assert request.full_url == 'https://bender.com'
environ['SCRIPT_NAME'] = '/Farnsworth'
request = WSGIRequestWrapper.from_wsgi_environ(environ)
assert request.full_url == 'https://bender.com/Farnsworth'
environ['PATH_INFO'] = '/PlanetExpress'
request = WSGIRequestWrapper.from_wsgi_environ(environ)
assert request.full_url == 'https://bender.com/Farnsworth/PlanetExpress'
environ['QUERY_STRING'] = 'Bender=antiquing'
request = WSGIRequestWrapper.from_wsgi_environ(environ)
assert request.full_url == \
'https://bender.com/Farnsworth/PlanetExpress?Bender=antiquing'
def wsgi_tracing_middleware(environ, start_response):
if environ['PATH_INFO'] == '/metrics':
return prometheus_app(environ, start_response)
# TODO find out if the route can be retrieved from somewhere
request = WSGIRequestWrapper.from_wsgi_environ(environ)
span = before_request(request=request, tracer=tracer)
nm = '%s %s %s' % (environ['wsgi.url_scheme'].upper(),
request.operation.upper(), environ['PATH_INFO'])
nm = _NORM_RE.sub('-', nm)
span.set_operation_name(nm)
# Wrapper around the real start_response object to log
# additional information to opentracing Span
def start_response_wrapper(status, response_headers, exc_info=None):
span.set_tag('error', exc_info is not None)
span.set_tag('http.status_code', status[:3])
span.finish()
return start_response(status, response_headers)
with opentracing_instrumentation.span_in_context(span):
return other_wsgi(environ, start_response_wrapper)