darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-08-24 00:07:02 -04:00
Code Issues Packages Projects Releases Wiki Activity

Use test_client() as context manager

Browse Source
This commit is contained in:
Athanasios Voutsadakis
2017-11-16 17:02:38 +00:00
parent 3b30895a73
commit bde734dc75
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/app/main/test_request_header.py
View File

@@ -16,11 +16,11 @@ def test_route_correct_secret_key(app_, check_proxy_header, header_value, expect
'CHECK_PROXY_HEADER': check_proxy_header, 'CHECK_PROXY_HEADER': check_proxy_header,
}): }):
client = app_.test_client() with app_.test_client() as client:
response = client.get( response = client.get(
path='/_status?elb=True', path='/_status?elb=True',
headers=[ headers=[
('X-Custom-forwarder', header_value), ('X-Custom-forwarder', header_value),
] ]
) )
assert response.status_code == expected_code assert response.status_code == expected_code