darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-09-09 17:45:32 -04:00
Code Issues Packages Projects Releases Wiki Activity

Use parametrize

Browse Source
This commit is contained in:
Athanasios Voutsadakis
2017-11-16 16:34:10 +00:00
parent 1c78b938b4
commit 8786ce79f4
1 changed files with 14 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,33 +1,26 @@
import pytest
from tests.conftest import set_config_values from tests.conftest import set_config_values
def test_route_correct_secret_key(app_, client): @pytest.mark.parametrize('check_proxy_header,header_value,expected_code', [
(True, 'key_1', 200),
(True, 'wrong_key', 403),
(False, 'wrong_key', 200),
(False, 'key_1', 200),
])
def test_route_correct_secret_key(app_, check_proxy_header, header_value, expected_code):
with set_config_values(app_, { with set_config_values(app_, {
'ROUTE_SECRET_KEY_1': 'key_1', 'ROUTE_SECRET_KEY_1': 'key_1',
'ROUTE_SECRET_KEY_2': '', 'ROUTE_SECRET_KEY_2': '',
'DEBUG': False, 'CHECK_PROXY_HEADER': check_proxy_header,
}): }):
client = app_.test_client()
response = client.get( response = client.get(
path='/_status', path='/_status?elb=True',
headers=[ headers=[
('X-Custom-forwarder', 'key_1'), ('X-Custom-forwarder', header_value),
] ]
) )
assert response.status_code == 200 assert response.status_code == expected_code
def test_route_incorrect_secret_key(app_, client):
with set_config_values(app_, {
'ROUTE_SECRET_KEY_1': 'key_1',
'ROUTE_SECRET_KEY_2': '',
'DEBUG': False,
}):
response = client.get(
path='/_status',
headers=[
('X-Custom-forwarder', 'wrong_key'),
]
)
assert response.status_code == 403