Refactored code to personalise subject

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

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

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",