mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-06 03:13:42 -05:00
123 lines
3.7 KiB
Python
123 lines
3.7 KiB
Python
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"]],
|
|
},
|
|
]
|
|
|
|
|
|
@pytest.mark.usefixtures("_mock_get_hierarchy_of_folders")
|
|
def test_template_list_yields_folders_visible_to_user(
|
|
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"],
|
|
)
|
|
|
|
|
|
@pytest.mark.usefixtures("_mock_get_hierarchy_of_folders")
|
|
def test_template_list_yields_all_folders_without_user(
|
|
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",
|
|
)
|