def test_dataset(self, client, organization, created_entities): visibility = OrgCustom(write=True) dataset = client.set_dataset(workspace=organization.name, visibility=visibility) created_entities.append(dataset) if visibility._to_public_within_org(): assert dataset._msg.dataset_visibility == _DatasetService.DatasetVisibilityEnum.ORG_SCOPED_PUBLIC else: assert dataset._msg.dataset_visibility == _DatasetService.DatasetVisibilityEnum.PRIVATE
def test_registered_model(self, client, organization, created_entities): visibility = OrgCustom(write=True) entity = client.set_registered_model(workspace=organization.name, visibility=visibility) created_entities.append(entity) if visibility._to_public_within_org(): assert entity._msg.visibility == _CommonCommonService.VisibilityEnum.ORG_SCOPED_PUBLIC else: assert entity._msg.visibility == _CommonCommonService.VisibilityEnum.PRIVATE
def test_project(self, client, organization): visibility = OrgCustom(write=True) entity = client.set_project(workspace=organization.name, visibility=visibility) try: if visibility._to_public_within_org(): assert entity._msg.project_visibility == _ProjectService.ORG_SCOPED_PUBLIC else: assert entity._msg.project_visibility == _ProjectService.PRIVATE finally: entity.delete() client._ctx.proj = None # otherwise client teardown tries to delete
def test_registered_model(self, client, organization): visibility = OrgCustom(write=True) entity = client.set_registered_model(workspace=organization.name, visibility=visibility) try: if visibility._to_public_within_org(): assert entity._msg.visibility == _CommonCommonService.VisibilityEnum.ORG_SCOPED_PUBLIC else: assert entity._msg.visibility == _CommonCommonService.VisibilityEnum.PRIVATE finally: entity.delete()
def test_dataset(self, client, organization): visibility = OrgCustom(write=True) entity = client.set_dataset(workspace=organization.name, visibility=visibility) try: if visibility._to_public_within_org(): assert entity._msg.dataset_visibility == _DatasetService.DatasetVisibilityEnum.ORG_SCOPED_PUBLIC else: assert entity._msg.dataset_visibility == _DatasetService.DatasetVisibilityEnum.PRIVATE finally: entity.delete()
def test_endpoint(self, client, organization, created_entities): visibility = OrgCustom(write=True) endpoint = client.set_endpoint( path=_utils.generate_default_name(), workspace=organization.name, visibility=visibility, ) created_entities.append(endpoint) endpoint_json = endpoint._get_json_by_id(endpoint._conn, endpoint.workspace, endpoint.id) if visibility._to_public_within_org(): assert endpoint_json['creator_request']['visibility'] == "ORG_SCOPED_PUBLIC" else: assert endpoint_json['creator_request']['visibility'] == "PRIVATE"
def test_repository(self, client, organization): visibility = OrgCustom(write=True) repo = client.set_repository( name=_utils.generate_default_name(), workspace=organization.name, visibility=visibility, ) try: retrieved_visibility = repo._get_proto_by_id( repo._conn, repo.id).repository_visibility if visibility._to_public_within_org(): assert retrieved_visibility == _VersioningService.RepositoryVisibilityEnum.ORG_SCOPED_PUBLIC else: assert retrieved_visibility == _VersioningService.RepositoryVisibilityEnum.PRIVATE finally: repo.delete()