diff --git a/app/utils.py b/app/utils.py index f8e5491f6..b4a92d5a8 100644 --- a/app/utils.py +++ b/app/utils.py @@ -157,7 +157,9 @@ def get_sample_template(template_type): if template_type == 'sms': return SMSPreviewTemplate({'content': 'any', 'template_type': 'sms'}) if template_type == 'letter': - return LetterImageTemplate({'content': 'any', 'subject': '', 'template_type': 'letter'}) + return LetterImageTemplate( + {'content': 'any', 'subject': '', 'template_type': 'letter'}, postage='second', image_url='x', page_count=1 + ) def generate_notifications_csv(**kwargs): diff --git a/tests/app/test_utils.py b/tests/app/test_utils.py index 6fae70a20..061282eff 100644 --- a/tests/app/test_utils.py +++ b/tests/app/test_utils.py @@ -7,6 +7,7 @@ import pytest from bs4 import BeautifulSoup from flask import url_for from freezegun import freeze_time +from notifications_utils.template import Template from app import format_datetime_relative from app.utils import ( @@ -18,6 +19,7 @@ from app.utils import ( get_letter_printing_statement, get_letter_validation_error, get_logo_cdn_domain, + get_sample_template, is_less_than_90_days_ago, printing_today_or_tomorrow, ) @@ -576,3 +578,9 @@ def test_get_letter_validation_error_for_known_errors( @freeze_time('2020-02-14T12:00:00') def test_is_less_than_90_days_ago(date_from_db, expected_result): assert is_less_than_90_days_ago(date_from_db) == expected_result + + +@pytest.mark.parametrize("template_type", ["sms", "letter", "email"]) +def test_get_sample_template_returns_template(template_type): + template = get_sample_template(template_type) + assert isinstance(template, Template)