mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-08-18 21:49:37 -04:00
Merge pull request #4258 from alphagov/speed-up-templates-page-179736794
Optimise load time for service "Templates" page
This commit is contained in:
@@ -13,6 +13,7 @@ from tests.conftest import (
|
||||
create_active_caseworking_user,
|
||||
create_active_user_view_permissions,
|
||||
create_active_user_with_permissions,
|
||||
create_template,
|
||||
normalize_spaces,
|
||||
)
|
||||
|
||||
@@ -916,17 +917,17 @@ def test_manage_folder_users_doesnt_change_permissions_current_user_cannot_manag
|
||||
|
||||
|
||||
def test_delete_template_folder_should_request_confirmation(
|
||||
client_request, service_one, mock_get_template_folders, mocker,
|
||||
client_request,
|
||||
service_one,
|
||||
mock_get_template_folders,
|
||||
mocker,
|
||||
mock_get_service_templates_when_no_templates_exist,
|
||||
):
|
||||
mocker.patch('app.models.service.Service.active_users', [])
|
||||
folder_id = str(uuid.uuid4())
|
||||
mock_get_template_folders.side_effect = [[
|
||||
_folder('sacrifice', folder_id, None),
|
||||
], []]
|
||||
mocker.patch(
|
||||
'app.models.service.Service.get_templates',
|
||||
return_value=[],
|
||||
)
|
||||
page = client_request.get(
|
||||
'main.delete_template_folder', service_id=service_one['id'],
|
||||
template_folder_id=folder_id,
|
||||
@@ -954,17 +955,19 @@ def test_delete_template_folder_should_request_confirmation(
|
||||
|
||||
|
||||
def test_delete_template_folder_should_detect_non_empty_folder_on_get(
|
||||
client_request, service_one, mock_get_template_folders, mocker
|
||||
client_request,
|
||||
service_one,
|
||||
mock_get_template_folders,
|
||||
mocker
|
||||
):
|
||||
folder_id = str(uuid.uuid4())
|
||||
template_id = str(uuid.uuid4())
|
||||
mock_get_template_folders.side_effect = [
|
||||
[_folder("can't touch me", folder_id, None)],
|
||||
[]
|
||||
]
|
||||
mocker.patch(
|
||||
'app.models.service.Service.get_templates',
|
||||
return_value=[{'id': template_id, 'name': 'template'}],
|
||||
'app.service_api_client.get_service_templates',
|
||||
return_value={'data': [create_template(folder=folder_id)]}
|
||||
)
|
||||
client_request.get(
|
||||
'main.delete_template_folder', service_id=service_one['id'],
|
||||
@@ -983,16 +986,19 @@ def test_delete_template_folder_should_detect_non_empty_folder_on_get(
|
||||
None,
|
||||
PARENT_FOLDER_ID,
|
||||
))
|
||||
def test_delete_folder(client_request, service_one, mock_get_template_folders, mocker, parent_folder_id):
|
||||
def test_delete_folder(
|
||||
client_request,
|
||||
service_one,
|
||||
mock_get_template_folders,
|
||||
mocker,
|
||||
parent_folder_id,
|
||||
mock_get_service_templates_when_no_templates_exist,
|
||||
):
|
||||
mock_delete = mocker.patch('app.template_folder_api_client.delete_template_folder')
|
||||
folder_id = str(uuid.uuid4())
|
||||
mock_get_template_folders.side_effect = [[
|
||||
_folder('sacrifice', folder_id, parent_folder_id),
|
||||
], []]
|
||||
mocker.patch(
|
||||
'app.models.service.Service.get_templates',
|
||||
return_value=[],
|
||||
)
|
||||
|
||||
client_request.post(
|
||||
'main.delete_template_folder',
|
||||
|
||||
@@ -1,184 +1,10 @@
|
||||
import uuid
|
||||
|
||||
import pytest
|
||||
|
||||
from app.models.organisation import Organisation
|
||||
from app.models.service import Service
|
||||
from app.models.user import User
|
||||
from tests import organisation_json, service_json
|
||||
from tests.conftest import ORGANISATION_ID, create_folder, create_template
|
||||
|
||||
INV_PARENT_FOLDER_ID = '7e979e79-d970-43a5-ac69-b625a8d147b0'
|
||||
INV_CHILD_1_FOLDER_ID = '92ee1ee0-e4ee-4dcc-b1a7-a5da9ebcfa2b'
|
||||
VIS_PARENT_FOLDER_ID = 'bbbb222b-2b22-2b22-222b-b222b22b2222'
|
||||
INV_CHILD_2_FOLDER_ID = 'fafe723f-1d39-4a10-865f-e551e03d8886'
|
||||
|
||||
|
||||
def _get_all_folders(active_user_with_permissions):
|
||||
return [
|
||||
{
|
||||
'name': "Invisible folder",
|
||||
'id': str(uuid.uuid4()),
|
||||
'parent_id': None,
|
||||
'users_with_permission': []
|
||||
},
|
||||
{
|
||||
'name': "Parent 1 - invisible",
|
||||
'id': INV_PARENT_FOLDER_ID,
|
||||
'parent_id': None,
|
||||
'users_with_permission': []
|
||||
},
|
||||
{
|
||||
'name': "1's Visible child",
|
||||
'id': str(uuid.uuid4()),
|
||||
'parent_id': INV_PARENT_FOLDER_ID,
|
||||
'users_with_permission': [active_user_with_permissions['id']],
|
||||
},
|
||||
{
|
||||
'name': "1's Invisible child",
|
||||
'id': INV_CHILD_1_FOLDER_ID,
|
||||
'parent_id': INV_PARENT_FOLDER_ID,
|
||||
'users_with_permission': []
|
||||
},
|
||||
{
|
||||
'name': "1's Visible grandchild",
|
||||
'id': str(uuid.uuid4()),
|
||||
'parent_id': INV_CHILD_1_FOLDER_ID,
|
||||
'users_with_permission': [active_user_with_permissions['id']],
|
||||
},
|
||||
{
|
||||
'name': "Parent 2 - visible",
|
||||
'id': VIS_PARENT_FOLDER_ID,
|
||||
'parent_id': None,
|
||||
'users_with_permission': [active_user_with_permissions['id']],
|
||||
},
|
||||
{
|
||||
'name': "2's Visible child",
|
||||
'id': str(uuid.uuid4()),
|
||||
'parent_id': VIS_PARENT_FOLDER_ID,
|
||||
'users_with_permission': [active_user_with_permissions['id']],
|
||||
},
|
||||
{
|
||||
'name': "2's Invisible child",
|
||||
'id': INV_CHILD_2_FOLDER_ID,
|
||||
'parent_id': VIS_PARENT_FOLDER_ID,
|
||||
'users_with_permission': []
|
||||
},
|
||||
{
|
||||
'name': "2's Visible grandchild",
|
||||
'id': str(uuid.uuid4()),
|
||||
'parent_id': INV_CHILD_2_FOLDER_ID,
|
||||
'users_with_permission': [active_user_with_permissions['id']],
|
||||
},
|
||||
]
|
||||
|
||||
|
||||
def test_get_user_template_folders_only_returns_folders_visible_to_user(
|
||||
notify_admin,
|
||||
mock_get_template_folders,
|
||||
service_one,
|
||||
active_user_with_permissions,
|
||||
mocker
|
||||
):
|
||||
mock_get_template_folders.return_value = _get_all_folders(active_user_with_permissions)
|
||||
service = Service(service_one)
|
||||
result = service.get_user_template_folders(User(active_user_with_permissions))
|
||||
assert result == [
|
||||
{
|
||||
'name': ["Parent 1 - invisible", "1's Visible child"],
|
||||
'id': mocker.ANY,
|
||||
'parent_id': None,
|
||||
'users_with_permission': [active_user_with_permissions['id']],
|
||||
},
|
||||
{
|
||||
'name': ["Parent 1 - invisible", ["1's Invisible child", "1's Visible grandchild"]],
|
||||
'id': mocker.ANY,
|
||||
'parent_id': None,
|
||||
'users_with_permission': [active_user_with_permissions['id']],
|
||||
},
|
||||
{
|
||||
'name': "2's Visible child",
|
||||
'id': mocker.ANY,
|
||||
'parent_id': VIS_PARENT_FOLDER_ID,
|
||||
'users_with_permission': [active_user_with_permissions['id']],
|
||||
},
|
||||
{
|
||||
'name': ["2's Invisible child", "2's Visible grandchild"],
|
||||
'id': mocker.ANY,
|
||||
'parent_id': VIS_PARENT_FOLDER_ID,
|
||||
'users_with_permission': [active_user_with_permissions['id']],
|
||||
},
|
||||
{
|
||||
'name': "Parent 2 - visible",
|
||||
'id': VIS_PARENT_FOLDER_ID,
|
||||
'parent_id': None,
|
||||
'users_with_permission': [active_user_with_permissions['id']],
|
||||
},
|
||||
]
|
||||
|
||||
|
||||
def test_get_template_folders_shows_user_folders_when_user_id_passed_in(
|
||||
notify_admin,
|
||||
mock_get_template_folders,
|
||||
service_one,
|
||||
active_user_with_permissions,
|
||||
mocker
|
||||
):
|
||||
mock_get_template_folders.return_value = _get_all_folders(active_user_with_permissions)
|
||||
service = Service(service_one)
|
||||
result = service.get_template_folders(user=User(active_user_with_permissions))
|
||||
assert result == [
|
||||
{
|
||||
'name': ["Parent 1 - invisible", "1's Visible child"],
|
||||
'id': mocker.ANY,
|
||||
'parent_id': None,
|
||||
'users_with_permission': [active_user_with_permissions['id']]
|
||||
},
|
||||
{
|
||||
'name': ["Parent 1 - invisible", ["1's Invisible child", "1's Visible grandchild"]],
|
||||
'id': mocker.ANY,
|
||||
'parent_id': None,
|
||||
'users_with_permission': [active_user_with_permissions['id']]
|
||||
},
|
||||
{
|
||||
'name': "Parent 2 - visible",
|
||||
'id': VIS_PARENT_FOLDER_ID,
|
||||
'parent_id': None,
|
||||
'users_with_permission': [active_user_with_permissions['id']]
|
||||
},
|
||||
]
|
||||
|
||||
|
||||
def test_get_template_folders_shows_all_folders_when_user_id_not_passed_in(
|
||||
mock_get_template_folders,
|
||||
service_one,
|
||||
active_user_with_permissions,
|
||||
mocker
|
||||
):
|
||||
mock_get_template_folders.return_value = _get_all_folders(active_user_with_permissions)
|
||||
service = Service(service_one)
|
||||
result = service.get_template_folders()
|
||||
assert result == [
|
||||
{
|
||||
'name': "Invisible folder",
|
||||
'id': mocker.ANY,
|
||||
'parent_id': None,
|
||||
'users_with_permission': []
|
||||
},
|
||||
{
|
||||
'name': "Parent 1 - invisible",
|
||||
'id': INV_PARENT_FOLDER_ID,
|
||||
'parent_id': None,
|
||||
'users_with_permission': []
|
||||
},
|
||||
{
|
||||
'name': "Parent 2 - visible",
|
||||
'id': VIS_PARENT_FOLDER_ID,
|
||||
'parent_id': None,
|
||||
'users_with_permission': [active_user_with_permissions['id']],
|
||||
}
|
||||
]
|
||||
|
||||
|
||||
def test_organisation_type_when_services_organisation_has_no_org_type(mocker, service_one):
|
||||
service = Service(service_one)
|
||||
|
||||
125
tests/app/models/test_template_list.py
Normal file
125
tests/app/models/test_template_list.py
Normal file
@@ -0,0 +1,125 @@
|
||||
import uuid
|
||||
|
||||
import pytest
|
||||
|
||||
from app.models.service import Service
|
||||
from app.models.template_list import TemplateList
|
||||
from app.models.user import User
|
||||
|
||||
INV_PARENT_FOLDER_ID = '7e979e79-d970-43a5-ac69-b625a8d147b0'
|
||||
INV_CHILD_1_FOLDER_ID = '92ee1ee0-e4ee-4dcc-b1a7-a5da9ebcfa2b'
|
||||
VIS_PARENT_FOLDER_ID = 'bbbb222b-2b22-2b22-222b-b222b22b2222'
|
||||
INV_CHILD_2_FOLDER_ID = 'fafe723f-1d39-4a10-865f-e551e03d8886'
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mock_get_hierarchy_of_folders(
|
||||
mock_get_template_folders,
|
||||
active_user_with_permissions
|
||||
):
|
||||
mock_get_template_folders.return_value = [
|
||||
{
|
||||
'name': "Invisible folder",
|
||||
'id': str(uuid.uuid4()),
|
||||
'parent_id': None,
|
||||
'users_with_permission': []
|
||||
},
|
||||
{
|
||||
'name': "Parent 1 - invisible",
|
||||
'id': INV_PARENT_FOLDER_ID,
|
||||
'parent_id': None,
|
||||
'users_with_permission': []
|
||||
},
|
||||
{
|
||||
'name': "1's Visible child",
|
||||
'id': str(uuid.uuid4()),
|
||||
'parent_id': INV_PARENT_FOLDER_ID,
|
||||
'users_with_permission': [active_user_with_permissions['id']],
|
||||
},
|
||||
{
|
||||
'name': "1's Invisible child",
|
||||
'id': INV_CHILD_1_FOLDER_ID,
|
||||
'parent_id': INV_PARENT_FOLDER_ID,
|
||||
'users_with_permission': []
|
||||
},
|
||||
{
|
||||
'name': "1's Visible grandchild",
|
||||
'id': str(uuid.uuid4()),
|
||||
'parent_id': INV_CHILD_1_FOLDER_ID,
|
||||
'users_with_permission': [active_user_with_permissions['id']],
|
||||
},
|
||||
{
|
||||
'name': "Parent 2 - visible",
|
||||
'id': VIS_PARENT_FOLDER_ID,
|
||||
'parent_id': None,
|
||||
'users_with_permission': [active_user_with_permissions['id']],
|
||||
},
|
||||
{
|
||||
'name': "2's Visible child",
|
||||
'id': str(uuid.uuid4()),
|
||||
'parent_id': VIS_PARENT_FOLDER_ID,
|
||||
'users_with_permission': [active_user_with_permissions['id']],
|
||||
},
|
||||
{
|
||||
'name': "2's Invisible child",
|
||||
'id': INV_CHILD_2_FOLDER_ID,
|
||||
'parent_id': VIS_PARENT_FOLDER_ID,
|
||||
'users_with_permission': []
|
||||
},
|
||||
{
|
||||
'name': "2's Visible grandchild",
|
||||
'id': str(uuid.uuid4()),
|
||||
'parent_id': INV_CHILD_2_FOLDER_ID,
|
||||
'users_with_permission': [active_user_with_permissions['id']],
|
||||
},
|
||||
]
|
||||
|
||||
|
||||
def test_template_list_yields_folders_visible_to_user(
|
||||
mock_get_hierarchy_of_folders,
|
||||
mock_get_service_templates,
|
||||
service_one,
|
||||
active_user_with_permissions,
|
||||
):
|
||||
service = Service(service_one)
|
||||
user = User(active_user_with_permissions)
|
||||
|
||||
result_folder_names = tuple(
|
||||
result.name for result in
|
||||
TemplateList(service=service, user=user)
|
||||
if result.is_folder
|
||||
)
|
||||
|
||||
assert result_folder_names == (
|
||||
["Parent 1 - invisible", "1's Visible child"],
|
||||
["Parent 1 - invisible", ["1's Invisible child", "1's Visible grandchild"]],
|
||||
"Parent 2 - visible",
|
||||
"2's Visible child",
|
||||
["2's Invisible child", "2's Visible grandchild"],
|
||||
)
|
||||
|
||||
|
||||
def test_template_list_yields_all_folders_without_user(
|
||||
mock_get_hierarchy_of_folders,
|
||||
mock_get_service_templates,
|
||||
service_one,
|
||||
):
|
||||
service = Service(service_one)
|
||||
|
||||
result_folder_names = tuple(
|
||||
result.name for result in
|
||||
TemplateList(service=service)
|
||||
if result.is_folder
|
||||
)
|
||||
|
||||
assert result_folder_names == (
|
||||
"Invisible folder",
|
||||
"Parent 1 - invisible",
|
||||
"1's Invisible child",
|
||||
"1's Visible grandchild",
|
||||
"1's Visible child",
|
||||
"Parent 2 - visible",
|
||||
"2's Invisible child",
|
||||
"2's Visible grandchild",
|
||||
"2's Visible child",
|
||||
)
|
||||
Reference in New Issue
Block a user