darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-03 09:51:11 -05:00
Code Issues Packages Projects Releases Wiki Activity

Added authorization headers for all requests

Browse Source
This commit is contained in:
Rebecca Law
2016-01-15 16:22:03 +00:00
parent 5f59b631e1
commit 0ad292300d
10 changed files with 400 additions and 204 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/app/template/test_rest.py
View File

@@ -1,7 +1,8 @@
import json
from app.models import Template
from flask import url_for
from tests import create_authorization_header
def test_get_template_list(notify_api, notify_db, notify_db_session, sample_template):
"""
@@ -9,7 +10,11 @@ def test_get_template_list(notify_api, notify_db, notify_db_session, sample_temp
"""
with notify_api.test_request_context():
with notify_api.test_client() as client:
response = client.get(url_for('template.get_template'))
auth_header = create_authorization_header(service_id=sample_template.service_id,
path=url_for('template.get_template'),
method='GET')
response = client.get(url_for('template.get_template'),
headers=[auth_header])
assert response.status_code == 200
json_resp = json.loads(response.get_data(as_text=True))
assert len(json_resp['data']) == 1
@@ -23,8 +28,13 @@ def test_get_template(notify_api, notify_db, notify_db_session, sample_template)
"""
with notify_api.test_request_context():
with notify_api.test_client() as client:
auth_header = create_authorization_header(service_id=sample_template.service_id,
path=url_for('template.get_template',
template_id=sample_template.id),
method='GET')
resp = client.get(url_for(
'template.get_template', template_id=sample_template.id))
'template.get_template', template_id=sample_template.id),
headers=[auth_header])
assert resp.status_code == 200
json_resp = json.loads(resp.get_data(as_text=True))
assert json_resp['data']['name'] == sample_template.name