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"
示例#3
0
    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
        )
示例#4
0
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 []
示例#6
0
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)