notifications-api/tests/app/public_contracts/test_GET_notification.py

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_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_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')