30 lines
1.0 KiB
Rust
30 lines
1.0 KiB
Rust
|
use crate::parser::NyashParser;
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn parse_peek_with_block_arm() {
|
||
|
|
let src = r#"
|
||
|
|
local x = 2
|
||
|
|
local y = peek x {
|
||
|
|
1 => { local a = 10 a }
|
||
|
|
2 => { 20 }
|
||
|
|
else => { 30 }
|
||
|
|
}
|
||
|
|
"#;
|
||
|
|
let ast = NyashParser::parse_from_string(src).expect("parse ok");
|
||
|
|
// Quick structural check: ensure AST contains PeekExpr and Program nodes inside arms
|
||
|
|
fn find_peek(ast: &crate::ast::ASTNode) -> bool {
|
||
|
|
match ast {
|
||
|
|
crate::ast::ASTNode::PeekExpr { arms, else_expr, .. } => {
|
||
|
|
// Expect at least one Program arm
|
||
|
|
let has_block = arms.iter().any(|(_, e)| matches!(e, crate::ast::ASTNode::Program { .. }));
|
||
|
|
let else_is_block = matches!(**else_expr, crate::ast::ASTNode::Program { .. });
|
||
|
|
has_block && else_is_block
|
||
|
|
}
|
||
|
|
crate::ast::ASTNode::Program { statements, .. } => statements.iter().any(find_peek),
|
||
|
|
_ => false,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
assert!(find_peek(&ast), "expected peek with block arms in AST");
|
||
|
|
}
|
||
|
|
|