def test_external_partitions_tags(): repository_handle = get_bar_repo_handle() data = sync_get_external_partition_tags(repository_handle, 'baz_partitions', 'c') assert isinstance(data, ExternalPartitionTagsData) assert data.tags assert data.tags['foo'] == 'bar'
def test_external_partitions_tags(): repository_handle = get_bar_repo_handle() data = sync_get_external_partition_tags(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( repository_handle, partition_set_name, partition_name )
def get_partition_tags(graphene_info, 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') result = sync_get_external_partition_tags(repository_handle, partition_set_name, partition_name) if isinstance(result, ExternalPartitionTagsData): return graphene_info.schema.type_named('PartitionTags')( results=[ graphene_info.schema.type_named('PipelineTag')(key=key, value=value) for key, value in result.tags.items() ] ) else: return graphene_info.schema.type_named('PythonError')(result.error)
def get_partition_tags(graphene_info, 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') result = sync_get_external_partition_tags(repository_handle, partition_set_name, partition_name) if isinstance(result, ExternalPartitionTagsData): return [ graphene_info.schema.type_named('PipelineTag')(key=key, value=value) for key, value in result.tags.items() ] else: # TODO: surface user-facing error here, using the serialized error # https://github.com/dagster-io/dagster/issues/2576 return []
def test_external_partitions_tags_error(): repository_handle = get_bar_repo_handle() error = sync_get_external_partition_tags(repository_handle, 'error_partition_tags', 'c') assert isinstance(error, ExternalPartitionExecutionErrorData)