From 012b560673f6a8794d3a3315cb95c15450f7d568 Mon Sep 17 00:00:00 2001 From: Chris Hill-Scott Date: Mon, 12 Nov 2018 12:39:14 +0000 Subject: [PATCH] Make a helper function for generating templates MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Having this stuff split across various points in `conftest.py` was making my brain melt. This way I feel like it’s easier to see the input/output of the test. --- tests/app/main/views/test_template_folders.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/tests/app/main/views/test_template_folders.py b/tests/app/main/views/test_template_folders.py index c03de789f..f8d2e0771 100644 --- a/tests/app/main/views/test_template_folders.py +++ b/tests/app/main/views/test_template_folders.py @@ -17,6 +17,14 @@ def _folder(name, folder_id=None, parent=None): } +def _template(template_type, name): + return { + 'id': str(uuid.uuid4()), + 'name': name, + 'template_type': template_type, + } + + @pytest.mark.parametrize('parent_folder_id', [None, PARENT_FOLDER_ID]) def test_add_page_shows_option_for_folder( client_request, @@ -150,6 +158,14 @@ def test_should_show_templates_folder_page( _folder('folder_one_two', parent=PARENT_FOLDER_ID), _folder('folder_one_one', CHILD_FOLDER_ID, parent=PARENT_FOLDER_ID), ] + mock_get_service_templates.return_value = [ + _template('sms', 'sms_template_one'), + _template('sms', 'sms_template_two'), + _template('email', 'email_template_one'), + _template('email', 'email_template_two'), + _template('letter', 'letter_template_one'), + _template('letter', 'letter_template_two'), + ] service_one['permissions'] += ['letter', 'edit_folders']