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

Backfill missing tests for requires_auth function

Browse Source
This commit is contained in:
Ben Thorner
2021-07-29 12:09:01 +01:00
parent cd0a950571
commit 323feedb1f
1 changed files with 17 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -6,7 +6,7 @@ import pytest
from flask import current_app, g, request
from notifications_python_client.authentication import create_jwt_token
from app import api_user
from app import db
from app.authentication.auth import (
GENERAL_TOKEN_ERROR_MESSAGE,
AuthError,
@@ -288,14 +288,28 @@ def test_requires_auth_returns_error_when_service_inactive(
assert exc.value.short_message == 'Invalid token: service is archived'
def test_requires_auth_should_attach_the_current_api_key_to_current_app(
def test_requires_auth_should_assign_global_variables(
client,
sample_api_key,
service_jwt_token,
):
request.headers = {'Authorization': 'Bearer {}'.format(service_jwt_token)}
requires_auth()
assert str(api_user.id) == str(sample_api_key.id)
assert g.api_user.id == sample_api_key.id
assert g.service_id == sample_api_key.service_id
assert g.authenticated_service.id == str(sample_api_key.service_id)
def test_requires_auth_errors_if_service_has_no_api_keys(
client,
sample_api_key,
service_jwt_token,
):
db.session.delete(sample_api_key)
request.headers = {'Authorization': 'Bearer {}'.format(service_jwt_token)}
with pytest.raises(AuthError) as exc:
requires_auth()
assert exc.value.short_message == 'Invalid token: service has no API keys'
def test_requires_auth_should_cache_service_and_api_key_lookups(