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