mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-06 03:13:42 -05:00
34 lines
909 B
Python
34 lines
909 B
Python
import pytest
|
|
|
|
from tests.conftest import set_config_values
|
|
|
|
|
|
@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(
|
|
notify_admin, check_proxy_header, header_value, expected_code
|
|
):
|
|
with set_config_values(
|
|
notify_admin,
|
|
{
|
|
"ROUTE_SECRET_KEY_1": "key_1",
|
|
"ROUTE_SECRET_KEY_2": "",
|
|
"CHECK_PROXY_HEADER": check_proxy_header,
|
|
},
|
|
):
|
|
with notify_admin.test_client() as client:
|
|
response = client.get(
|
|
path="/_status?elb=True",
|
|
headers=[
|
|
("X-Custom-forwarder", header_value),
|
|
],
|
|
)
|
|
assert response.status_code == expected_code
|