From 47d047a3a88756231124f5f960b42fa935e4801f Mon Sep 17 00:00:00 2001 From: Pea Tyczynska Date: Mon, 24 Sep 2018 11:53:40 +0100 Subject: [PATCH] Add postage to fixtures and test postage display on notification page --- tests/__init__.py | 4 ++++ tests/app/main/views/test_notifications.py | 3 +++ 2 files changed, 7 insertions(+) diff --git a/tests/__init__.py b/tests/__init__.py index 6f412363e..1fc6e83db 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -369,6 +369,7 @@ def notification_json( reply_to_text=None, client_reference=None, created_by_name=None, + postage=None, ): if template is None: template = template_json(service_id, str(generate_uuid()), type_=template_type) @@ -388,6 +389,8 @@ def notification_json( if status is None: status = 'delivered' links = {} + if template_type == 'letter': + postage = 'second' if with_links: links = { @@ -415,6 +418,7 @@ def notification_json( 'service': service_id, 'template_version': template['version'], 'personalisation': personalisation or {}, + 'postage': postage, 'notification_type': template_type, 'reply_to_text': reply_to_text, 'client_reference': client_reference, diff --git a/tests/app/main/views/test_notifications.py b/tests/app/main/views/test_notifications.py index a0c133ea9..780ac0d55 100644 --- a/tests/app/main/views/test_notifications.py +++ b/tests/app/main/views/test_notifications.py @@ -157,6 +157,9 @@ def test_notification_page_shows_page_for_letter_notification( 'sample template sent by Test User on 1 January at 1:01am' ) assert normalize_spaces(page.select('main p:nth-of-type(2)')[0].text) == ( + 'Sent by second class' + ) + assert normalize_spaces(page.select('main p:nth-of-type(3)')[0].text) == ( 'Estimated delivery date: 6 January' ) assert page.select('p.notification-status') == []