notifications-api/tests/app/v2/templates/test_templates_schemas.py

import uuid

import pytest
from flask import json

from app.models import EMAIL_TYPE, SMS_TYPE, TEMPLATE_TYPES
from app.v2.templates.templates_schemas import (
    get_all_template_request,
    get_all_template_response
)
from app.schema_validation import validate
from jsonschema.exceptions import ValidationError


valid_json_get_all_response = [
    {
        "templates": [
            {"id": str(uuid.uuid4()), "version": 1, "uri": "http://template/id"},
            {"id": str(uuid.uuid4()), "version": 2, "uri": "http://template/id"}
        ]
    },
    {
        "templates": [{"id": str(uuid.uuid4()), "version": 1, "uri": "http://template/id"}]
    },
    {
        "templates": []
    }
]

invalid_json_get_all_response = [
    ({
        "templates": [
            {"id": 'invalid_id', "version": 1, "uri": "http://template/id"}
        ]
    }, ['templates is not a valid UUID']),
    ({
        "templates": [
            {"id": str(uuid.uuid4()), "version": 'invalid_version', "uri": "http://template/id"}
        ]
    }, ['templates invalid_version is not of type integer']),
    ({
        "templates": [
            {"id": str(uuid.uuid4()), "version": 1, "uri": "invalid_uri"}
        ]
    }, ['templates invalid_uri is not a valid URI.']),
    ({}, ['templates is a required property']),
    ({
        "templates": [{"version": 1, "uri": "http://template/id"}]
    }, ['templates id is a required property']),
    ({
        "templates": [{"id": str(uuid.uuid4()), "uri": "http://template/id"}]
    }, ['templates version is a required property']),
    ({
        "templates": [{"id": str(uuid.uuid4()), "version": 1}]
    }, ['templates uri is a required property']),
    ({
        "templates": [{"version": 1}]
    }, ['templates id is a required property', 'templates uri is a required property']),
]


@pytest.mark.parametrize("template_type", TEMPLATE_TYPES)
def test_get_all_template_request_schema_against_no_args_is_valid(template_type):
    data = {}
    assert validate(data, get_all_template_request) == data


@pytest.mark.parametrize("template_type", TEMPLATE_TYPES)
def test_get_all_template_request_schema_against_valid_args_is_valid(template_type):
    data = {'type': template_type}
    assert validate(data, get_all_template_request) == data


@pytest.mark.parametrize("template_type", TEMPLATE_TYPES)
def test_get_all_template_request_schema_against_invalid_args_is_invalid(template_type):
    data = {'type': 'unknown'}

    with pytest.raises(ValidationError) as e:
        validate(data, get_all_template_request)
    errors = json.loads(str(e.value))

    assert errors['status_code'] == 400
    assert len(errors['errors']) == 1
    assert errors['errors'][0]['message'] == 'type unknown is not one of [sms, email, letter]'


@pytest.mark.parametrize("response", valid_json_get_all_response)
def test_valid_get_all_templates_response_schema_is_valid(response):
    assert validate(response, get_all_template_response) == response


@pytest.mark.parametrize("response,error_messages", invalid_json_get_all_response)
def test_invalid_get_all_templates_response_schema_is_invalid(response, error_messages):
    with pytest.raises(ValidationError) as e:
        validate(response, get_all_template_response)
    errors = json.loads(str(e.value))

    assert errors['status_code'] == 400
    assert len(errors['errors']) == len(error_messages)
    for error in errors['errors']:
        assert error['message'] in error_messages