diff --git a/app/v2/template/post_template.py b/app/v2/template/post_template.py index ca94ee157..c4ecbc6fe 100644 --- a/app/v2/template/post_template.py +++ b/app/v2/template/post_template.py @@ -6,6 +6,7 @@ from werkzeug.exceptions import abort from app import api_user from app.dao import templates_dao +from app.models import SMS_TYPE from app.schema_validation import validate from app.utils import get_template_instance from app.v2.errors import BadRequestError @@ -16,6 +17,9 @@ from app.v2.template.template_schemas import post_template_preview_request, crea @v2_template_blueprint.route("//preview", methods=['POST']) def post_template_preview(template_id): _data = request.get_json() + if _data is None: + _data = {} + _data['id'] = template_id data = validate(_data, post_template_preview_request) @@ -28,7 +32,10 @@ def post_template_preview(template_id): check_placeholders(template_object) + subject = template_object.subject if template.template_type != SMS_TYPE else None + resp = create_post_template_preview_response(template=template, + subject=subject, body=str(template_object)) return jsonify(resp), 200 diff --git a/app/v2/template/template_schemas.py b/app/v2/template/template_schemas.py index 0ad632b9a..dac2c4cde 100644 --- a/app/v2/template/template_schemas.py +++ b/app/v2/template/template_schemas.py @@ -49,7 +49,7 @@ post_template_preview_request = { "id": uuid, "personalisation": personalisation }, - "required": ["id", "personalisation"] + "required": ["id"] } post_template_preview_response = { @@ -68,11 +68,11 @@ post_template_preview_response = { } -def create_post_template_preview_response(template, body): +def create_post_template_preview_response(template, subject, body): return { "id": template.id, "type": template.template_type, "version": template.version, "body": body, - "subject": template.subject + "subject": subject } diff --git a/tests/app/db.py b/tests/app/db.py index bcd94c9c3..e2a24f1c3 100644 --- a/tests/app/db.py +++ b/tests/app/db.py @@ -39,6 +39,7 @@ def create_service(user=None, service_name="Sample service", service_id=None): def create_template( service, template_type=SMS_TYPE, + subject='Template subject', content='Dear Sir/Madam, Hello. Yours Truly, The Government.', template_id=None ): @@ -50,7 +51,7 @@ def create_template( 'created_by': service.created_by, } if template_type != SMS_TYPE: - data['subject'] = 'Template subject' + data['subject'] = subject template = Template(**data) dao_create_template(template) return template diff --git a/tests/app/v2/template/test_post_template.py b/tests/app/v2/template/test_post_template.py index a6c8d542d..4501852c2 100644 --- a/tests/app/v2/template/test_post_template.py +++ b/tests/app/v2/template/test_post_template.py @@ -3,27 +3,61 @@ import uuid from flask import json -from app.models import TEMPLATE_TYPES +from app.models import SMS_TYPE, TEMPLATE_TYPES from tests import create_authorization_header from tests.app.db import create_template -valid_data = { +valid_personalisation = { 'personalisation': {'Name': 'Jo'} } +valid_post = [ + ( + "Some subject", + "Some content", + None, + "Some subject", + "Some content" + ), + ( + "Some subject", + "Dear ((Name)), Hello. Yours Truly, The Government.", + valid_personalisation, + "Some subject", + "Dear Jo, Hello. Yours Truly, The Government." + ), + ( + "Message for ((Name))", + "Dear ((Name)), Hello. Yours Truly, The Government.", + valid_personalisation, + "Message for Jo", + "Dear Jo, Hello. Yours Truly, The Government." + ), + ( + "Message for ((Name))", + "Some content", + valid_personalisation, + "Message for Jo", + "Some content" + ), +] + @pytest.mark.parametrize("tmp_type", TEMPLATE_TYPES) -def test_valid_post_template_returns_200(client, sample_service, tmp_type): +@pytest.mark.parametrize("subject,content,post_data,expected_subject,expected_content", valid_post) +def test_valid_post_template_returns_200( + client, sample_service, tmp_type, subject, content, post_data, expected_subject, expected_content): template = create_template( sample_service, template_type=tmp_type, - content='Dear ((Name)), Hello. Yours Truly, The Government.') + subject=subject, + content=content) auth_header = create_authorization_header(service_id=sample_service.id) response = client.post( path='/v2/template/{}/preview'.format(template.id), - data=json.dumps(valid_data), + data=json.dumps(post_data), headers=[('Content-Type', 'application/json'), auth_header]) assert response.status_code == 200 @@ -31,7 +65,9 @@ def test_valid_post_template_returns_200(client, sample_service, tmp_type): resp_json = json.loads(response.get_data(as_text=True)) assert resp_json['id'] == str(template.id) - assert 'Dear {}'.format(valid_data['personalisation']['Name']) in resp_json['body'] + if tmp_type != SMS_TYPE: + assert expected_subject in resp_json['subject'] + assert expected_content in resp_json['body'] @pytest.mark.parametrize("tmp_type", TEMPLATE_TYPES) @@ -45,7 +81,7 @@ def test_invalid_post_template_returns_400(client, sample_service, tmp_type): response = client.post( path='/v2/template/{}/preview'.format(template.id), - data=json.dumps(valid_data), + data=json.dumps(valid_personalisation), headers=[('Content-Type', 'application/json'), auth_header]) assert response.status_code == 400 @@ -61,7 +97,7 @@ def test_post_template_with_non_existent_template_id_returns_404(client, fake_uu response = client.post( path='/v2/template/{}/preview'.format(fake_uuid), - data=json.dumps(valid_data), + data=json.dumps(valid_personalisation), headers=[('Content-Type', 'application/json'), auth_header]) assert response.status_code == 404 diff --git a/tests/app/v2/template/test_template_schemas.py b/tests/app/v2/template/test_template_schemas.py index 97fca1ca4..c89802942 100644 --- a/tests/app/v2/template/test_template_schemas.py +++ b/tests/app/v2/template/test_template_schemas.py @@ -54,7 +54,6 @@ invalid_json_post_args = [ ({"id": "invalid_uuid", "personalisation": {"key": "value"}}, ["id is not a valid UUID"]), ({"id": str(uuid.uuid4()), "personalisation": "invalid_personalisation"}, ["personalisation should contain key value pairs"]), - ({"id": str(uuid.uuid4())}, ["personalisation is a required property"]), ({"personalisation": "invalid_personalisation"}, ["id is a required property", "personalisation is a required property",