arbol 0.1.0

Tree-sitter rust AST query tool
Documentation 
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97

use std::fs;
use std::io::Write;
use std::path::Path;

use tempfile::TempDir;

use crate::dump_json;

fn write(path: &Path, content: &str) {
    if let Some(parent) = path.parent() {
        fs::create_dir_all(parent).unwrap();
    }
    let mut f = fs::File::create(path).unwrap();
    f.write_all(content.as_bytes()).unwrap();
}

#[test]
fn depth_zero_returns_only_root_node() {
    let td = TempDir::new().unwrap();
    let root = td.path();
    write(
        &root.join("Cargo.toml"),
        "[package]\nname='d0'\nversion='0.0.0'\nedition='2021'\n",
    );
    write(&root.join("src/lib.rs"), "pub fn alpha() {}\n");
    let full = dump_json(root, false, false, usize::MAX).unwrap();
    let shallow = dump_json(root, false, false, 0).unwrap();
    assert_eq!(full.len(), shallow.len(), "same file count");
    assert_eq!(shallow[0].nodes.len(), 1, "depth 0 should include only the root node");
}

#[test]
fn with_source_includes_short_text() {
    let td = TempDir::new().unwrap();
    let root = td.path();
    write(
        &root.join("Cargo.toml"),
        "[package]\nname='src'\nversion='0.0.0'\nedition='2021'\n",
    );
    write(&root.join("src/lib.rs"), "pub fn beta() { }\n");
    let asts = dump_json(root, false, true, 4).unwrap();
    assert!(!asts.is_empty());
    let has_text = asts[0].nodes.iter().any(|n| n.text.as_ref().is_some());
    assert!(
        has_text,
        "expected at least one node to include source text under threshold"
    );
}

#[test]
fn large_source_omits_long_text() {
    let td = TempDir::new().unwrap();
    let root = td.path();
    write(
        &root.join("Cargo.toml"),
        "[package]\nname='big'\nversion='0.0.0'\nedition='2021'\n",
    );
    // create a body well over 240 bytes
    let big_body = "x".repeat(400);
    write(
        &root.join("src/lib.rs"),
        &format!("pub fn big() {{ /*{}*/ }}\n", big_body),
    );
    let asts = dump_json(root, false, true, usize::MAX).unwrap();
    // At least one node (likely root) should have text omitted due to length
    let any_none = asts[0].nodes.iter().any(|n| n.text.is_none());
    assert!(
        any_none,
        "expected at least one node with no text because span exceeded threshold"
    );
}

#[test]
fn deterministic_multiple_runs() {
    let td = TempDir::new().unwrap();
    let root = td.path();
    write(
        &root.join("Cargo.toml"),
        "[package]\nname='det'\nversion='0.0.0'\nedition='2021'\n",
    );
    write(&root.join("src/lib.rs"), "pub fn a() {} pub fn b() {} pub fn c() {}\n");
    let run1 = dump_json(root, false, false, usize::MAX).unwrap();
    let run2 = dump_json(root, false, false, usize::MAX).unwrap();
    assert_eq!(run1, run2, "dump_json output should be deterministic across runs");
    // Byte ranges should be coherent
    for file in run1 {
        for node in file.nodes.windows(2) {
            let a = &node[0];
            let b = &node[1];
            assert!(a.start_byte <= a.end_byte, "node start <= end");
            assert!(
                a.end_byte <= b.end_byte || a.start_byte <= b.start_byte,
                "non-decreasing progression"
            );
        }
    }
}