import pytest from flask import json from app.models import EMAIL_TYPE, SMS_TYPE, TEMPLATE_TYPES from app.utils import DATETIME_FORMAT from tests import create_service_authorization_header from tests.app.db import create_template valid_version_params = [None, 1] @pytest.mark.parametrize( "tmp_type, expected_name, expected_subject", [ (SMS_TYPE, "sms Template Name", None), (EMAIL_TYPE, "email Template Name", "Template subject"), ], ) @pytest.mark.parametrize("version", valid_version_params) def test_get_template_by_id_returns_200( client, sample_service, tmp_type, expected_name, expected_subject, version ): template = create_template(sample_service, template_type=tmp_type) auth_header = create_service_authorization_header(service_id=sample_service.id) version_path = "/version/{}".format(version) if version else "" response = client.get( path="/v2/template/{}{}".format(template.id, version_path), headers=[("Content-Type", "application/json"), auth_header], ) assert response.status_code == 200 assert response.headers["Content-type"] == "application/json" json_response = json.loads(response.get_data(as_text=True)) expected_response = { "id": "{}".format(template.id), "type": "{}".format(template.template_type), "created_at": template.created_at.strftime(DATETIME_FORMAT), "updated_at": None, "version": template.version, "created_by": template.created_by.email_address, "body": template.content, "subject": expected_subject, "name": expected_name, "personalisation": {}, } assert json_response == expected_response @pytest.mark.parametrize( "create_template_args, expected_personalisation", [ ( { "template_type": SMS_TYPE, "content": "Hello ((placeholder)) ((conditional??yes))", }, { "placeholder": {"required": True}, "conditional": {"required": True}, }, ), ( { "template_type": EMAIL_TYPE, "subject": "((subject))", "content": "((content))", }, { "subject": {"required": True}, "content": {"required": True}, }, ), ], ) @pytest.mark.parametrize("version", valid_version_params) def test_get_template_by_id_returns_placeholders( client, sample_service, version, create_template_args, expected_personalisation, ): template = create_template(sample_service, **create_template_args) auth_header = create_service_authorization_header(service_id=sample_service.id) version_path = "/version/{}".format(version) if version else "" response = client.get( path="/v2/template/{}{}".format(template.id, version_path), headers=[("Content-Type", "application/json"), auth_header], ) json_response = json.loads(response.get_data(as_text=True)) assert json_response["personalisation"] == expected_personalisation def test_get_template_with_non_existent_template_id_returns_404( client, fake_uuid, sample_service ): auth_header = create_service_authorization_header(service_id=sample_service.id) response = client.get( path="/v2/template/{}".format(fake_uuid), headers=[("Content-Type", "application/json"), auth_header], ) assert response.status_code == 404 assert response.headers["Content-type"] == "application/json" json_response = json.loads(response.get_data(as_text=True)) assert json_response == { "errors": [{"error": "NoResultFound", "message": "No result found"}], "status_code": 404, } @pytest.mark.parametrize("tmp_type", TEMPLATE_TYPES) def test_get_template_with_non_existent_version_returns_404( client, sample_service, tmp_type ): template = create_template(sample_service, template_type=tmp_type) auth_header = create_service_authorization_header(service_id=sample_service.id) invalid_version = template.version + 1 response = client.get( path="/v2/template/{}/version/{}".format(template.id, invalid_version), headers=[("Content-Type", "application/json"), auth_header], ) assert response.status_code == 404 assert response.headers["Content-type"] == "application/json" json_response = json.loads(response.get_data(as_text=True)) assert json_response == { "errors": [{"error": "NoResultFound", "message": "No result found"}], "status_code": 404, }