66 lines
2.0 KiB
Rust
66 lines
2.0 KiB
Rust
|
use crate::parser::NyashParser;
|
||
|
|
|
||
|
|
fn parse(src: &str) -> crate::ast::ASTNode { NyashParser::parse_from_string(src).expect("parse ok") }
|
||
|
|
|
||
|
|
fn no_toplevel_funccall(ast: &crate::ast::ASTNode) -> bool {
|
||
|
|
match ast {
|
||
|
|
crate::ast::ASTNode::Program { statements, .. } => {
|
||
|
|
!statements.iter().any(|n| matches!(n, crate::ast::ASTNode::FunctionCall { .. }))
|
||
|
|
}
|
||
|
|
_ => true,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
fn box_has_methods(ast: &crate::ast::ASTNode, box_name: &str, methods: &[&str]) -> bool {
|
||
|
|
fn check_box(b: &crate::ast::ASTNode, box_name: &str, methods: &[&str]) -> bool {
|
||
|
|
if let crate::ast::ASTNode::BoxDeclaration { name, methods: m, is_static, .. } = b {
|
||
|
|
if name == box_name && *is_static {
|
||
|
|
return methods.iter().all(|k| {
|
||
|
|
if let Some(node) = m.get(*k) {
|
||
|
|
matches!(node, crate::ast::ASTNode::FunctionDeclaration { name, .. } if name == *k)
|
||
|
|
} else { false }
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|
||
|
|
false
|
||
|
|
}
|
||
|
|
match ast {
|
||
|
|
crate::ast::ASTNode::Program { statements, .. } => statements.iter().any(|n| check_box(n, box_name, methods)),
|
||
|
|
_ => false,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn static_box_methods_no_stray_call_compact() {
|
||
|
|
let src = r#"
|
||
|
|
static box S {
|
||
|
|
f(a) { return a }
|
||
|
|
g(b) { return b }
|
||
|
|
}
|
||
|
|
"#;
|
||
|
|
let ast = parse(src);
|
||
|
|
assert!(no_toplevel_funccall(&ast), "no top-level FunctionCall expected");
|
||
|
|
assert!(box_has_methods(&ast, "S", &["f", "g"]), "static box S should have f and g methods");
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn static_box_methods_no_stray_call_newline_seams() {
|
||
|
|
// Newlines between ) and {, and tight seam between } and next method head
|
||
|
|
let src = r#"
|
||
|
|
static box S {
|
||
|
|
parse_float(s)
|
||
|
|
{
|
||
|
|
return s
|
||
|
|
}
|
||
|
|
is_empty_or_whitespace(s)
|
||
|
|
{
|
||
|
|
return 0
|
||
|
|
}
|
||
|
|
}
|
||
|
|
"#;
|
||
|
|
let ast = parse(src);
|
||
|
|
assert!(no_toplevel_funccall(&ast), "no top-level FunctionCall expected at seams");
|
||
|
|
assert!(box_has_methods(&ast, "S", &["parse_float", "is_empty_or_whitespace"]));
|
||
|
|
}
|
||
|
|
|