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 from tests.conftest import ORGANISATION_ID 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): service = Service(service_one) service._dict['organisation_id'] = ORGANISATION_ID org = organisation_json(organisation_type=None) mocker.patch('app.organisations_client.get_organisation', return_value=org) assert not org['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) service._dict['organisation'] = ORGANISATION_ID org = organisation_json(organisation_type='local') mocker.patch('app.organisations_client.get_organisation', return_value=org) assert service.organisation_type == 'local' def test_organisation_name_comes_from_cache(mocker, service_one): mock_redis_get = mocker.patch( 'app.extensions.RedisClient.get', return_value=b'"Borchester Council"', ) mock_get_organisation = mocker.patch('app.organisations_client.get_organisation') service = Service(service_one) service._dict['organisation'] = ORGANISATION_ID assert service.organisation_name == 'Borchester Council' mock_redis_get.assert_called_once_with(f'organisation-{ORGANISATION_ID}-name') assert mock_get_organisation.called is False def test_organisation_name_goes_into_cache(mocker, service_one): mocker.patch( 'app.extensions.RedisClient.get', return_value=None, ) mock_redis_set = mocker.patch( 'app.extensions.RedisClient.set', ) mocker.patch( 'app.organisations_client.get_organisation', return_value=organisation_json(), ) service = Service(service_one) service._dict['organisation'] = ORGANISATION_ID assert service.organisation_name == 'Test Organisation' mock_redis_set.assert_called_once_with( f'organisation-{ORGANISATION_ID}-name', '"Test Organisation"', ex=604800, ) def test_service_without_organisation_doesnt_need_org_api(mocker, service_one): mock_redis_get = mocker.patch('app.extensions.RedisClient.get') mock_get_organisation = mocker.patch('app.organisations_client.get_organisation') service = Service(service_one) service._dict['organisation'] = None assert service.organisation_id is None assert service.organisation_name is None assert isinstance(service.organisation, Organisation) assert mock_redis_get.called is False assert mock_get_organisation.called is False def test_bad_permission_raises(service_one): with pytest.raises(KeyError) as e: Service(service_one).has_permission('foo') assert str(e.value) == "'foo is not a service permission'"