From ccfa1c82289ba816557a9473026872518e1d370c Mon Sep 17 00:00:00 2001 From: "Pea (Malgorzata Tyczynska)" Date: Wed, 26 Jun 2019 18:01:20 +0100 Subject: [PATCH] Revert "Ensure dao_create_template doesn't clash with new trigger" --- app/dao/templates_dao.py | 21 +++++++++------------ tests/app/v2/template/test_get_template.py | 4 ++-- 2 files changed, 11 insertions(+), 14 deletions(-) diff --git a/app/dao/templates_dao.py b/app/dao/templates_dao.py index 88b479bac..63079ef68 100644 --- a/app/dao/templates_dao.py +++ b/app/dao/templates_dao.py @@ -24,19 +24,16 @@ def dao_create_template(template): template.id = uuid.uuid4() # must be set now so version history model can use same id template.archived = False - # The template redacted is now being inserted via a trigger - # this will catch any Notify templates being added by a db migration - if not TemplateRedacted.query.filter(TemplateRedacted.template_id == template.id).first(): - redacted_dict = { - "template": template, - "redact_personalisation": False, - } - if template.created_by: - redacted_dict.update({"updated_by": template.created_by}) - else: - redacted_dict.update({"updated_by_id": template.created_by_id}) + redacted_dict = { + "template": template, + "redact_personalisation": False, + } + if template.created_by: + redacted_dict.update({"updated_by": template.created_by}) + else: + redacted_dict.update({"updated_by_id": template.created_by_id}) - template.template_redacted = TemplateRedacted(**redacted_dict) + template.template_redacted = TemplateRedacted(**redacted_dict) db.session.add(template) diff --git a/tests/app/v2/template/test_get_template.py b/tests/app/v2/template/test_get_template.py index aae49a8b9..dadf82983 100644 --- a/tests/app/v2/template/test_get_template.py +++ b/tests/app/v2/template/test_get_template.py @@ -17,7 +17,7 @@ valid_version_params = [None, 1] ]) @pytest.mark.parametrize("version", valid_version_params) def test_get_template_by_id_returns_200( - client, sample_service, mock, tmp_type, expected_name, expected_subject, version, postage + client, sample_service, tmp_type, expected_name, expected_subject, version, postage ): template = create_template(sample_service, template_type=tmp_type) auth_header = create_authorization_header(service_id=sample_service.id) @@ -36,7 +36,7 @@ def test_get_template_by_id_returns_200( 'id': '{}'.format(template.id), 'type': '{}'.format(template.template_type), 'created_at': template.created_at.strftime(DATETIME_FORMAT), - 'updated_at': mock.ANY, + 'updated_at': None, 'version': template.version, 'created_by': template.created_by.email_address, 'body': template.content,