diff --git a/tests/app/dao/test_templates_dao.py b/tests/app/dao/test_templates_dao.py index 43cc88fd7..607330097 100644 --- a/tests/app/dao/test_templates_dao.py +++ b/tests/app/dao/test_templates_dao.py @@ -15,7 +15,7 @@ from app.dao.templates_dao import ( from app.models import Template, TemplateHistory, TemplateRedacted from tests.app.conftest import sample_template as create_sample_template -from tests.app.db import create_template +from tests.app.db import create_template, create_letter_contact @pytest.mark.parametrize('template_type, subject', [ @@ -53,6 +53,23 @@ def test_create_template_creates_redact_entry(sample_service): assert redacted.updated_by_id == sample_service.created_by_id +def test_create_template_with_reply_to(sample_service, sample_user): + letter_contact = create_letter_contact(sample_service, 'Edinburgh, ED1 1AA') + + data = { + 'name': 'Sample Template', + 'template_type': "letter", + 'content': "Template content", + 'service': sample_service, + 'created_by': sample_user, + 'reply_to': letter_contact.id, + } + template = Template(**data) + dao_create_template(template) + + assert dao_get_all_templates_for_service(sample_service.id)[0].reply_to == letter_contact.id + + def test_update_template(sample_service, sample_user): data = { 'name': 'Sample Template', @@ -71,6 +88,29 @@ def test_update_template(sample_service, sample_user): assert dao_get_all_templates_for_service(sample_service.id)[0].name == 'new name' +def test_update_template_reply_to(sample_service, sample_user): + letter_contact = create_letter_contact(sample_service, 'Edinburgh, ED1 1AA') + + data = { + 'name': 'Sample Template', + 'template_type': "letter", + 'content': "Template content", + 'service': sample_service, + 'created_by': sample_user, + } + template = Template(**data) + dao_create_template(template) + created = dao_get_all_templates_for_service(sample_service.id)[0] + assert created.reply_to is None + + created.reply_to = letter_contact.id + dao_update_template(created) + assert dao_get_all_templates_for_service(sample_service.id)[0].reply_to == letter_contact.id + + template_history = TemplateHistory.query.filter_by(id=created.id, version=2).one() + assert template_history.service_letter_contact_id + + def test_redact_template(sample_template): redacted = TemplateRedacted.query.one() assert redacted.template_id == sample_template.id