def test_push(): assert request_id.get() is None request_id.push(None) assert request_id.get() is None request_id.push('id') assert request_id.get() == 'id' request_id.push(None) assert request_id.get() is None
def test_decorator(id): @request_id.decorator(lambda: id) def test(): assert request_id.get() == id assert request_id.get() is None test() assert request_id.get() is None
def test_push(): assert request_id.get() is None request_id.push(None) assert request_id.get() is None request_id.push('id') assert request_id.get() == 'id' request_id.push(None) assert request_id.get() is None
def test_decorator(id): @request_id.decorator(lambda: id) def test(): assert request_id.get() == id assert request_id.get() is None test() assert request_id.get() is None
def app(environ, start_response): start_response(200, []) return [ environ.get('REQUEST_ID'), request_id.get(), request_context.extra['request_id'], ]
def send(request, **kwargs): rid = request_id.get() if rid and HEADER not in request.headers: request.headers[HEADER] = rid try: return func(request, **kwargs) except Exception as e: record_request(request, None, e) raise
def test(): assert request_id.get() == id
def test_context(id): assert request_id.get() is None with request_id.context(id): assert request_id.get() == id assert request_id.get() is None
def app(environ, start_response): start_response(200, []) return [ environ.get('REQUEST_ID'), request_id.get(), ]
def test_context_existing_id(id): request_id.push('existing') assert request_id.get() == 'existing' with request_id.context(id): assert request_id.get() == id assert request_id.get() == 'existing'
def test_context_existing_id(id): request_id.push('existing') assert request_id.get() == 'existing' with request_id.context(id): assert request_id.get() == id assert request_id.get() == 'existing'
def test_context(id): assert request_id.get() is None with request_id.context(id): assert request_id.get() == id assert request_id.get() is None
def test(): assert request_id.get() == id
def delay(task, *args, **kwargs): id = request_id.get() if id: kwargs['request_id'] = id return task.delay(*args, **kwargs)