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)
