darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-21 07:51:13 -05:00
Code Issues Packages Projects Releases Wiki Activity

Colocate requires_auth test with the others

Browse Source 
This is to help minimise the diff for the next commit.
This commit is contained in:
Ben Thorner
2021-07-29 11:55:04 +01:00
parent f5d2edaa0a
commit cd0a950571
1 changed files with 22 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
tests/app/authentication/test_authentication.py
View File

@@ -298,6 +298,28 @@ def test_requires_auth_should_attach_the_current_api_key_to_current_app(
assert str(api_user.id) == str(sample_api_key.id) assert str(api_user.id) == str(sample_api_key.id)
def test_requires_auth_should_cache_service_and_api_key_lookups(
mocker,
client,
service_jwt_token
):
mock_get_api_keys = mocker.patch(
'app.serialised_models.get_model_api_keys',
wraps=get_model_api_keys,
)
mock_get_service = mocker.patch(
'app.serialised_models.dao_fetch_service_by_id',
wraps=dao_fetch_service_by_id,
)
request.headers = {'Authorization': f'Bearer {service_jwt_token}'}
requires_auth()
requires_auth() # second request
mock_get_api_keys.assert_called_once()
mock_get_service.assert_called_once()
def test_requires_internal_auth_checks_proxy_key( def test_requires_internal_auth_checks_proxy_key(
client, client,
mocker, mocker,
@@ -336,25 +358,3 @@ def test_requires_internal_auth_sets_global_variables(
request.headers = {'Authorization': 'Bearer {}'.format(internal_jwt_token)} request.headers = {'Authorization': 'Bearer {}'.format(internal_jwt_token)}
requires_my_internal_app_auth() requires_my_internal_app_auth()
assert g.service_id == 'my-internal-app' assert g.service_id == 'my-internal-app'
def test_requires_auth_should_cache_service_and_api_key_lookups(
mocker,
client,
service_jwt_token
):
mock_get_api_keys = mocker.patch(
'app.serialised_models.get_model_api_keys',
wraps=get_model_api_keys,
)
mock_get_service = mocker.patch(
'app.serialised_models.dao_fetch_service_by_id',
wraps=dao_fetch_service_by_id,
)
request.headers = {'Authorization': f'Bearer {service_jwt_token}'}
requires_auth()
requires_auth() # second request
mock_get_api_keys.assert_called_once()
mock_get_service.assert_called_once()