about summary refs log tree commit diff
path: root/tvix/store/src/tests/directory_nodes_iterator.rs
blob: 8f591f0560cef69ba2d3d3d25b60a86b4af8333d (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
use crate::proto::node::Node;
use crate::proto::Directory;
use crate::proto::DirectoryNode;
use crate::proto::FileNode;
use crate::proto::SymlinkNode;

#[test]
fn iterator() -> anyhow::Result<()> {
    let d = Directory {
        directories: vec![
            DirectoryNode {
                name: "c".to_string(),
                ..DirectoryNode::default()
            },
            DirectoryNode {
                name: "d".to_string(),
                ..DirectoryNode::default()
            },
            DirectoryNode {
                name: "h".to_string(),
                ..DirectoryNode::default()
            },
            DirectoryNode {
                name: "l".to_string(),
                ..DirectoryNode::default()
            },
        ],
        files: vec![
            FileNode {
                name: "b".to_string(),
                ..FileNode::default()
            },
            FileNode {
                name: "e".to_string(),
                ..FileNode::default()
            },
            FileNode {
                name: "g".to_string(),
                ..FileNode::default()
            },
            FileNode {
                name: "j".to_string(),
                ..FileNode::default()
            },
        ],
        symlinks: vec![
            SymlinkNode {
                name: "a".to_string(),
                ..SymlinkNode::default()
            },
            SymlinkNode {
                name: "f".to_string(),
                ..SymlinkNode::default()
            },
            SymlinkNode {
                name: "i".to_string(),
                ..SymlinkNode::default()
            },
            SymlinkNode {
                name: "k".to_string(),
                ..SymlinkNode::default()
            },
        ],
    };

    let mut node_names: Vec<String> = vec![];

    for node in d.nodes() {
        match node {
            Node::Directory(n) => node_names.push(n.name),
            Node::File(n) => node_names.push(n.name),
            Node::Symlink(n) => node_names.push(n.name),
        };
    }

    assert_eq!(
        vec!["a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l"],
        node_names
    );

    Ok(())
}