import uuid from app.models.service import Service from app.models.user import User from tests import organisation_json 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( app_, 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( app_, 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, organisation_one): service = Service(service_one) mocker.patch('app.organisations_client.get_service_organisation', return_value=organisation_one) assert not organisation_one['organisation_type'] assert service.organisation_type == 'central' def test_organisation_type_when_service_and_its_org_both_have_an_org_type(mocker, service_one): # service_one has an organisation_type of 'central' service = Service(service_one) org = organisation_json(organisation_type='local') mocker.patch('app.organisations_client.get_service_organisation', return_value=org) assert service.organisation_type == 'local'