Skip to content

Commit 5b52aa2

Browse files
author
Ben Thorner
committed
Simplify test setup for Template List model
1 parent 5db2581 commit 5b52aa2

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

tests/app/models/test_template_list.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
import uuid
22

3+
import pytest
4+
35
from app.models.service import Service
46
from app.models.template_list import TemplateList
57
from app.models.user import User
@@ -10,8 +12,12 @@
1012
INV_CHILD_2_FOLDER_ID = 'fafe723f-1d39-4a10-865f-e551e03d8886'
1113

1214

13-
def _get_all_folders(active_user_with_permissions):
14-
return [
15+
@pytest.fixture
16+
def mock_get_hierarchy_of_folders(
17+
mock_get_template_folders,
18+
active_user_with_permissions
19+
):
20+
mock_get_template_folders.return_value = [
1521
{
1622
'name': "Invisible folder",
1723
'id': str(uuid.uuid4()),
@@ -70,14 +76,11 @@ def _get_all_folders(active_user_with_permissions):
7076

7177

7278
def test_template_list_yields_folders_visible_to_user(
73-
notify_admin,
74-
mock_get_template_folders,
79+
mock_get_hierarchy_of_folders,
7580
mock_get_service_templates,
7681
service_one,
7782
active_user_with_permissions,
78-
mocker
7983
):
80-
mock_get_template_folders.return_value = _get_all_folders(active_user_with_permissions)
8184
service = Service(service_one)
8285
user = User(active_user_with_permissions)
8386

@@ -97,13 +100,10 @@ def test_template_list_yields_folders_visible_to_user(
97100

98101

99102
def test_template_list_yields_all_folders_without_user(
100-
mock_get_template_folders,
103+
mock_get_hierarchy_of_folders,
101104
mock_get_service_templates,
102105
service_one,
103-
active_user_with_permissions,
104-
mocker
105106
):
106-
mock_get_template_folders.return_value = _get_all_folders(active_user_with_permissions)
107107
service = Service(service_one)
108108

109109
result_folder_names = tuple(

0 commit comments

Comments
 (0)