darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-05-05 16:38:59 -04:00
Code Issues Packages Projects Releases Wiki Activity

Both trace_id and span_id headers must be present

Browse Source
This commit is contained in:
Athanasios Voutsadakis
2018-08-14 18:12:06 +01:00
parent 6786ff8fde
commit 984e09cc6a
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
app/notify_client/__init__.py
View File

@@ -38,6 +38,7 @@ class NotifyAdminAPIClient(BaseAPIClient):
if not has_request_context(): if not has_request_context():
return headers return headers
headers['X-B3-TraceId'] = request.request_id headers['X-B3-TraceId'] = request.request_id
headers['X-B3-SpanId'] = request.span_id
return headers return headers
def check_inactive_service(self): def check_inactive_service(self):

3
tests/app/notify_client/test_notify_admin_api_client.py
View File

@@ -99,6 +99,7 @@ def test_generate_headers_sets_request_id_if_in_request_context(app_):
headers = api_client.generate_headers('api_token') headers = api_client.generate_headers('api_token')
assert set(headers.keys()) == { assert set(headers.keys()) == {
'Authorization', 'Content-type', 'User-agent', 'X-Custom-Forwarder', 'X-B3-TraceId' 'Authorization', 'Content-type', 'User-agent', 'X-Custom-Forwarder', 'X-B3-TraceId', 'X-B3-SpanId',
} }
assert headers['X-B3-TraceId'] == request_context.request.request_id assert headers['X-B3-TraceId'] == request_context.request.request_id
assert headers['X-B3-SpanId'] == request_context.request.span_id