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

Merge branch 'master' into caching-with-redis

Browse Source
This commit is contained in:
Martyn Inglis
2016-11-10 11:09:04 +00:00
parent aea2356ae1 bec20d3854
commit 23cc37e2d1
3 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/authentication/auth.py
View File

@@ -27,7 +27,7 @@ def get_auth_token(req):
if not auth_header: if not auth_header:
raise AuthError('Unauthorized, authentication token must be provided', 401) raise AuthError('Unauthorized, authentication token must be provided', 401)
auth_scheme = auth_header[:7] auth_scheme = auth_header[:7].title()
if auth_scheme != 'Bearer ': if auth_scheme != 'Bearer ':
raise AuthError('Unauthorized, authentication bearer scheme must be used', 401) raise AuthError('Unauthorized, authentication bearer scheme must be used', 401)

2
app/celery/tasks.py
View File

@@ -136,7 +136,7 @@ def send_sms(self,
) )
current_app.logger.info( current_app.logger.info(
"SMS {} created at {}".format(notification_id, created_at) "SMS {} created at {} for job {}".format(notification_id, created_at, notification.get('job', None))
) )
except SQLAlchemyError as e: except SQLAlchemyError as e:

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

@@ -56,13 +56,14 @@ def test_should_not_allow_invalid_secret(notify_api, sample_api_key):
assert data['message'] == {"token": ['Invalid token: signature, api token is not valid']} assert data['message'] == {"token": ['Invalid token: signature, api token is not valid']}
def test_should_allow_valid_token(notify_api, sample_api_key): @pytest.mark.parametrize('scheme', ['bearer', 'Bearer'])
def test_should_allow_valid_token(notify_api, sample_api_key, scheme):
with notify_api.test_request_context(): with notify_api.test_request_context():
with notify_api.test_client() as client: with notify_api.test_client() as client:
token = __create_get_token(sample_api_key.service_id) token = __create_get_token(sample_api_key.service_id)
response = client.get( response = client.get(
'/service/{}'.format(str(sample_api_key.service_id)), '/service/{}'.format(str(sample_api_key.service_id)),
headers={'Authorization': 'Bearer {}'.format(token)} headers={'Authorization': '{} {}'.format(scheme, token)}
) )
assert response.status_code == 200 assert response.status_code == 200