mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-22 16:31:15 -05:00
0312e2a528
In response to [1]. [1]: https://github.com/alphagov/notifications-api/pull/3300#discussion_r681653248
81 lines
3.0 KiB
Python
81 lines
3.0 KiB
Python
from app.dao.api_key_dao import save_model_api_key
|
|
from app.models import KEY_TYPE_NORMAL, ApiKey
|
|
from app.v2.notifications.notification_schemas import (
|
|
get_notification_response,
|
|
get_notifications_response,
|
|
)
|
|
from tests import create_service_authorization_header
|
|
|
|
from . import return_json_from_response, validate, validate_v0
|
|
|
|
|
|
def _get_notification(client, notification, url):
|
|
save_model_api_key(ApiKey(
|
|
service=notification.service,
|
|
name='api_key',
|
|
created_by=notification.service.created_by,
|
|
key_type=KEY_TYPE_NORMAL
|
|
))
|
|
auth_header = create_service_authorization_header(service_id=notification.service_id)
|
|
return client.get(url, headers=[auth_header])
|
|
|
|
|
|
# v2
|
|
|
|
def test_get_v2_sms_contract(client, sample_notification):
|
|
response_json = return_json_from_response(_get_notification(
|
|
client, sample_notification, '/v2/notifications/{}'.format(sample_notification.id)
|
|
))
|
|
validate(response_json, get_notification_response)
|
|
|
|
|
|
def test_get_v2_email_contract(client, sample_email_notification):
|
|
response_json = return_json_from_response(_get_notification(
|
|
client, sample_email_notification, '/v2/notifications/{}'.format(sample_email_notification.id)
|
|
))
|
|
validate(response_json, get_notification_response)
|
|
|
|
|
|
def test_get_v2_notifications_contract(client, sample_notification):
|
|
response_json = return_json_from_response(_get_notification(
|
|
client, sample_notification, '/v2/notifications'
|
|
))
|
|
validate(response_json, get_notifications_response)
|
|
|
|
|
|
# v0
|
|
|
|
def test_get_api_sms_contract(client, sample_notification):
|
|
response_json = return_json_from_response(_get_notification(
|
|
client, sample_notification, '/notifications/{}'.format(sample_notification.id)
|
|
))
|
|
validate_v0(response_json, 'GET_notification_return_sms.json')
|
|
|
|
|
|
def test_get_api_email_contract(client, sample_email_notification):
|
|
response_json = return_json_from_response(_get_notification(
|
|
client, sample_email_notification, '/notifications/{}'.format(sample_email_notification.id)
|
|
))
|
|
validate_v0(response_json, 'GET_notification_return_email.json')
|
|
|
|
|
|
def test_get_job_sms_contract(client, sample_notification):
|
|
response_json = return_json_from_response(_get_notification(
|
|
client, sample_notification, '/notifications/{}'.format(sample_notification.id)
|
|
))
|
|
validate_v0(response_json, 'GET_notification_return_sms.json')
|
|
|
|
|
|
def test_get_job_email_contract(client, sample_email_notification):
|
|
response_json = return_json_from_response(_get_notification(
|
|
client, sample_email_notification, '/notifications/{}'.format(sample_email_notification.id)
|
|
))
|
|
validate_v0(response_json, 'GET_notification_return_email.json')
|
|
|
|
|
|
def test_get_notifications_contract(client, sample_notification, sample_email_notification):
|
|
response_json = return_json_from_response(_get_notification(
|
|
client, sample_notification, '/notifications'
|
|
))
|
|
validate_v0(response_json, 'GET_notifications_return.json')
|