def test_external_partition_names_error_grpc():
with get_bar_repo_handle() as repository_handle:
with pytest.raises(DagsterUserCodeProcessError, match="womp womp"):
sync_get_external_partition_names_grpc(
repository_handle.repository_location.client,
repository_handle,
"error_partitions",
)
def test_external_partition_names_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, match="womp womp"):
sync_get_external_partition_names_grpc(
repository_location.client,
repository_handle,
"error_partitions",
)
def get_external_partition_names(self, repository_handle, partition_set_name):
check.inst_param(repository_handle, "repository_handle", RepositoryHandle)
check.str_param(partition_set_name, "partition_set_name")
return sync_get_external_partition_names_grpc(
self._handle.client, repository_handle, partition_set_name
)
def test_external_partition_names_grpc():
with get_bar_repo_handle() as repository_handle:
data = sync_get_external_partition_names_grpc(
repository_handle.repository_location.client, repository_handle, "baz_partitions"
)
assert isinstance(data, ExternalPartitionNamesData)
assert data.partition_names == list(string.ascii_lowercase)
def test_external_partition_names_error_grpc():
repository_handle = get_bar_grpc_repo_handle()
error = sync_get_external_partition_names_grpc(
repository_handle.repository_location_handle.client, repository_handle,
'error_partitions')
assert isinstance(error, ExternalPartitionExecutionErrorData)
assert 'womp womp' in error.error.to_string()
def get_external_partition_names(
self, repository_handle: RepositoryHandle, partition_set_name: str
) -> "ExternalPartitionNamesData":
check.inst_param(repository_handle, "repository_handle", RepositoryHandle)
check.str_param(partition_set_name, "partition_set_name")
return sync_get_external_partition_names_grpc(
self.client, repository_handle, partition_set_name
)
def test_external_partition_names_error_grpc():
with get_bar_grpc_repo_handle() as repository_handle:
error = sync_get_external_partition_names_grpc(
repository_handle.repository_location_handle.client,
repository_handle,
"error_partitions",
)
assert isinstance(error, ExternalPartitionExecutionErrorData)
assert "womp womp" in error.error.to_string()
def test_external_partition_names_grpc():
repository_handle = get_bar_repo_handle()
data = sync_get_external_partition_names_grpc(repository_handle,
'baz_partitions')
assert isinstance(data, ExternalPartitionNamesData)
assert data.partition_names == list(string.ascii_lowercase)
