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()
