Function test_parse_function

aiscript-vm/src/parser/stmt_test.rs:13–96

()

    #[test]
    fn test_parse_function() {
        rootless_mutate(|mutation| {
            let context = Context {
                mutation,
                strings: InternedStringSet::new(mutation),
            };
            let source = r#"
                fn test1() {
                    """
                    test1 doc
                    """
                    print("Hello, world!");
                }

                fn test2(a: int, b: int) -> int {
                    return a + b;
                }

                fn test3(a, b,) -> int {
                    return a + b;
                }
            "#;
            let mut parser = Parser::new(context, source);
            let result = parser.parse().unwrap();
            let Stmt::Function(FunctionDecl {
                name,
                mangled_name,
                doc,
                params,
                return_type,
                body,
                line,
                ..
            }) = &result.statements[0]
            else {
                panic!("Unexpected statement type");
            };
            assert_eq!(name.lexeme, "test1");
            assert_eq!(mangled_name, "script$test1");
            assert_eq!(doc.unwrap().lexeme, "test1 doc");
            assert_eq!(params.len(), 0);
            assert!(return_type.is_none());
            assert_eq!(body.len(), 1);
            assert_eq!(line, &2);

            let Stmt::Function(FunctionDecl {
                name,
                mangled_name,
                doc,
                params,
                return_type,
                ..
            }) = &result.statements[1]
            else {
                panic!("Unexpected statement type");
            };
            assert_eq!(name.lexeme, "test2");
            assert_eq!(mangled_name, "script$test2");