def test_external_partitions_tags_error_grpc():
with get_bar_repo_handle() as repository_handle:
with pytest.raises(DagsterUserCodeProcessError):
sync_get_external_partition_tags_grpc(
repository_handle.repository_location.client,
repository_handle,
"error_partition_tags",
"c",
)
def test_external_partitions_tags_error_grpc(instance):
with get_bar_repo_repository_location(instance) as repository_location:
repository_handle = repository_location.get_repository(
"bar_repo").handle
with pytest.raises(DagsterUserCodeProcessError):
sync_get_external_partition_tags_grpc(
repository_location.client,
repository_handle,
"error_partition_tags",
"c",
)
def test_external_partitions_tags_grpc():
repository_handle = get_bar_repo_handle()
data = sync_get_external_partition_tags_grpc(repository_handle,
'baz_partitions', 'c')
assert isinstance(data, ExternalPartitionTagsData)
assert data.tags
assert data.tags['foo'] == 'bar'
def get_external_partition_tags(self, repository_handle, partition_set_name, partition_name):
check.inst_param(repository_handle, "repository_handle", RepositoryHandle)
check.str_param(partition_set_name, "partition_set_name")
check.str_param(partition_name, "partition_name")
return sync_get_external_partition_tags_grpc(
self._handle.client, repository_handle, partition_set_name, partition_name
)
def test_external_partitions_tags_error_grpc():
repository_handle = get_bar_grpc_repo_handle()
error = sync_get_external_partition_tags_grpc(
repository_handle.repository_location_handle.client,
repository_handle,
'error_partition_tags',
'c',
)
assert isinstance(error, ExternalPartitionExecutionErrorData)
def test_external_partitions_tags_error_grpc():
with get_bar_grpc_repo_handle() as repository_handle:
error = sync_get_external_partition_tags_grpc(
repository_handle.repository_location_handle.client,
repository_handle,
"error_partition_tags",
"c",
)
assert isinstance(error, ExternalPartitionExecutionErrorData)
def get_external_partition_tags(
self, repository_handle: RepositoryHandle, partition_set_name: str, partition_name: str
) -> "ExternalPartitionTagsData":
check.inst_param(repository_handle, "repository_handle", RepositoryHandle)
check.str_param(partition_set_name, "partition_set_name")
check.str_param(partition_name, "partition_name")
return sync_get_external_partition_tags_grpc(
self.client, repository_handle, partition_set_name, partition_name
)
def test_external_partitions_tags_grpc():
with get_bar_repo_handle() as repository_handle:
data = sync_get_external_partition_tags_grpc(
repository_handle.repository_location.client,
repository_handle,
"baz_partitions",
"c",
)
assert isinstance(data, ExternalPartitionTagsData)
assert data.tags
assert data.tags["foo"] == "bar"
