def test_middleware_overwrites_header(environ, id, monkeypatch):
def proxy(environ, start_response):
start_response(200, [('X-Request-Id', 'other-id')])
return 'ok'
monkeypatch.setattr(request_id, 'generate', lambda: id)
middleware = request_id.RequestIdMiddleware(proxy)
body, status, headers = run_wsgi(middleware, environ)
assert headers['X-Request-Id'] == id
def test_middleware_soft_request_timeout_disabled_by_default(
environ, sentry_messages):
def app(environ, start_response):
start_response(200, [])
return []
mw = talisker.sentry.get_middleware(app)
body, _, _ = conftest.run_wsgi(mw, environ)
list(body)
assert len(sentry_messages) == 0
def test_middleware_soft_request_timeout(monkeypatch, environ,
sentry_messages):
monkeypatch.setitem(os.environ, 'TALISKER_SOFT_REQUEST_TIMEOUT', '0')
def app(environ, start_response):
start_response(200, [])
return []
mw = talisker.sentry.get_middleware(app)
body, _, _ = conftest.run_wsgi(mw, environ)
list(body)
assert 'Start_response over timeout: 0' == sentry_messages[0]['message']
def test_middleware_alt_header(environ, id):
middleware = request_id.RequestIdMiddleware(app, 'X-Alternate')
environ['HTTP_X_ALTERNATE'] = id
body, status, headers = run_wsgi(middleware, environ)
assert list(set(body)) == [id]
assert ('X-Alternate', id) in headers
def test_middleware_without_id(environ, id, monkeypatch):
monkeypatch.setattr(request_id, 'generate', lambda: id)
middleware = request_id.RequestIdMiddleware(app)
body, status, headers = run_wsgi(middleware, environ)
assert list(set(body)) == [id]
assert ('X-Request-Id', id) in headers
def test_middleware_with_id(environ, id):
middleware = request_id.RequestIdMiddleware(app)
environ['HTTP_X_REQUEST_ID'] = id
body, status, headers = run_wsgi(middleware, environ)
assert list(set(body)) == [id]
assert ('X-Request-Id', id) in headers
