hakorune/src/tests/parser_static_box_members.rs

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"]
    ));
}