vb6parse 1.2.4

vb6parse is a library for parsing and analyzing VB6 code, from projects, to controls, to modules, and forms.
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120

//! Tests for VB6 label statements and their interaction with control flow.
//!
//! Labels in VB6 are identifiers followed by a colon, used primarily with `GoTo` and `GoSub`.
//! This test file verifies that labels are properly parsed and that End statements
//! immediately following labels are correctly recognized.

use vb6parse::*;

const SNAPSHOT_PATH: &str = "../../snapshots/parsers/cst/control_flow/labels";

#[test]
fn label_followed_by_end_function() {
    let source = r"
Function TestFunction() As Long
    TestFunction = 1
    Exit Function
ExitEnd:
End Function
";
    let (cst_opt, failures) = ConcreteSyntaxTree::from_text("test.bas", source).unpack();
    assert_eq!(failures.len(), 0, "Expected no parse failures.");
    let cst = cst_opt.expect("CST should be parsed");
    let tree = cst.to_serializable();

    let text = format!("{tree:#?}");
    assert!(
        !text.contains("Unknown"),
        "Should not contain Unknown tokens, but found:\n{text}"
    );

    let mut settings = insta::Settings::clone_current();
    settings.set_snapshot_path(SNAPSHOT_PATH);
    settings.set_prepend_module_to_snapshot(false);
    let _guard = settings.bind_to_scope();
    insta::assert_yaml_snapshot!(tree);
}

#[test]
fn label_followed_by_end_sub() {
    let source = r"
Sub TestSub()
    Exit Sub
CleanupLabel:
End Sub
";
    let (cst_opt, failures) = ConcreteSyntaxTree::from_text("test.bas", source).unpack();
    assert_eq!(failures.len(), 0, "Expected no parse failures.");
    let cst = cst_opt.expect("CST should be parsed");
    let tree = cst.to_serializable();

    let text = format!("{tree:#?}");
    assert!(
        !text.contains("Unknown"),
        "Should not contain Unknown tokens"
    );

    let mut settings = insta::Settings::clone_current();
    settings.set_snapshot_path(SNAPSHOT_PATH);
    settings.set_prepend_module_to_snapshot(false);
    let _guard = settings.bind_to_scope();
    insta::assert_yaml_snapshot!(tree);
}

#[test]
fn label_followed_by_end_property() {
    let source = r#"
Property Get TestProperty() As String
    TestProperty = "test"
    Exit Property
ErrorHandler:
End Property
"#;
    let (cst_opt, failures) = ConcreteSyntaxTree::from_text("test.bas", source).unpack();
    assert_eq!(failures.len(), 0, "Expected no parse failures.");
    let cst = cst_opt.expect("CST should be parsed");
    let tree = cst.to_serializable();

    let text = format!("{tree:#?}");
    assert!(
        !text.contains("Unknown"),
        "Should not contain Unknown tokens"
    );

    let mut settings = insta::Settings::clone_current();
    settings.set_snapshot_path(SNAPSHOT_PATH);
    settings.set_prepend_module_to_snapshot(false);
    let _guard = settings.bind_to_scope();
    insta::assert_yaml_snapshot!(tree);
}

#[test]
fn multiple_labels_in_function() {
    let source = r"
Function MultiLabel() As Integer
    On Error GoTo ErrorHandler
    MultiLabel = 1
    Exit Function
ErrorHandler:
    MultiLabel = -1
    GoTo CleanupLabel
CleanupLabel:
End Function
";
    let (cst_opt, failures) = ConcreteSyntaxTree::from_text("test.bas", source).unpack();
    assert_eq!(failures.len(), 0, "Expected no parse failures.");
    let cst = cst_opt.expect("CST should be parsed");
    let tree = cst.to_serializable();

    let text = format!("{tree:#?}");
    assert!(
        !text.contains("Unknown"),
        "Should not contain Unknown tokens"
    );

    let mut settings = insta::Settings::clone_current();
    settings.set_snapshot_path(SNAPSHOT_PATH);
    settings.set_prepend_module_to_snapshot(false);
    let _guard = settings.bind_to_scope();
    insta::assert_yaml_snapshot!(tree);
}