about summary refs log tree commit diff
path: root/tvix/castore/src/proto/tests/directory_nodes_iterator.rs
blob: 68f147a332100732362dda9e84bcf14abcfeaf29 (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
use crate::proto::Directory;
use crate::proto::DirectoryNode;
use crate::proto::FileNode;
use crate::proto::NamedNode;
use crate::proto::SymlinkNode;

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

    // We keep this strings here and convert to string to make the comparison
    // less messy.
    let mut node_names: Vec<String> = vec![];

    for node in d.nodes() {
        node_names.push(String::from_utf8(node.get_name().to_vec()).unwrap());
    }

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