diff --git a/src/parser/tests.rs b/src/parser/tests.rs index 508a2c1..0ed23b0 100644 --- a/src/parser/tests.rs +++ b/src/parser/tests.rs @@ -42,6 +42,24 @@ fn test_parse_no_function_context() { assert!(tree.is_err()) } +#[test] +fn test_parse_multiple_functions() { + let raw = " + fn foo() { + let x = 2; + return x; + } + + fn bar() { + let y = 5; + return y; + } + "; + let tokens = tokenize(raw); + let tree = parse(tokens); + assert!(tree.is_ok()) +} + #[test] fn test_parse_variable_declaration() { let raw = "