diff --git a/.checkov.baseline b/.checkov.baseline index 19875f2e3..796f224f6 100644 --- a/.checkov.baseline +++ b/.checkov.baseline @@ -613,6 +613,25 @@ } ] }, + { + "file": "/checkov_pipeline_synth.json", + "findings": [ + { + "resource": "AWS::IAM::Role.PipelineRoleDCFDBB91", + "check_ids": [ + "CKV_AWS_107", + "CKV_AWS_108", + "CKV_AWS_111" + ] + }, + { + "resource": "AWS::S3::Bucket.thistableartifactsbucketDB1C8C64", + "check_ids": [ + "CKV_AWS_18" + ] + } + ] + }, { "file": "/frontend/docker/prod/Dockerfile", "findings": [ diff --git a/backend/dataall/core/environment/api/resolvers.py b/backend/dataall/core/environment/api/resolvers.py index 2f3301c74..01a3c1bde 100644 --- a/backend/dataall/core/environment/api/resolvers.py +++ b/backend/dataall/core/environment/api/resolvers.py @@ -14,7 +14,6 @@ from dataall.core.organizations.api.resolvers import Context, exceptions, get_organization_simplified - log = logging.getLogger() @@ -223,6 +222,7 @@ def generate_environment_access_token(context, source, environmentUri: str = Non def get_environment_stack(context: Context, source: Environment, **kwargs): return StackService.resolve_parent_obj_stack( targetUri=source.environmentUri, + targetType='environment', environmentUri=source.environmentUri, ) @@ -275,8 +275,7 @@ def resolve_environment(context, source, **kwargs): """Resolves the environment for a environmental resource""" if not source: return None - with context.engine.scoped_session() as session: - return EnvironmentService.get_environment_by_uri(session, source.environmentUri) + return EnvironmentService.find_environment_by_uri(uri=source.environmentUri) def resolve_parameters(context, source: Environment, **kwargs): diff --git a/backend/dataall/core/organizations/services/organization_service.py b/backend/dataall/core/organizations/services/organization_service.py index 696ae0881..cbc81a1ce 100644 --- a/backend/dataall/core/organizations/services/organization_service.py +++ b/backend/dataall/core/organizations/services/organization_service.py @@ -305,7 +305,7 @@ def resolve_organization_by_env(uri): context = get_context() with context.db_engine.scoped_session() as session: env = EnvironmentRepository.get_environment_by_uri(session, uri) - return OrganizationRepository.find_organization_by_uri(session, env.organizationUri) + return OrganizationService.get_organization(uri=env.organizationUri) @staticmethod @ResourcePolicyService.has_resource_permission(GET_ORGANIZATION) diff --git a/backend/dataall/core/stacks/services/stack_service.py b/backend/dataall/core/stacks/services/stack_service.py index d02d9ba48..46bacd555 100644 --- a/backend/dataall/core/stacks/services/stack_service.py +++ b/backend/dataall/core/stacks/services/stack_service.py @@ -65,9 +65,16 @@ def map_target_type_to_log_config_path(**kwargs): class StackService: @staticmethod - def resolve_parent_obj_stack(targetUri: str, environmentUri: str): + def resolve_parent_obj_stack(targetUri: str, targetType: str, environmentUri: str): context = get_context() with context.db_engine.scoped_session() as session: + ResourcePolicyService.check_user_resource_permission( + session=session, + username=context.username, + groups=context.groups, + resource_uri=targetUri, + permission_name=TargetType.get_resource_read_permission_name(targetType), + ) env: Environment = EnvironmentRepository.get_environment_by_uri(session, environmentUri) stack: Stack = StackRepository.find_stack_by_target_uri(session, target_uri=targetUri) if not stack: diff --git a/backend/dataall/core/vpc/services/vpc_service.py b/backend/dataall/core/vpc/services/vpc_service.py index 48c9c726c..fbdf8f092 100644 --- a/backend/dataall/core/vpc/services/vpc_service.py +++ b/backend/dataall/core/vpc/services/vpc_service.py @@ -1,5 +1,8 @@ +import logging + from dataall.base.context import get_context from dataall.base.db import exceptions +from dataall.base.db.exceptions import ResourceUnauthorized from dataall.core.permissions.services.group_policy_service import GroupPolicyService from dataall.core.environment.db.environment_repositories import EnvironmentRepository from dataall.core.activity.db.activity_models import Activity @@ -7,10 +10,12 @@ from dataall.core.permissions.services.tenant_policy_service import TenantPolicyService from dataall.core.vpc.db.vpc_repositories import VpcRepository from dataall.core.vpc.db.vpc_models import Vpc -from dataall.core.permissions.services.network_permissions import NETWORK_ALL, DELETE_NETWORK +from dataall.core.permissions.services.network_permissions import NETWORK_ALL, DELETE_NETWORK, GET_NETWORK from dataall.core.permissions.services.environment_permissions import CREATE_NETWORK from dataall.core.permissions.services.tenant_permissions import MANAGE_ENVIRONMENTS +log = logging.getLogger(__name__) + def _session(): return get_context().db_engine.scoped_session() @@ -90,4 +95,19 @@ def delete_network(uri): @staticmethod def get_environment_networks(environment_uri): with _session() as session: - return VpcRepository.get_environment_networks(session=session, environment_uri=environment_uri) + nets = [] + all_nets = VpcRepository.get_environment_networks(session=session, environment_uri=environment_uri) + for net in all_nets: + try: + ResourcePolicyService.check_user_resource_permission( + session=session, + username=get_context().username, + groups=get_context().groups, + resource_uri=net.vpcUri, + permission_name=GET_NETWORK, + ) + except ResourceUnauthorized as exc: + log.info(exc) + else: + nets += net + return nets diff --git a/backend/dataall/modules/datapipelines/api/resolvers.py b/backend/dataall/modules/datapipelines/api/resolvers.py index 3c2bde886..47e676816 100644 --- a/backend/dataall/modules/datapipelines/api/resolvers.py +++ b/backend/dataall/modules/datapipelines/api/resolvers.py @@ -105,5 +105,6 @@ def resolve_stack(context, source: DataPipeline, **kwargs): return None return StackService.resolve_parent_obj_stack( targetUri=source.DataPipelineUri, + targetType='pipeline', environmentUri=source.environmentUri, ) diff --git a/backend/dataall/modules/datapipelines/services/datapipelines_service.py b/backend/dataall/modules/datapipelines/services/datapipelines_service.py index de277d20d..08dc46986 100644 --- a/backend/dataall/modules/datapipelines/services/datapipelines_service.py +++ b/backend/dataall/modules/datapipelines/services/datapipelines_service.py @@ -255,6 +255,7 @@ def _delete_repository(target_uri, accountid, cdk_role_arn, region, repo_name): @staticmethod @TenantPolicyService.has_tenant_permission(MANAGE_PIPELINES) + @ResourcePolicyService.has_resource_permission(UPDATE_PIPELINE, param_name='envPipelineUri') def delete_pipeline_environment(envPipelineUri: str): with _session() as session: DatapipelinesRepository.delete_pipeline_environment(session=session, envPipelineUri=envPipelineUri) diff --git a/backend/dataall/modules/datasets_base/api/resolvers.py b/backend/dataall/modules/datasets_base/api/resolvers.py index 73f6539c7..017256ae6 100644 --- a/backend/dataall/modules/datasets_base/api/resolvers.py +++ b/backend/dataall/modules/datasets_base/api/resolvers.py @@ -58,8 +58,7 @@ def get_dataset_organization(context, source: DatasetBase, **kwargs): def get_dataset_environment(context, source: DatasetBase, **kwargs): if not source: return None - with context.engine.scoped_session() as session: - return EnvironmentService.get_environment_by_uri(session, source.environmentUri) + return EnvironmentService.find_environment_by_uri(uri=source.environmentUri) def get_dataset_owners_group(context, source: DatasetBase, **kwargs): @@ -79,5 +78,6 @@ def resolve_dataset_stack(context: Context, source: DatasetBase, **kwargs): return None return StackService.resolve_parent_obj_stack( targetUri=source.datasetUri, + targetType='dataset', environmentUri=source.environmentUri, ) diff --git a/backend/dataall/modules/mlstudio/api/resolvers.py b/backend/dataall/modules/mlstudio/api/resolvers.py index e38d72ae6..6d4e07f8b 100644 --- a/backend/dataall/modules/mlstudio/api/resolvers.py +++ b/backend/dataall/modules/mlstudio/api/resolvers.py @@ -122,6 +122,7 @@ def resolve_sagemaker_studio_user_stack(context: Context, source: SagemakerStudi return None return StackService.resolve_parent_obj_stack( targetUri=source.sagemakerStudioUserUri, + targetType='mlstudio', environmentUri=source.environmentUri, ) diff --git a/backend/dataall/modules/notebooks/api/resolvers.py b/backend/dataall/modules/notebooks/api/resolvers.py index de6235305..e7e111fc3 100644 --- a/backend/dataall/modules/notebooks/api/resolvers.py +++ b/backend/dataall/modules/notebooks/api/resolvers.py @@ -90,6 +90,7 @@ def resolve_notebook_stack(context: Context, source: SagemakerNotebook, **kwargs return None return StackService.resolve_parent_obj_stack( targetUri=source.notebookUri, + targetType='notebook', environmentUri=source.environmentUri, ) diff --git a/backend/dataall/modules/redshift_datasets/api/datasets/resolvers.py b/backend/dataall/modules/redshift_datasets/api/datasets/resolvers.py index e95480351..aca1b2e79 100644 --- a/backend/dataall/modules/redshift_datasets/api/datasets/resolvers.py +++ b/backend/dataall/modules/redshift_datasets/api/datasets/resolvers.py @@ -87,8 +87,7 @@ def resolve_dataset_environment( ): # TODO- duplicated with S3 datasets - follow-up PR if not source: return None - with context.engine.scoped_session() as session: - return EnvironmentService.get_environment_by_uri(session, source.environmentUri) + return EnvironmentService.find_environment_by_uri(uri=source.environmentUri) def resolve_dataset_owners_group( diff --git a/backend/dataall/modules/s3_datasets/api/dataset/resolvers.py b/backend/dataall/modules/s3_datasets/api/dataset/resolvers.py index 11a183f2b..ef8acdbb7 100644 --- a/backend/dataall/modules/s3_datasets/api/dataset/resolvers.py +++ b/backend/dataall/modules/s3_datasets/api/dataset/resolvers.py @@ -85,8 +85,7 @@ def get_dataset_organization(context, source: S3Dataset, **kwargs): def get_dataset_environment(context, source: S3Dataset, **kwargs): if not source: return None - with context.engine.scoped_session() as session: - return EnvironmentService.get_environment_by_uri(session, source.environmentUri) + return EnvironmentService.find_environment_by_uri(uri=source.environmentUri) def get_dataset_owners_group(context, source: S3Dataset, **kwargs): @@ -133,6 +132,7 @@ def resolve_dataset_stack(context: Context, source: S3Dataset, **kwargs): return None return StackService.resolve_parent_obj_stack( targetUri=source.datasetUri, + targetType='dataset', environmentUri=source.environmentUri, ) diff --git a/backend/dataall/modules/s3_datasets/api/storage_location/resolvers.py b/backend/dataall/modules/s3_datasets/api/storage_location/resolvers.py index 3d6847029..39aa2d8cf 100644 --- a/backend/dataall/modules/s3_datasets/api/storage_location/resolvers.py +++ b/backend/dataall/modules/s3_datasets/api/storage_location/resolvers.py @@ -1,9 +1,10 @@ from dataall.base.api.context import Context -from dataall.modules.catalog.db.glossary_repositories import GlossaryRepository from dataall.base.db.exceptions import RequiredParameter from dataall.base.feature_toggle_checker import is_feature_enabled +from dataall.modules.catalog.db.glossary_repositories import GlossaryRepository +from dataall.modules.s3_datasets.db.dataset_models import DatasetStorageLocation from dataall.modules.s3_datasets.services.dataset_location_service import DatasetLocationService -from dataall.modules.s3_datasets.db.dataset_models import DatasetStorageLocation, S3Dataset +from dataall.modules.s3_datasets.services.dataset_service import DatasetService def _validate_input(input: dict): @@ -46,9 +47,7 @@ def remove_storage_location(context, source, locationUri: str = None): def resolve_dataset(context, source: DatasetStorageLocation, **kwargs): if not source: return None - with context.engine.scoped_session() as session: - d = session.query(S3Dataset).get(source.datasetUri) - return d + return DatasetService.find_dataset(uri=source.datasetUri) def resolve_glossary_terms(context: Context, source: DatasetStorageLocation, **kwargs): diff --git a/backend/dataall/modules/s3_datasets/services/dataset_service.py b/backend/dataall/modules/s3_datasets/services/dataset_service.py index a279c97db..6d3010bf4 100644 --- a/backend/dataall/modules/s3_datasets/services/dataset_service.py +++ b/backend/dataall/modules/s3_datasets/services/dataset_service.py @@ -38,6 +38,7 @@ DATASET_ALL, DATASET_READ, IMPORT_DATASET, + GET_DATASET, DATASET_TABLE_ALL, ) from dataall.modules.datasets_base.services.dataset_list_permissions import LIST_ENVIRONMENT_DATASETS @@ -242,6 +243,11 @@ def get_dataset(uri): dataset.userRoleForDataset = DatasetRole.Admin.value return dataset + @classmethod + @ResourcePolicyService.has_resource_permission(GET_DATASET) + def find_dataset(cls, uri): + return DatasetService.get_dataset(uri) + @staticmethod @TenantPolicyService.has_tenant_permission(MANAGE_DATASETS) @ResourcePolicyService.has_resource_permission(CREDENTIALS_DATASET) diff --git a/frontend/src/modules/DatasetsBase/views/DatasetList.js b/frontend/src/modules/DatasetsBase/views/DatasetList.js index b6835a547..f3611a239 100644 --- a/frontend/src/modules/DatasetsBase/views/DatasetList.js +++ b/frontend/src/modules/DatasetsBase/views/DatasetList.js @@ -94,10 +94,12 @@ const DatasetList = () => { const fetchItems = useCallback(async () => { setLoading(true); const response = await client.query(listDatasets({ filter })); - if (!response.errors) { + if (response.data.listDatasets !== null) { setItems(response.data.listDatasets); } else { - dispatch({ type: SET_ERROR, error: response.errors[0].message }); + response.errors.forEach((err) => + dispatch({ type: SET_ERROR, error: err.message }) + ); } setLoading(false); }, [client, dispatch, filter]); diff --git a/frontend/src/modules/Folders/components/FolderOverview.js b/frontend/src/modules/Folders/components/FolderOverview.js index 98dcfc84b..c1b6c469c 100644 --- a/frontend/src/modules/Folders/components/FolderOverview.js +++ b/frontend/src/modules/Folders/components/FolderOverview.js @@ -27,11 +27,11 @@ export const FolderOverview = (props) => { diff --git a/frontend/src/modules/Folders/components/FolderS3Properties.js b/frontend/src/modules/Folders/components/FolderS3Properties.js index 5118896f5..1447d0fe8 100644 --- a/frontend/src/modules/Folders/components/FolderS3Properties.js +++ b/frontend/src/modules/Folders/components/FolderS3Properties.js @@ -9,7 +9,7 @@ import { export const FolderS3Properties = (props) => { const { folder } = props; - + if (folder.dataset === null) return null; return ( diff --git a/frontend/src/modules/Folders/views/FolderView.js b/frontend/src/modules/Folders/views/FolderView.js index 184b258bf..7b70a7e70 100644 --- a/frontend/src/modules/Folders/views/FolderView.js +++ b/frontend/src/modules/Folders/views/FolderView.js @@ -233,17 +233,18 @@ const FolderView = () => { const fetchItem = useCallback(async () => { setLoading(true); const response = await client.query(getDatasetStorageLocation(params.uri)); - if (!response.errors && response.data.getDatasetStorageLocation !== null) { + if (response.data.getDatasetStorageLocation !== null) { setFolder(response.data.getDatasetStorageLocation); setIsAdmin( ['Creator', 'Admin', 'Owner'].indexOf( - response.data.getDatasetStorageLocation.dataset.userRoleForDataset + response.data.getDatasetStorageLocation.dataset?.userRoleForDataset ) !== -1 ); } else { setFolder(null); - const error = response.errors[0].message; - dispatch({ type: SET_ERROR, error }); + response.errors.forEach((err) => + dispatch({ type: SET_ERROR, error: err.message }) + ); } setLoading(false); }, [client, dispatch, params.uri]); diff --git a/frontend/src/modules/S3_Datasets/components/DatasetOverview.js b/frontend/src/modules/S3_Datasets/components/DatasetOverview.js index 351e8a3ce..0599d33b2 100644 --- a/frontend/src/modules/S3_Datasets/components/DatasetOverview.js +++ b/frontend/src/modules/S3_Datasets/components/DatasetOverview.js @@ -26,7 +26,7 @@ export const DatasetOverview = (props) => { { const fetchItem = useCallback(async () => { setLoading(true); const response = await client.query(getDataset(params.uri)); - if (!response.errors && response.data.getDataset !== null) { + if (response.data.getDataset !== null) { setDataset(response.data.getDataset); setIsAdmin( ['BusinessOwner', 'Admin', 'DataSteward', 'Creator'].indexOf( diff --git a/tests/conftest.py b/tests/conftest.py index 99a17b4c4..39c7e7664 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -1,20 +1,23 @@ import os from dataclasses import dataclass +from glob import glob from unittest.mock import MagicMock import pytest from starlette.testclient import TestClient +from dataall.base.config import config from dataall.base.db import get_engine, create_schema_and_tables, Engine from dataall.base.loader import load_modules, ImportMode, list_loaded_modules -from glob import glob - from dataall.core.groups.db.group_models import Group from dataall.core.permissions.services.permission_service import PermissionService -from dataall.core.permissions.services.tenant_policy_service import TenantPolicyService from dataall.core.permissions.services.tenant_permissions import TENANT_ALL +from dataall.core.permissions.services.tenant_policy_service import TenantPolicyService from tests.client import create_app, ClientWrapper +for module in config.get_property('modules'): + config.set_property(f'modules.{module}.active', True) + load_modules(modes=ImportMode.all()) ENVNAME = os.environ.get('envname', 'pytest') diff --git a/tests/test_permissions.py b/tests/test_permissions.py new file mode 100644 index 000000000..80c664389 --- /dev/null +++ b/tests/test_permissions.py @@ -0,0 +1,1241 @@ +import inspect +import logging +from contextlib import suppress +from dataclasses import dataclass +from enum import Enum +from typing import Mapping, Tuple +from unittest.mock import MagicMock, patch, ANY + +import pytest +from assertpy import assert_that + +from dataall.base.api import bootstrap +from dataall.base.context import RequestContext +from dataall.core.permissions.services.environment_permissions import ( + GET_ENVIRONMENT, + LIST_ENVIRONMENT_CONSUMPTION_ROLES, + LIST_ENVIRONMENT_NETWORKS, + CREDENTIALS_ENVIRONMENT, + LIST_ENVIRONMENT_GROUPS, + LIST_ENVIRONMENT_GROUP_PERMISSIONS, + ADD_ENVIRONMENT_CONSUMPTION_ROLES, + CREATE_NETWORK, + ENABLE_ENVIRONMENT_SUBSCRIPTIONS, + DELETE_ENVIRONMENT, + INVITE_ENVIRONMENT_GROUP, + REMOVE_ENVIRONMENT_CONSUMPTION_ROLE, + REMOVE_ENVIRONMENT_GROUP, + UPDATE_ENVIRONMENT, + UPDATE_ENVIRONMENT_GROUP, +) +from dataall.core.permissions.services.network_permissions import GET_NETWORK, DELETE_NETWORK +from dataall.core.permissions.services.organization_permissions import ( + GET_ORGANIZATION, + UPDATE_ORGANIZATION, + DELETE_ORGANIZATION, + INVITE_ORGANIZATION_GROUP, + LINK_ENVIRONMENT, + REMOVE_ORGANIZATION_GROUP, +) +from dataall.core.permissions.services.tenant_permissions import MANAGE_ENVIRONMENTS, MANAGE_ORGANIZATIONS +from dataall.modules.catalog.services.glossaries_permissions import MANAGE_GLOSSARIES +from dataall.modules.dashboards.services.dashboard_permissions import ( + MANAGE_DASHBOARDS, + SHARE_DASHBOARD, + GET_DASHBOARD, + CREATE_DASHBOARD, + DELETE_DASHBOARD, + UPDATE_DASHBOARD, +) +from dataall.modules.datapipelines.services.datapipelines_permissions import ( + GET_PIPELINE, + MANAGE_PIPELINES, + CREDENTIALS_PIPELINE, + CREATE_PIPELINE, + DELETE_PIPELINE, + UPDATE_PIPELINE, +) +from dataall.modules.datasets_base.services.dataset_list_permissions import LIST_ENVIRONMENT_DATASETS +from dataall.modules.metadata_forms.services.metadata_form_permissions import ( + MANAGE_METADATA_FORMS, + ATTACH_METADATA_FORM, +) +from dataall.modules.mlstudio.services.mlstudio_permissions import ( + GET_SGMSTUDIO_USER, + MANAGE_SGMSTUDIO_USERS, + SGMSTUDIO_USER_URL, + DELETE_SGMSTUDIO_USER, + CREATE_SGMSTUDIO_USER, +) +from dataall.modules.notebooks.services.notebook_permissions import ( + GET_NOTEBOOK, + MANAGE_NOTEBOOKS, + DELETE_NOTEBOOK, + CREATE_NOTEBOOK, +) +from dataall.modules.omics.services.omics_permissions import MANAGE_OMICS_RUNS, CREATE_OMICS_RUN +from dataall.modules.redshift_datasets.services.redshift_connection_permissions import GET_REDSHIFT_CONNECTION +from dataall.modules.redshift_datasets.services.redshift_dataset_permissions import ( + MANAGE_REDSHIFT_DATASETS, + ADD_TABLES_REDSHIFT_DATASET, + DELETE_REDSHIFT_DATASET, + DELETE_REDSHIFT_DATASET_TABLE, + IMPORT_REDSHIFT_DATASET, + UPDATE_REDSHIFT_DATASET_TABLE, + UPDATE_REDSHIFT_DATASET, + GET_REDSHIFT_DATASET, + GET_REDSHIFT_DATASET_TABLE, +) +from dataall.modules.s3_datasets.services.dataset_permissions import ( + GET_DATASET, + GET_DATASET_TABLE, + MANAGE_DATASETS, + GET_DATASET_FOLDER, + CREDENTIALS_DATASET, + CREATE_TABLE_DATA_FILTER, + DELETE_DATASET_TABLE, + DELETE_DATASET, + DELETE_DATASET_FOLDER, + DELETE_TABLE_DATA_FILTER, + UPDATE_DATASET, + PROFILE_DATASET_TABLE, + CRAWL_DATASET, + UPDATE_DATASET_TABLE, + SYNC_DATASET, + UPDATE_DATASET_FOLDER, + LIST_TABLE_DATA_FILTERS, + CREATE_DATASET, + CREATE_DATASET_FOLDER, +) +from dataall.modules.shares_base.services.share_permissions import ( + MANAGE_SHARES, + GET_SHARE_OBJECT, + LIST_ENVIRONMENT_SHARED_WITH_OBJECTS, + ADD_ITEM, + APPROVE_SHARE_OBJECT, + SUBMIT_SHARE_OBJECT, + DELETE_SHARE_OBJECT, + REJECT_SHARE_OBJECT, + REMOVE_ITEM, + CREATE_SHARE_OBJECT, +) +from dataall.modules.worksheets.services.worksheet_permissions import ( + MANAGE_WORKSHEETS, + GET_WORKSHEET, + RUN_ATHENA_QUERY, + DELETE_WORKSHEET, + UPDATE_WORKSHEET, +) + + +class IgnoreReason(Enum): + TENANT = 'tenant action, no need for tenant permission check' + APPSUPPORT = 'permissions do not apply to application support features' + BACKPORT = 'outside of this PR to be able to backport to v2.6.2' + INTRAMODULE = 'returns intra-module data' + USERROLEINRESOURCE = 'checks user permissions for a particular feature' + PUBLIC = 'public by design' + SIMPLIFIED = 'returns a simplified response' + USERLIMITED = 'returns user resources in application' + CUSTOM = 'custom permissions checks' + NOTREQUIRED = 'permission check is not required' + + +def field_id(type_name: str, field_name: str) -> str: + return f'{type_name}_{field_name}' + + +@dataclass +class TestData: + resource_ignore: IgnoreReason = None + resource_perm: str = None + tenant_ignore: IgnoreReason = None + tenant_perm: str = None + tenant_admin_ignore: IgnoreReason = IgnoreReason.NOTREQUIRED + tenant_admin_perm: bool = False + + def get_perm(self, _type: str) -> str: + return getattr(self, f'{_type}_perm') + + def get_ignore(self, _type) -> IgnoreReason: + return getattr(self, f'{_type}_ignore') + + def get(self, _type) -> Tuple[str, IgnoreReason]: + return self.get_perm(_type), self.get_ignore(_type) + + def __post_init__(self): + if not bool(self.resource_perm) ^ bool(self.resource_ignore): + raise ValueError('Either resource_perm or resource_ignore must be set') + if not bool(self.tenant_perm) ^ bool(self.tenant_ignore): + raise ValueError('Either tenant_perm or tenant_ignore must be set') + + +EXPECTED_RESOLVERS: Mapping[str, TestData] = { + field_id('AttachedMetadataForm', 'entityName'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('AttachedMetadataForm', 'fields'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('AttachedMetadataForm', 'metadataForm'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('AttachedMetadataFormField', 'field'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('AttachedMetadataFormField', 'hasTenantPermissions'): TestData( + resource_ignore=IgnoreReason.USERROLEINRESOURCE, tenant_ignore=IgnoreReason.TENANT, tenant_admin_perm=True + ), + field_id('Category', 'associations'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Category', 'categories'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Category', 'children'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Category', 'stats'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Category', 'terms'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('ConsumptionRole', 'managedPolicies'): TestData( + resource_perm=GET_ENVIRONMENT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Dashboard', 'environment'): TestData( + resource_perm=GET_ENVIRONMENT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Dashboard', 'terms'): TestData( + resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Dashboard', 'upvotes'): TestData( + resource_ignore=IgnoreReason.APPSUPPORT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Dashboard', 'userRoleForDashboard'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('DataPipeline', 'cloneUrlHttp'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('DataPipeline', 'developmentEnvironments'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('DataPipeline', 'environment'): TestData( + resource_perm=GET_ENVIRONMENT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('DataPipeline', 'organization'): TestData( + resource_perm=GET_ORGANIZATION, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('DataPipeline', 'stack'): TestData(resource_perm=GET_PIPELINE, tenant_ignore=IgnoreReason.NOTREQUIRED), + field_id('DataPipeline', 'userRoleForPipeline'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Dataset', 'environment'): TestData(resource_perm=GET_ENVIRONMENT, tenant_ignore=IgnoreReason.NOTREQUIRED), + field_id('Dataset', 'locations'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Dataset', 'owners'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Dataset', 'stack'): TestData(resource_perm=GET_DATASET, tenant_ignore=IgnoreReason.NOTREQUIRED), + field_id('Dataset', 'statistics'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Dataset', 'stewards'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Dataset', 'tables'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Dataset', 'terms'): TestData(resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.NOTREQUIRED), + field_id('Dataset', 'userRoleForDataset'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('DatasetBase', 'environment'): TestData( + resource_perm=GET_ENVIRONMENT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('DatasetBase', 'owners'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('DatasetBase', 'stack'): TestData(resource_perm=GET_DATASET, tenant_ignore=IgnoreReason.NOTREQUIRED), + field_id('DatasetBase', 'stewards'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('DatasetBase', 'userRoleForDataset'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('DatasetProfilingRun', 'dataset'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('DatasetProfilingRun', 'results'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('DatasetProfilingRun', 'status'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('DatasetStorageLocation', 'dataset'): TestData( + resource_perm=GET_DATASET, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('DatasetStorageLocation', 'terms'): TestData( + resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('DatasetTable', 'GlueTableProperties'): TestData( + resource_perm=GET_DATASET_TABLE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('DatasetTable', 'columns'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('DatasetTable', 'dataset'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('DatasetTable', 'terms'): TestData( + resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('DatasetTableColumn', 'terms'): TestData( + resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Environment', 'networks'): TestData(resource_perm=GET_NETWORK, tenant_ignore=IgnoreReason.NOTREQUIRED), + field_id('Environment', 'organization'): TestData( + resource_ignore=IgnoreReason.SIMPLIFIED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Environment', 'parameters'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Environment', 'stack'): TestData(resource_perm=GET_ENVIRONMENT, tenant_ignore=IgnoreReason.NOTREQUIRED), + field_id('Environment', 'userRoleInEnvironment'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('EnvironmentSimplified', 'networks'): TestData( + resource_perm=GET_NETWORK, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('EnvironmentSimplified', 'organization'): TestData( + resource_ignore=IgnoreReason.SIMPLIFIED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Feed', 'messages'): TestData( + resource_ignore=IgnoreReason.APPSUPPORT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Glossary', 'associations'): TestData( + resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Glossary', 'categories'): TestData( + resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Glossary', 'children'): TestData( + resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Glossary', 'stats'): TestData( + resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Glossary', 'tree'): TestData(resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.NOTREQUIRED), + field_id('Glossary', 'userRoleForGlossary'): TestData( + resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('GlossaryTermLink', 'target'): TestData( + resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('GlossaryTermLink', 'term'): TestData( + resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Group', 'environmentPermissions'): TestData( + resource_perm=LIST_ENVIRONMENT_GROUP_PERMISSIONS, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Group', 'tenantPermissions'): TestData( + resource_ignore=IgnoreReason.USERROLEINRESOURCE, tenant_ignore=IgnoreReason.TENANT, tenant_admin_perm=True + ), + field_id('MetadataForm', 'fields'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('MetadataForm', 'homeEntityName'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('MetadataForm', 'userRole'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('MetadataFormField', 'glossaryNodeName'): TestData( + resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('MetadataFormSearchResult', 'hasTenantPermissions'): TestData( + resource_ignore=IgnoreReason.USERROLEINRESOURCE, tenant_ignore=IgnoreReason.TENANT, tenant_admin_perm=True + ), + field_id('Mutation', 'DisableDataSubscriptions'): TestData( + tenant_perm=MANAGE_ENVIRONMENTS, resource_perm=ENABLE_ENVIRONMENT_SUBSCRIPTIONS + ), + field_id('Mutation', 'addConnectionGroupPermission'): TestData( + tenant_ignore=IgnoreReason.BACKPORT, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'addConsumptionRoleToEnvironment'): TestData( + tenant_perm=MANAGE_ENVIRONMENTS, resource_perm=ADD_ENVIRONMENT_CONSUMPTION_ROLES + ), + field_id('Mutation', 'addRedshiftDatasetTables'): TestData( + tenant_perm=MANAGE_REDSHIFT_DATASETS, resource_perm=ADD_TABLES_REDSHIFT_DATASET + ), + field_id('Mutation', 'addSharedItem'): TestData(tenant_perm=MANAGE_SHARES, resource_perm=ADD_ITEM), + field_id('Mutation', 'approveDashboardShare'): TestData( + tenant_perm=MANAGE_DASHBOARDS, resource_perm=SHARE_DASHBOARD + ), + field_id('Mutation', 'approveShareExtension'): TestData( + tenant_perm=MANAGE_SHARES, resource_perm=APPROVE_SHARE_OBJECT + ), + field_id('Mutation', 'approveShareObject'): TestData(tenant_perm=MANAGE_SHARES, resource_perm=APPROVE_SHARE_OBJECT), + field_id('Mutation', 'approveTermAssociation'): TestData( + tenant_perm=MANAGE_GLOSSARIES, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'archiveOrganization'): TestData( + tenant_perm=MANAGE_ORGANIZATIONS, resource_perm=DELETE_ORGANIZATION + ), + field_id('Mutation', 'batchMetadataFormFieldUpdates'): TestData( + tenant_perm=MANAGE_METADATA_FORMS, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'cancelShareExtension'): TestData( + tenant_perm=MANAGE_SHARES, resource_perm=SUBMIT_SHARE_OBJECT + ), + field_id('Mutation', 'createAttachedMetadataForm'): TestData( + tenant_ignore=IgnoreReason.BACKPORT, resource_perm=ATTACH_METADATA_FORM + ), + field_id('Mutation', 'createCategory'): TestData( + tenant_perm=MANAGE_GLOSSARIES, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'createDataPipeline'): TestData(tenant_perm=MANAGE_PIPELINES, resource_perm=CREATE_PIPELINE), + field_id('Mutation', 'createDataPipelineEnvironment'): TestData( + tenant_perm=MANAGE_PIPELINES, resource_perm=CREATE_PIPELINE + ), + field_id('Mutation', 'createDataset'): TestData(tenant_perm=MANAGE_DATASETS, resource_perm=CREATE_DATASET), + field_id('Mutation', 'createDatasetStorageLocation'): TestData( + tenant_perm=MANAGE_DATASETS, resource_perm=CREATE_DATASET_FOLDER + ), + field_id('Mutation', 'createEnvironment'): TestData( + tenant_perm=MANAGE_ENVIRONMENTS, resource_perm=LINK_ENVIRONMENT + ), + field_id('Mutation', 'createGlossary'): TestData( + tenant_perm=MANAGE_GLOSSARIES, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'createMetadataForm'): TestData( + tenant_perm=MANAGE_METADATA_FORMS, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'createMetadataFormFields'): TestData( + tenant_perm=MANAGE_METADATA_FORMS, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'createMetadataFormVersion'): TestData( + tenant_perm=MANAGE_METADATA_FORMS, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'createNetwork'): TestData(tenant_perm=MANAGE_ENVIRONMENTS, resource_perm=CREATE_NETWORK), + field_id('Mutation', 'createOmicsRun'): TestData(tenant_perm=MANAGE_OMICS_RUNS, resource_perm=CREATE_OMICS_RUN), + field_id('Mutation', 'createOrganization'): TestData( + tenant_perm=MANAGE_ORGANIZATIONS, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'createQuicksightDataSourceSet'): TestData( + tenant_ignore=IgnoreReason.TENANT, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'createRedshiftConnection'): TestData( + tenant_ignore=IgnoreReason.BACKPORT, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'createSagemakerNotebook'): TestData( + tenant_perm=MANAGE_NOTEBOOKS, resource_perm=CREATE_NOTEBOOK + ), + field_id('Mutation', 'createSagemakerStudioUser'): TestData( + tenant_perm=MANAGE_SGMSTUDIO_USERS, resource_perm=CREATE_SGMSTUDIO_USER + ), + field_id('Mutation', 'createShareObject'): TestData(tenant_perm=MANAGE_SHARES, resource_perm=CREATE_SHARE_OBJECT), + field_id('Mutation', 'createTableDataFilter'): TestData( + tenant_perm=MANAGE_DATASETS, resource_perm=CREATE_TABLE_DATA_FILTER + ), + field_id('Mutation', 'createTerm'): TestData( + tenant_perm=MANAGE_GLOSSARIES, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'createWorksheet'): TestData( + tenant_perm=MANAGE_WORKSHEETS, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'deleteAttachedMetadataForm'): TestData( + tenant_ignore=IgnoreReason.BACKPORT, resource_perm=ATTACH_METADATA_FORM + ), + field_id('Mutation', 'deleteCategory'): TestData( + tenant_perm=MANAGE_GLOSSARIES, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'deleteConnectionGroupPermission'): TestData( + tenant_ignore=IgnoreReason.BACKPORT, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'deleteDashboard'): TestData(tenant_perm=MANAGE_DASHBOARDS, resource_perm=DELETE_DASHBOARD), + field_id('Mutation', 'deleteDataPipeline'): TestData(tenant_perm=MANAGE_PIPELINES, resource_perm=DELETE_PIPELINE), + field_id('Mutation', 'deleteDataPipelineEnvironment'): TestData( + tenant_perm=MANAGE_PIPELINES, resource_perm=UPDATE_PIPELINE + ), + field_id('Mutation', 'deleteDataset'): TestData(tenant_perm=MANAGE_DATASETS, resource_perm=DELETE_DATASET), + field_id('Mutation', 'deleteDatasetStorageLocation'): TestData( + tenant_perm=MANAGE_DATASETS, resource_perm=DELETE_DATASET_FOLDER + ), + field_id('Mutation', 'deleteDatasetTable'): TestData( + tenant_perm=MANAGE_DATASETS, resource_perm=DELETE_DATASET_TABLE + ), + field_id('Mutation', 'deleteEnvironment'): TestData( + tenant_perm=MANAGE_ENVIRONMENTS, resource_perm=DELETE_ENVIRONMENT + ), + field_id('Mutation', 'deleteGlossary'): TestData( + tenant_perm=MANAGE_GLOSSARIES, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'deleteMetadataForm'): TestData( + tenant_perm=MANAGE_METADATA_FORMS, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'deleteMetadataFormField'): TestData( + tenant_perm=MANAGE_METADATA_FORMS, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'deleteMetadataFormVersion'): TestData( + tenant_perm=MANAGE_METADATA_FORMS, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'deleteNetwork'): TestData(tenant_perm=MANAGE_ENVIRONMENTS, resource_perm=DELETE_NETWORK), + field_id('Mutation', 'deleteNotification'): TestData( + tenant_ignore=IgnoreReason.APPSUPPORT, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'deleteOmicsRun'): TestData( + tenant_perm=MANAGE_OMICS_RUNS, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'deleteRedshiftConnection'): TestData( + tenant_ignore=IgnoreReason.BACKPORT, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'deleteRedshiftDataset'): TestData( + tenant_perm=MANAGE_REDSHIFT_DATASETS, resource_perm=DELETE_REDSHIFT_DATASET + ), + field_id('Mutation', 'deleteRedshiftDatasetTable'): TestData( + tenant_perm=MANAGE_REDSHIFT_DATASETS, resource_perm=DELETE_REDSHIFT_DATASET_TABLE + ), + field_id('Mutation', 'deleteSagemakerNotebook'): TestData( + tenant_perm=MANAGE_NOTEBOOKS, resource_perm=DELETE_NOTEBOOK + ), + field_id('Mutation', 'deleteSagemakerStudioUser'): TestData( + tenant_perm=MANAGE_SGMSTUDIO_USERS, resource_perm=DELETE_SGMSTUDIO_USER + ), + field_id('Mutation', 'deleteShareObject'): TestData(tenant_perm=MANAGE_SHARES, resource_perm=DELETE_SHARE_OBJECT), + field_id('Mutation', 'deleteTableDataFilter'): TestData( + tenant_perm=MANAGE_DATASETS, resource_perm=DELETE_TABLE_DATA_FILTER + ), + field_id('Mutation', 'deleteTerm'): TestData( + tenant_perm=MANAGE_GLOSSARIES, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'deleteWorksheet'): TestData(tenant_perm=MANAGE_WORKSHEETS, resource_perm=DELETE_WORKSHEET), + field_id('Mutation', 'dismissTermAssociation'): TestData( + tenant_perm=MANAGE_GLOSSARIES, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'enableDataSubscriptions'): TestData( + tenant_perm=MANAGE_ENVIRONMENTS, resource_perm=ENABLE_ENVIRONMENT_SUBSCRIPTIONS + ), + field_id('Mutation', 'generateDatasetAccessToken'): TestData( + tenant_perm=MANAGE_DATASETS, resource_perm=CREDENTIALS_DATASET + ), + field_id('Mutation', 'importDashboard'): TestData(tenant_perm=MANAGE_DASHBOARDS, resource_perm=CREATE_DASHBOARD), + field_id('Mutation', 'importDataset'): TestData(tenant_perm=MANAGE_DATASETS, resource_perm=CREATE_DATASET), + field_id('Mutation', 'importRedshiftDataset'): TestData( + tenant_perm=MANAGE_REDSHIFT_DATASETS, resource_perm=IMPORT_REDSHIFT_DATASET + ), + field_id('Mutation', 'inviteGroupOnEnvironment'): TestData( + tenant_perm=MANAGE_ENVIRONMENTS, resource_perm=INVITE_ENVIRONMENT_GROUP + ), + field_id('Mutation', 'inviteGroupToOrganization'): TestData( + tenant_perm=MANAGE_ORGANIZATIONS, resource_perm=INVITE_ORGANIZATION_GROUP + ), + field_id('Mutation', 'markNotificationAsRead'): TestData( + tenant_ignore=IgnoreReason.APPSUPPORT, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'postFeedMessage'): TestData( + tenant_ignore=IgnoreReason.APPSUPPORT, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'reApplyItemsShareObject'): TestData( + tenant_perm=MANAGE_SHARES, resource_perm=APPROVE_SHARE_OBJECT + ), + field_id('Mutation', 'reApplyShareObjectItemsOnDataset'): TestData( + tenant_perm=MANAGE_DATASETS, resource_perm=UPDATE_DATASET + ), + field_id('Mutation', 'rejectDashboardShare'): TestData( + tenant_perm=MANAGE_DASHBOARDS, resource_perm=SHARE_DASHBOARD + ), + field_id('Mutation', 'rejectShareObject'): TestData(tenant_perm=MANAGE_SHARES, resource_perm=REJECT_SHARE_OBJECT), + field_id('Mutation', 'removeConsumptionRoleFromEnvironment'): TestData( + tenant_perm=MANAGE_ENVIRONMENTS, resource_perm=REMOVE_ENVIRONMENT_CONSUMPTION_ROLE + ), + field_id('Mutation', 'removeGroupFromEnvironment'): TestData( + tenant_perm=MANAGE_ENVIRONMENTS, resource_perm=REMOVE_ENVIRONMENT_GROUP + ), + field_id('Mutation', 'removeGroupFromOrganization'): TestData( + tenant_perm=MANAGE_ORGANIZATIONS, resource_perm=REMOVE_ORGANIZATION_GROUP + ), + field_id('Mutation', 'removeShareItemFilter'): TestData( + tenant_perm=MANAGE_SHARES, resource_perm=APPROVE_SHARE_OBJECT + ), + field_id('Mutation', 'removeSharedItem'): TestData(tenant_perm=MANAGE_SHARES, resource_perm=REMOVE_ITEM), + field_id('Mutation', 'requestDashboardShare'): TestData( + tenant_perm=MANAGE_DASHBOARDS, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'revokeItemsShareObject'): TestData(tenant_perm=MANAGE_SHARES, resource_perm=GET_SHARE_OBJECT), + field_id('Mutation', 'startDatasetProfilingRun'): TestData( + tenant_perm=MANAGE_DATASETS, resource_perm=PROFILE_DATASET_TABLE + ), + field_id('Mutation', 'startGlueCrawler'): TestData(tenant_perm=MANAGE_DATASETS, resource_perm=CRAWL_DATASET), + field_id('Mutation', 'startMaintenanceWindow'): TestData( + tenant_ignore=IgnoreReason.TENANT, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'startReindexCatalog'): TestData( + tenant_ignore=IgnoreReason.TENANT, resource_ignore=IgnoreReason.TENANT, tenant_admin_perm=True + ), + field_id('Mutation', 'startSagemakerNotebook'): TestData(tenant_perm=MANAGE_NOTEBOOKS, resource_perm=GET_NOTEBOOK), + field_id('Mutation', 'stopMaintenanceWindow'): TestData( + tenant_ignore=IgnoreReason.TENANT, resource_ignore=IgnoreReason.TENANT, tenant_admin_perm=True + ), + field_id('Mutation', 'stopSagemakerNotebook'): TestData(tenant_perm=MANAGE_NOTEBOOKS, resource_perm=GET_NOTEBOOK), + field_id('Mutation', 'submitShareExtension'): TestData( + tenant_perm=MANAGE_SHARES, resource_perm=SUBMIT_SHARE_OBJECT + ), + field_id('Mutation', 'submitShareObject'): TestData(tenant_perm=MANAGE_SHARES, resource_perm=SUBMIT_SHARE_OBJECT), + field_id('Mutation', 'syncDatasetTableColumns'): TestData( + tenant_perm=MANAGE_DATASETS, resource_perm=UPDATE_DATASET_TABLE + ), + field_id('Mutation', 'syncTables'): TestData(tenant_perm=MANAGE_DATASETS, resource_perm=SYNC_DATASET), + field_id('Mutation', 'upVote'): TestData( + tenant_ignore=IgnoreReason.APPSUPPORT, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'updateCategory'): TestData( + tenant_perm=MANAGE_GLOSSARIES, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'updateConsumptionRole'): TestData( + tenant_perm=MANAGE_ENVIRONMENTS, resource_perm=REMOVE_ENVIRONMENT_CONSUMPTION_ROLE + ), + field_id('Mutation', 'updateDashboard'): TestData(tenant_perm=MANAGE_DASHBOARDS, resource_perm=UPDATE_DASHBOARD), + field_id('Mutation', 'updateDataPipeline'): TestData(tenant_perm=MANAGE_PIPELINES, resource_perm=UPDATE_PIPELINE), + field_id('Mutation', 'updateDataPipelineEnvironment'): TestData( + tenant_perm=MANAGE_PIPELINES, resource_perm=UPDATE_PIPELINE + ), + field_id('Mutation', 'updateDataset'): TestData(tenant_perm=MANAGE_DATASETS, resource_perm=UPDATE_DATASET), + field_id('Mutation', 'updateDatasetStorageLocation'): TestData( + tenant_perm=MANAGE_DATASETS, resource_perm=UPDATE_DATASET_FOLDER + ), + field_id('Mutation', 'updateDatasetTable'): TestData( + tenant_perm=MANAGE_DATASETS, resource_perm=UPDATE_DATASET_TABLE + ), + field_id('Mutation', 'updateDatasetTableColumn'): TestData( + tenant_perm=MANAGE_DATASETS, resource_perm=UPDATE_DATASET_TABLE + ), + field_id('Mutation', 'updateEnvironment'): TestData( + tenant_perm=MANAGE_ENVIRONMENTS, resource_perm=UPDATE_ENVIRONMENT + ), + field_id('Mutation', 'updateGlossary'): TestData( + tenant_perm=MANAGE_GLOSSARIES, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'updateGroupEnvironmentPermissions'): TestData( + tenant_perm=MANAGE_ENVIRONMENTS, resource_perm=UPDATE_ENVIRONMENT_GROUP + ), + field_id('Mutation', 'updateGroupTenantPermissions'): TestData( + tenant_ignore=IgnoreReason.TENANT, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'updateKeyValueTags'): TestData( + tenant_perm=MANAGE_ENVIRONMENTS, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'updateOrganization'): TestData( + tenant_perm=MANAGE_ORGANIZATIONS, resource_perm=UPDATE_ORGANIZATION + ), + field_id('Mutation', 'updateOrganizationGroup'): TestData( + tenant_perm=MANAGE_ORGANIZATIONS, resource_perm=INVITE_ORGANIZATION_GROUP + ), + field_id('Mutation', 'updateRedshiftDataset'): TestData( + tenant_perm=MANAGE_REDSHIFT_DATASETS, resource_perm=UPDATE_REDSHIFT_DATASET + ), + field_id('Mutation', 'updateRedshiftDatasetTable'): TestData( + tenant_perm=MANAGE_REDSHIFT_DATASETS, resource_perm=UPDATE_REDSHIFT_DATASET_TABLE + ), + field_id('Mutation', 'updateSSMParameter'): TestData( + tenant_ignore=IgnoreReason.TENANT, resource_ignore=IgnoreReason.TENANT, tenant_admin_perm=True + ), + field_id('Mutation', 'updateShareExpirationPeriod'): TestData( + tenant_perm=MANAGE_SHARES, resource_perm=SUBMIT_SHARE_OBJECT + ), + field_id('Mutation', 'updateShareExtensionReason'): TestData( + tenant_perm=MANAGE_SHARES, resource_perm=SUBMIT_SHARE_OBJECT + ), + field_id('Mutation', 'updateShareItemFilters'): TestData( + tenant_perm=MANAGE_SHARES, resource_perm=APPROVE_SHARE_OBJECT + ), + field_id('Mutation', 'updateShareRejectReason'): TestData( + tenant_perm=MANAGE_SHARES, resource_perm=REJECT_SHARE_OBJECT + ), + field_id('Mutation', 'updateShareRequestReason'): TestData( + tenant_perm=MANAGE_SHARES, resource_perm=SUBMIT_SHARE_OBJECT + ), + field_id('Mutation', 'updateStack'): TestData( + tenant_perm=MANAGE_ENVIRONMENTS, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'updateTerm'): TestData( + tenant_perm=MANAGE_GLOSSARIES, resource_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Mutation', 'updateWorksheet'): TestData(tenant_perm=MANAGE_WORKSHEETS, resource_perm=UPDATE_WORKSHEET), + field_id('Mutation', 'verifyDatasetShareObjects'): TestData( + tenant_perm=MANAGE_DATASETS, resource_perm=UPDATE_DATASET + ), + field_id('Mutation', 'verifyItemsShareObject'): TestData(tenant_perm=MANAGE_SHARES, resource_perm=GET_SHARE_OBJECT), + field_id('OmicsRun', 'environment'): TestData( + resource_perm=GET_ENVIRONMENT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('OmicsRun', 'organization'): TestData( + resource_perm=GET_ORGANIZATION, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('OmicsRun', 'status'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('OmicsRun', 'workflow'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Organization', 'environments'): TestData( + resource_perm=GET_ORGANIZATION, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Organization', 'stats'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Organization', 'userRoleInOrganization'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Permission', 'type'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'countDeletedNotifications'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'countReadNotifications'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'countUnreadNotifications'): TestData( + resource_ignore=IgnoreReason.USERLIMITED, tenant_ignore=IgnoreReason.USERLIMITED + ), + field_id('Query', 'countUpVotes'): TestData(resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.PUBLIC), + field_id('Query', 'generateEnvironmentAccessToken'): TestData( + tenant_perm=MANAGE_ENVIRONMENTS, resource_perm=CREDENTIALS_ENVIRONMENT + ), + field_id('Query', 'getAttachedMetadataForm'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getAuthorSession'): TestData(resource_perm=CREATE_DASHBOARD, tenant_perm=MANAGE_DASHBOARDS), + field_id('Query', 'getCDKExecPolicyPresignedUrl'): TestData( + resource_perm=LINK_ENVIRONMENT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getConsumptionRolePolicies'): TestData( + resource_perm=GET_ENVIRONMENT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getDashboard'): TestData(resource_perm=GET_DASHBOARD, tenant_ignore=IgnoreReason.NOTREQUIRED), + field_id('Query', 'getDataPipeline'): TestData(resource_perm=GET_PIPELINE, tenant_ignore=IgnoreReason.NOTREQUIRED), + field_id('Query', 'getDataPipelineCredsLinux'): TestData( + resource_perm=CREDENTIALS_PIPELINE, tenant_perm=MANAGE_PIPELINES + ), + field_id('Query', 'getDataset'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), # TODO Review + field_id('Query', 'getDatasetAssumeRoleUrl'): TestData( + tenant_perm=MANAGE_DATASETS, resource_perm=CREDENTIALS_DATASET + ), + field_id('Query', 'getDatasetPresignedUrl'): TestData( + tenant_perm=MANAGE_DATASETS, resource_perm=CREDENTIALS_DATASET + ), + field_id('Query', 'getDatasetSharedAssumeRoleUrl'): TestData( + tenant_perm=MANAGE_DATASETS, resource_perm=CREDENTIALS_DATASET + ), + field_id('Query', 'getDatasetStorageLocation'): TestData( + resource_perm=GET_DATASET_FOLDER, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getDatasetTable'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), # TODO Review + field_id('Query', 'getDatasetTableProfilingRun'): TestData( + resource_ignore=IgnoreReason.CUSTOM, tenant_ignore=IgnoreReason.CUSTOM + ), + field_id('Query', 'getEntityMetadataFormPermissions'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getEnvironment'): TestData( + resource_perm=GET_ENVIRONMENT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getEnvironmentAssumeRoleUrl'): TestData( + tenant_perm=MANAGE_ENVIRONMENTS, resource_perm=CREDENTIALS_ENVIRONMENT + ), + field_id('Query', 'getEnvironmentMLStudioDomain'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), # TODO add resource_perm GET_ENVIRONMENT + field_id('Query', 'getFeed'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getGlossary'): TestData(resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.PUBLIC), + field_id('Query', 'getGroup'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getGroupsForUser'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getMaintenanceWindowStatus'): TestData( + resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.PUBLIC + ), + field_id('Query', 'getMetadataForm'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getMonitoringDashboardId'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getMonitoringVPCConnectionId'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getOmicsWorkflow'): TestData( + resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.PUBLIC + ), + field_id('Query', 'getOrganization'): TestData( + resource_perm=GET_ORGANIZATION, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getPivotRoleExternalId'): TestData( + resource_perm=LINK_ENVIRONMENT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getPivotRoleName'): TestData( + resource_perm=LINK_ENVIRONMENT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getPivotRolePresignedUrl'): TestData( + resource_perm=LINK_ENVIRONMENT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getPlatformReaderSession'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getReaderSession'): TestData( + resource_perm=GET_DASHBOARD, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getRedshiftDataset'): TestData( + resource_perm=GET_REDSHIFT_DATASET, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getRedshiftDatasetTable'): TestData( + resource_perm=GET_REDSHIFT_DATASET_TABLE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getRedshiftDatasetTableColumns'): TestData( + resource_perm=GET_REDSHIFT_DATASET_TABLE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getS3ConsumptionData'): TestData( + resource_perm=GET_SHARE_OBJECT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getSagemakerNotebook'): TestData( + resource_perm=GET_NOTEBOOK, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getSagemakerNotebookPresignedUrl'): TestData( + tenant_perm=MANAGE_NOTEBOOKS, resource_perm=GET_NOTEBOOK + ), + field_id('Query', 'getSagemakerStudioUser'): TestData( + resource_perm=GET_SGMSTUDIO_USER, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getSagemakerStudioUserPresignedUrl'): TestData( + tenant_perm=MANAGE_SGMSTUDIO_USERS, + resource_perm=SGMSTUDIO_USER_URL, + ), + field_id('Query', 'getShareItemDataFilters'): TestData( + resource_perm=GET_SHARE_OBJECT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getShareLogs'): TestData(resource_ignore=IgnoreReason.CUSTOM, tenant_ignore=IgnoreReason.CUSTOM), + field_id('Query', 'getShareObject'): TestData( + resource_perm=GET_SHARE_OBJECT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getShareRequestsFromMe'): TestData( + tenant_ignore=IgnoreReason.USERLIMITED, resource_ignore=IgnoreReason.USERLIMITED + ), + field_id('Query', 'getShareRequestsToMe'): TestData( + resource_ignore=IgnoreReason.USERLIMITED, tenant_ignore=IgnoreReason.USERLIMITED + ), + field_id('Query', 'getSharedDatasetTables'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), # TODO Review + field_id('Query', 'getStack'): TestData(resource_perm='getStack', tenant_ignore=IgnoreReason.NOTREQUIRED), + field_id('Query', 'getStackLogs'): TestData(resource_perm='getStack', tenant_ignore=IgnoreReason.NOTREQUIRED), + field_id('Query', 'getTrustAccount'): TestData( + resource_perm=LINK_ENVIRONMENT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'getVote'): TestData(resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.PUBLIC), + field_id('Query', 'getWorksheet'): TestData(resource_perm=GET_WORKSHEET, tenant_ignore=IgnoreReason.NOTREQUIRED), + field_id('Query', 'listAllConsumptionRoles'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.USERLIMITED + ), + field_id('Query', 'listAllEnvironmentConsumptionRoles'): TestData( + resource_perm=LIST_ENVIRONMENT_CONSUMPTION_ROLES, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listAllEnvironmentGroups'): TestData( + resource_perm=LIST_ENVIRONMENT_GROUPS, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listAllGroups'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.USERLIMITED + ), + field_id('Query', 'listAttachedMetadataForms'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.TENANT, tenant_admin_perm=True + ), + field_id('Query', 'listConnectionGroupNoPermissions'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listConnectionGroupPermissions'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listDashboardShares'): TestData( + resource_ignore=IgnoreReason.USERLIMITED, tenant_ignore=IgnoreReason.USERLIMITED + ), + field_id('Query', 'listDataPipelines'): TestData( + resource_ignore=IgnoreReason.USERLIMITED, tenant_ignore=IgnoreReason.USERLIMITED + ), + field_id('Query', 'listDatasetTableColumns'): TestData( + resource_ignore=IgnoreReason.CUSTOM, tenant_ignore=IgnoreReason.CUSTOM + ), + field_id('Query', 'listDatasetTableProfilingRuns'): TestData( + resource_ignore=IgnoreReason.CUSTOM, tenant_ignore=IgnoreReason.CUSTOM + ), + field_id('Query', 'listDatasetTables'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listDatasets'): TestData( + resource_ignore=IgnoreReason.USERLIMITED, tenant_ignore=IgnoreReason.USERLIMITED + ), + field_id('Query', 'listDatasetsCreatedInEnvironment'): TestData( + resource_perm=LIST_ENVIRONMENT_DATASETS, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listEntityMetadataForms'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.TENANT, tenant_admin_perm=True + ), + field_id('Query', 'listEnvironmentConsumptionRoles'): TestData( + resource_perm=LIST_ENVIRONMENT_CONSUMPTION_ROLES, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listEnvironmentGroupInvitationPermissions'): TestData( + resource_ignore=IgnoreReason.USERROLEINRESOURCE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listEnvironmentGroups'): TestData( + resource_perm=LIST_ENVIRONMENT_GROUPS, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listEnvironmentInvitedGroups'): TestData( + resource_perm=LIST_ENVIRONMENT_GROUPS, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listEnvironmentNetworks'): TestData( + resource_perm=LIST_ENVIRONMENT_NETWORKS, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listEnvironmentRedshiftConnections'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listEnvironments'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.USERLIMITED + ), + field_id('Query', 'listGlossaries'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listGroups'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listInviteOrganizationPermissionsWithDescriptions'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listKeyValueTags'): TestData( + resource_perm='listKeyValueTags', tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listMetadataFormVersions'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listNotifications'): TestData( + resource_ignore=IgnoreReason.USERLIMITED, tenant_ignore=IgnoreReason.USERLIMITED + ), + field_id('Query', 'listOmicsRuns'): TestData( + resource_ignore=IgnoreReason.USERLIMITED, tenant_ignore=IgnoreReason.USERLIMITED + ), + field_id('Query', 'listOmicsWorkflows'): TestData( + resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.PUBLIC + ), + field_id('Query', 'listOrganizationGroupPermissions'): TestData( + resource_perm=GET_ORGANIZATION, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listOrganizationGroups'): TestData( + resource_perm=GET_ORGANIZATION, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listOrganizations'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.USERLIMITED + ), + field_id('Query', 'listOwnedDatasets'): TestData( + resource_ignore=IgnoreReason.USERLIMITED, tenant_ignore=IgnoreReason.USERLIMITED + ), + field_id('Query', 'listRedshiftConnectionSchemas'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listRedshiftDatasetTables'): TestData( + resource_perm=GET_REDSHIFT_DATASET, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listRedshiftSchemaDatasetTables'): TestData( + resource_perm=GET_REDSHIFT_DATASET, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listRedshiftSchemaTables'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listS3DatasetsOwnedByEnvGroup'): TestData( + resource_perm=LIST_ENVIRONMENT_DATASETS, tenant_ignore=IgnoreReason.USERLIMITED + ), + field_id('Query', 'listS3DatasetsSharedWithEnvGroup'): TestData( + resource_perm=LIST_ENVIRONMENT_DATASETS, tenant_ignore=IgnoreReason.NOTREQUIRED + ), # TODO Review + field_id('Query', 'listSagemakerNotebooks'): TestData( + resource_ignore=IgnoreReason.USERLIMITED, tenant_ignore=IgnoreReason.USERLIMITED + ), + field_id('Query', 'listSagemakerStudioUsers'): TestData( + resource_ignore=IgnoreReason.USERLIMITED, tenant_ignore=IgnoreReason.USERLIMITED + ), + field_id('Query', 'listSharedDatasetTableColumns'): TestData( + resource_perm=GET_DATASET_TABLE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listTableDataFilters'): TestData( + resource_perm=LIST_TABLE_DATA_FILTERS, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listTableDataFiltersByAttached'): TestData( + resource_perm=GET_SHARE_OBJECT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listTenantGroups'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.TENANT, tenant_admin_perm=True + ), + field_id('Query', 'listTenantPermissions'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.TENANT, tenant_admin_perm=True + ), + field_id('Query', 'listUserMetadataForms'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.TENANT, tenant_admin_perm=True + ), + field_id('Query', 'listUsersForGroup'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'listValidEnvironments'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.USERLIMITED + ), + field_id('Query', 'listWorksheets'): TestData( + resource_ignore=IgnoreReason.USERLIMITED, tenant_ignore=IgnoreReason.USERLIMITED + ), + field_id('Query', 'previewTable'): TestData(resource_ignore=IgnoreReason.CUSTOM, tenant_ignore=IgnoreReason.CUSTOM), + field_id('Query', 'queryEnums'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'runAthenaSqlQuery'): TestData(resource_perm=RUN_ATHENA_QUERY, tenant_perm=MANAGE_WORKSHEETS), + field_id('Query', 'searchDashboards'): TestData( + resource_ignore=IgnoreReason.USERLIMITED, tenant_ignore=IgnoreReason.USERLIMITED + ), + field_id('Query', 'searchEnvironmentDataItems'): TestData( + resource_perm=LIST_ENVIRONMENT_SHARED_WITH_OBJECTS, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Query', 'searchGlossary'): TestData( + resource_ignore=IgnoreReason.NOTREQUIRED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('RedshiftDataset', 'connection'): TestData( + resource_perm=GET_REDSHIFT_CONNECTION, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('RedshiftDataset', 'environment'): TestData( + resource_perm=GET_ENVIRONMENT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('RedshiftDataset', 'owners'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('RedshiftDataset', 'stewards'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('RedshiftDataset', 'terms'): TestData( + resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('RedshiftDataset', 'upvotes'): TestData( + resource_perm=GET_REDSHIFT_DATASET, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('RedshiftDataset', 'userRoleForDataset'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('RedshiftDatasetTable', 'dataset'): TestData( + resource_perm=GET_REDSHIFT_DATASET, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('RedshiftDatasetTable', 'terms'): TestData( + resource_ignore=IgnoreReason.PUBLIC, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('SagemakerNotebook', 'NotebookInstanceStatus'): TestData( + resource_perm=GET_NOTEBOOK, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('SagemakerNotebook', 'environment'): TestData( + resource_perm=GET_ENVIRONMENT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('SagemakerNotebook', 'organization'): TestData( + resource_perm=GET_ORGANIZATION, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('SagemakerNotebook', 'stack'): TestData( + resource_perm=GET_NOTEBOOK, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('SagemakerNotebook', 'userRoleForNotebook'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('SagemakerStudioDomain', 'environment'): TestData( + resource_perm=GET_ENVIRONMENT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('SagemakerStudioUser', 'environment'): TestData( + resource_perm=GET_ENVIRONMENT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('SagemakerStudioUser', 'organization'): TestData( + resource_perm=GET_ORGANIZATION, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('SagemakerStudioUser', 'sagemakerStudioUserApps'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('SagemakerStudioUser', 'sagemakerStudioUserStatus'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('SagemakerStudioUser', 'stack'): TestData( + resource_perm=GET_SGMSTUDIO_USER, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('SagemakerStudioUser', 'userRoleForSagemakerStudioUser'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('ShareObject', 'canViewLogs'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('ShareObject', 'dataset'): TestData( + resource_ignore=IgnoreReason.SIMPLIFIED, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('ShareObject', 'environment'): TestData( + resource_perm=GET_ENVIRONMENT, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('ShareObject', 'existingSharedItems'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('ShareObject', 'group'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('ShareObject', 'items'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('ShareObject', 'principal'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('ShareObject', 'statistics'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('ShareObject', 'userRoleForShareObject'): TestData( + resource_ignore=IgnoreReason.USERROLEINRESOURCE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('SharedDatabaseTableItem', 'sharedGlueDatabaseName'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Stack', 'EcsTaskId'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Stack', 'canViewLogs'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Stack', 'error'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Stack', 'events'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Stack', 'link'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Stack', 'outputs'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Stack', 'resources'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Term', 'associations'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Term', 'children'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Term', 'glossary'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Term', 'stats'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), + field_id('Worksheet', 'userRoleForWorksheet'): TestData( + resource_ignore=IgnoreReason.INTRAMODULE, tenant_ignore=IgnoreReason.NOTREQUIRED + ), +} + +ALL_RESOLVERS = {(_type, field) for _type in bootstrap().types for field in _type.fields if field.resolver} + + +@pytest.fixture(scope='function') +def mock_input_validation(mocker): + mocker.patch('dataall.modules.mlstudio.api.resolvers.RequestValidator', MagicMock()) + mocker.patch( + 'dataall.modules.mlstudio.services.mlstudio_service.SagemakerStudioCreationRequest.from_dict', MagicMock() + ) + mocker.patch('dataall.modules.notebooks.api.resolvers.RequestValidator', MagicMock()) + mocker.patch('dataall.modules.notebooks.services.notebook_service.NotebookCreationRequest.from_dict', MagicMock()) + mocker.patch('dataall.modules.s3_datasets.api.profiling.resolvers._validate_uri', MagicMock()) + mocker.patch('dataall.modules.s3_datasets.api.storage_location.resolvers._validate_input', MagicMock()) + mocker.patch('dataall.modules.s3_datasets.api.dataset.resolvers.RequestValidator', MagicMock()) + mocker.patch( + 'dataall.core.stacks.db.target_type_repositories.TargetType.get_resource_tenant_permission_name', + return_value='MANAGE_ENVIRONMENTS', + ) + mocker.patch('dataall.modules.shares_base.api.resolvers.RequestValidator', MagicMock()) + # mock aws calls for speed + mocker.patch('dataall.base.aws.sts.SessionHelper._get_parameter_value') + mocker.patch('dataall.base.aws.sts.SessionHelper.remote_session') + mocker.patch('boto3.client').side_effect = RuntimeError('mocked boto3 client') + + +ALL_PARAMS = [pytest.param(field, id=field_id(_type.name, field.name)) for _type, field in ALL_RESOLVERS] + + +def test_all_resolvers_have_test_data(): + """ + ensure that all EXPECTED_RESOURCES_PERMS have a corresponding query (to avoid stale entries) and vice versa + """ + assert_that(ALL_PARAMS).extracting(2).described_as( + 'stale or missing EXPECTED_RESOURCE_PERMS detected' + ).contains_only(*EXPECTED_RESOLVERS.keys()) + + +@pytest.mark.parametrize('field', ALL_PARAMS) +@pytest.mark.parametrize('perm_type', ['resource', 'tenant', 'tenant_admin']) +@patch('dataall.base.context._request_storage') +@patch('dataall.core.permissions.services.resource_policy_service.ResourcePolicyService.check_user_resource_permission') +@patch('dataall.core.permissions.services.group_policy_service.GroupPolicyService.check_group_environment_permission') +@patch('dataall.core.permissions.services.tenant_policy_service.TenantPolicyService.check_user_tenant_permission') +@patch('dataall.core.permissions.services.tenant_policy_service.TenantPolicyValidationService.is_tenant_admin') +@patch('dataall.core.stacks.db.target_type_repositories.TargetType.get_resource_read_permission_name') +def test_permissions( + mock_perm_name, + mock_check_tenant_admin, + mock_check_tenant, + mock_check_group, + mock_check_resource, + mock_storage, + field, + perm_type, + request, + mock_input_validation, +): + fid = request.node.callspec.id.split('-')[-1] + perm, reason = EXPECTED_RESOLVERS[fid].get(perm_type) + msg = f'{fid} -> {field.resolver.__code__.co_filename}:{field.resolver.__code__.co_firstlineno}' + logging.info(msg) + assert_that(field.resolver).is_not_none() + # Setup mock context + username = 'ausername' + groups = ['agroup'] + mock_storage.context = RequestContext(MagicMock(), username, groups, 'auserid') + mock_storage.context.db_engine.scoped_session().__enter__().query().filter().all.return_value = [MagicMock()] + mock_perm_name.return_value = perm + + iargs = {arg: MagicMock() for arg in inspect.signature(field.resolver).parameters.keys()} + with suppress(Exception): + field.resolver(**iargs) + + if not perm: # if no expected permission is defined, we expect the check to not be called + locals()[f'mock_check_{perm_type}'].assert_not_called() # nosemgrep + pytest.skip(msg + f' Reason: {reason.value}') + elif perm_type == 'resource': + mock_check_resource.assert_any_call( + session=ANY, + resource_uri=ANY, + username=username, + groups=groups, + permission_name=perm, + ) + elif perm_type == 'tenant': + mock_check_tenant.assert_any_call( + session=ANY, + username=username, + groups=groups, + tenant_name=ANY, + permission_name=perm, + ) + elif perm_type == 'tenant_admin': + mock_check_tenant_admin.assert_called() + else: + raise ValueError(f'unknown permission type {perm_type}') diff --git a/tests/test_tenant_unauthorized.py b/tests/test_tenant_unauthorized.py deleted file mode 100644 index f87b5dde7..000000000 --- a/tests/test_tenant_unauthorized.py +++ /dev/null @@ -1,88 +0,0 @@ -from unittest.mock import MagicMock, patch -import pytest -from assertpy import assert_that -from dataall.base.api import bootstrap -from dataall.base.loader import load_modules, ImportMode -from dataall.base.context import RequestContext -from dataall.base.db.exceptions import TenantUnauthorized -import inspect - - -load_modules(modes={ImportMode.API}) - -OPT_OUT_MUTATIONS = { - 'Mutation.updateGroupTenantPermissions': 'admin action. No need for tenant permission check', - 'Mutation.updateSSMParameter': 'admin action. No need for tenant permission check', - 'Mutation.createQuicksightDataSourceSet': 'admin action. No need for tenant permission check', - 'Mutation.startMaintenanceWindow': 'admin action. No need for tenant permission check', - 'Mutation.stopMaintenanceWindow': 'admin action. No need for tenant permission check', - 'Mutation.startReindexCatalog': 'admin action. No need for tenant permission check', - 'Mutation.markNotificationAsRead': 'tenant permissions do not apply to support notifications', - 'Mutation.deleteNotification': 'tenant permissions do not apply to support notifications', - 'Mutation.postFeedMessage': 'tenant permissions do not apply to support feed messages', - 'Mutation.upVote': 'tenant permissions do not apply to support votes', - 'Mutation.createAttachedMetadataForm': 'outside of this PR to be able to backport to v2.6.2', - 'Mutation.deleteAttachedMetadataForm': 'outside of this PR to be able to backport to v2.6.2', - 'Mutation.createRedshiftConnection': 'outside of this PR to be able to backport to v2.6.2', - 'Mutation.deleteRedshiftConnection': 'outside of this PR to be able to backport to v2.6.2', - 'Mutation.addConnectionGroupPermission': 'outside of this PR to be able to backport to v2.6.2', - 'Mutation.deleteConnectionGroupPermission': 'outside of this PR to be able to backport to v2.6.2', -} - -OPT_IN_QUERIES = [ - 'Query.generateEnvironmentAccessToken', - 'Query.getEnvironmentAssumeRoleUrl', - 'Query.getSagemakerStudioUserPresignedUrl', - 'Query.getSagemakerNotebookPresignedUrl', - 'Query.getDatasetAssumeRoleUrl', - 'Query.getDatasetPresignedUrl', - 'Query.getAuthorSession', - 'Query.getDatasetSharedAssumeRoleUrl', - 'Query.runAthenaSqlQuery', -] - -ALL_RESOLVERS = {(_type, field) for _type in bootstrap().types for field in _type.fields if field.resolver} - - -@pytest.fixture(scope='function') -def mock_input_validation(mocker): - mocker.patch('dataall.modules.mlstudio.api.resolvers.RequestValidator', MagicMock()) - mocker.patch( - 'dataall.modules.mlstudio.services.mlstudio_service.SagemakerStudioCreationRequest.from_dict', MagicMock() - ) - mocker.patch('dataall.modules.notebooks.api.resolvers.RequestValidator', MagicMock()) - mocker.patch('dataall.modules.notebooks.services.notebook_service.NotebookCreationRequest.from_dict', MagicMock()) - mocker.patch('dataall.modules.s3_datasets.api.profiling.resolvers._validate_uri', MagicMock()) - mocker.patch('dataall.modules.s3_datasets.api.storage_location.resolvers._validate_input', MagicMock()) - mocker.patch('dataall.modules.s3_datasets.api.dataset.resolvers.RequestValidator', MagicMock()) - mocker.patch( - 'dataall.core.stacks.db.target_type_repositories.TargetType.get_resource_tenant_permission_name', - return_value='MANAGE_ENVIRONMENTS', - ) - mocker.patch('dataall.modules.shares_base.api.resolvers.RequestValidator', MagicMock()) - - -@pytest.mark.parametrize( - '_type,field', - [ - pytest.param(_type, field, id=f'{_type.name}.{field.name}') - for _type, field in ALL_RESOLVERS - if _type.name in ['Query', 'Mutation'] - ], -) -@patch('dataall.base.context._request_storage') -def test_unauthorized_tenant_permissions( - mock_local, _type, field, mock_input_validation, db, userNoTenantPermissions, groupNoTenantPermissions -): - if _type.name == 'Mutation' and f'{_type.name}.{field.name}' in OPT_OUT_MUTATIONS.keys(): - pytest.skip(f'Skipping test for {field.name}: {OPT_OUT_MUTATIONS[f"{_type.name}.{field.name}"]}') - if _type.name == 'Query' and f'{_type.name}.{field.name}' not in OPT_IN_QUERIES: - pytest.skip(f'Skipping test for {field.name}: This Query does not require a tenant permission check.') - assert_that(field.resolver).is_not_none() - mock_local.context = RequestContext( - db, userNoTenantPermissions.username, [groupNoTenantPermissions.groupUri], userNoTenantPermissions - ) - # Mocking arguments - iargs = {arg: MagicMock() for arg in inspect.signature(field.resolver).parameters.keys()} - # Assert Unauthorized exception is raised - assert_that(field.resolver).raises(TenantUnauthorized).when_called_with(**iargs).contains('UnauthorizedOperation') diff --git a/tests_new/integration_tests/client.py b/tests_new/integration_tests/client.py index bef02d39e..cb756ce9e 100644 --- a/tests_new/integration_tests/client.py +++ b/tests_new/integration_tests/client.py @@ -1,12 +1,16 @@ -import requests +import logging import os import uuid +from pprint import pformat from urllib.parse import parse_qs, urlparse + +import requests from munch import DefaultMunch -from retrying import retry -from integration_tests.errors import GqlError from oauthlib.oauth2 import WebApplicationClient from requests_oauthlib import OAuth2Session +from retrying import retry + +from integration_tests.errors import GqlError ENVNAME = os.getenv('ENVNAME', 'dev') @@ -28,15 +32,18 @@ def __init__(self, username, password): wait_random_min=1000, wait_random_max=3000, ) - def query(self, query: str): + def query(self, query: dict): graphql_endpoint = os.path.join(os.environ['API_ENDPOINT'], 'graphql', 'api') headers = {'accesskeyid': 'none', 'SecretKey': 'none', 'Authorization': f'Bearer {self.access_token}'} r = requests.post(graphql_endpoint, json=query, headers=headers) - if errors := r.json().get('errors'): - raise GqlError(errors) + response = r.json() + if errors := response.get('errors'): + if any((response.get('data', {}) or {}).values()): # check if there are data + logging.warning(f'{query=} returned both data and errors:\n {pformat(response)}') + else: + raise GqlError(errors) r.raise_for_status() - - return DefaultMunch.fromDict(r.json()) + return DefaultMunch.fromDict(response) def _get_jwt_tokens(self): token = uuid.uuid4()