mortar_compiler 0.5.3

//! # parser_test.rs
//!
//! # parser_test.rs 文件
//!
//! ## Module Overview
//!
//! ## 模块概述
//!
//! Comprehensive tests for the parser.
//!
//! 解析器的综合测试。
//!
//! ## Source File Overview
//!
//! ## 源文件概述
//!
//! Tests parsing of nodes, choices, events, functions, and other core language constructs.
//!
//! 测试节点、选项、事件、函数和其他核心语言构造的解析。

use crate::ast::{
    Arg, ChoiceDest, ChoiceItem, Condition, Event, EventAction, FuncCall, FunctionDecl, NodeDef,
    NodeJump, NodeStmt, Param, Program, TopLevel, WithEventItem, WithEventsStmt,
};
use crate::parser::ParseHandler;

fn check_parsing(source: &str, expected_program: Program) {
    let program = ParseHandler::parse_source_code(source, false).unwrap();
    assert_eq!(program, expected_program);
}

#[test]
fn test_parse_empty_source() {
    let source = "";
    let program = ParseHandler::parse_source_code(source, false).unwrap();
    assert_eq!(program.body.len(), 0);
}

#[test]
fn test_parse_node_def() {
    let source = r#"
        node start_node {
            text: "Hello, world!"
        } -> next_node
    "#;
    let expected = Program {
        body: vec![TopLevel::NodeDef(NodeDef {
            name: "start_node".to_string(),
            name_span: Some((14, 24)), // Approximate span for "start_node"
            body: vec![NodeStmt::Text("Hello, world!".to_string())],
            jump: Some(NodeJump::Identifier(
                "next_node".to_string(),
                Some((74, 83)),
            )), // Updated to actual span
        })],
    };
    check_parsing(source, expected);
}

#[test]
fn test_parse_function_decl() {
    let source = r#"
        fn my_function(param1: String, param2: Number) -> String
    "#;
    let expected = Program {
        body: vec![TopLevel::FunctionDecl(FunctionDecl {
            name: "my_function".to_string(),
            name_span: Some((12, 23)), // Updated to actual span for "my_function"
            params: vec![
                Param {
                    name: "param1".to_string(),
                    type_name: "String".to_string(),
                },
                Param {
                    name: "param2".to_string(),
                    type_name: "Number".to_string(),
                },
            ],
            return_type: Some("String".to_string()),
        })],
    };
    check_parsing(source, expected);
}

#[test]
fn test_parse_node_with_events() {
    let source = r#"
        node event_node {
            text: "Hello"
            with events: [
                0, say("hello")
                1.5, say("world").wait(1)
            ]
        }
    "#;
    let expected = Program {
        body: vec![TopLevel::NodeDef(NodeDef {
            name: "event_node".to_string(),
            name_span: Some((14, 24)), // Approximate span
            body: vec![
                NodeStmt::Text("Hello".to_string()),
                NodeStmt::WithEvents(WithEventsStmt {
                    events: vec![
                        WithEventItem::InlineEvent(Event {
                            index: 0.0,
                            action: EventAction {
                                call: FuncCall {
                                    name: "say".to_string(),
                                    name_span: Some((99, 102)),
                                    args: vec![Arg::String("hello".to_string())],
                                },
                                chains: vec![],
                            },
                        }),
                        WithEventItem::InlineEvent(Event {
                            index: 1.5,
                            action: EventAction {
                                call: FuncCall {
                                    name: "say".to_string(),
                                    name_span: Some((133, 136)),
                                    args: vec![Arg::String("world".to_string())],
                                },
                                chains: vec![FuncCall {
                                    name: "wait".to_string(),
                                    name_span: Some((146, 150)),
                                    args: vec![Arg::Number(1.0)],
                                }],
                            },
                        }),
                    ],
                }),
            ],
            jump: None,
        })],
    };
    check_parsing(source, expected);
}

#[test]
fn test_parse_node_with_choices() {
    let source = r#"
        node choice_node {
            choice: [
                "Choice 1" -> next_node,
                "Choice 2".when(is_ready) -> return,
                ("Choice 3").when(check(arg1)) -> break,
                "Choice 4" -> [
                    "Nested 1" -> nested_node
                ]
            ]
        }
    "#;
    let expected = Program {
        body: vec![TopLevel::NodeDef(NodeDef {
            name: "choice_node".to_string(),
            name_span: Some((14, 25)), // Approximate span
            body: vec![NodeStmt::Choice(vec![
                ChoiceItem {
                    text: "Choice 1".to_string(),
                    condition: None,
                    target: ChoiceDest::Identifier("next_node".to_string(), Some((80, 89))),
                },
                ChoiceItem {
                    text: "Choice 2".to_string(),
                    condition: Some(Condition::Identifier("is_ready".to_string())),
                    target: ChoiceDest::Return,
                },
                ChoiceItem {
                    text: "Choice 3".to_string(),
                    condition: Some(Condition::FuncCall(FuncCall {
                        name: "check".to_string(),
                        name_span: Some((178, 183)), // Updated to actual span
                        args: vec![Arg::Identifier("arg1".to_string())],
                    })),
                    target: ChoiceDest::Break,
                },
                ChoiceItem {
                    text: "Choice 4".to_string(),
                    condition: None,
                    target: ChoiceDest::NestedChoices(vec![ChoiceItem {
                        text: "Nested 1".to_string(),
                        condition: None,
                        target: ChoiceDest::Identifier("nested_node".to_string(), Some((267, 278))),
                    }]),
                },
            ])],
            jump: None,
        })],
    };
    check_parsing(source, expected);
}

#[test]
fn test_parse_multiple_top_level() {
    let source = r#"
        fn func1()
        node node1 {}
        fn func2()
        node node2 {}
    "#;
    let program = ParseHandler::parse_source_code(source, false).unwrap();
    assert_eq!(program.body.len(), 4);
    assert!(matches!(program.body[0], TopLevel::FunctionDecl(_)));
    assert!(matches!(program.body[1], TopLevel::NodeDef(_)));
    assert!(matches!(program.body[2], TopLevel::FunctionDecl(_)));
    assert!(matches!(program.body[3], TopLevel::NodeDef(_)));
}

#[test]
fn test_parse_text_with_escape_sequences() {
    let source = r#"
        node escape_test {
            text: "Line 1\nLine 2"
            text: "Tab\there"
            text: "Quote: \"test\""
            text: "Backslash: \\"
        }
    "#;
    let program = ParseHandler::parse_source_code(source, false).unwrap();

    if let TopLevel::NodeDef(node) = &program.body[0] {
        assert_eq!(node.name, "escape_test");

        // Check newline escape
        if let NodeStmt::Text(text) = &node.body[0] {
            assert_eq!(text, "Line 1\nLine 2");
        } else {
            panic!("Expected Text statement");
        }

        // Check tab escape
        if let NodeStmt::Text(text) = &node.body[1] {
            assert_eq!(text, "Tab\there");
        } else {
            panic!("Expected Text statement");
        }

        // Check quote escape
        if let NodeStmt::Text(text) = &node.body[2] {
            assert_eq!(text, "Quote: \"test\"");
        } else {
            panic!("Expected Text statement");
        }

        // Check backslash escape
        if let NodeStmt::Text(text) = &node.body[3] {
            assert_eq!(text, "Backslash: \\");
        } else {
            panic!("Expected Text statement");
        }
    } else {
        panic!("Expected NodeDef");
    }
}

#[test]
fn test_parse_choice_with_escape_sequences() {
    let source = r#"
        node choice_escape {
            choice: [
                "Option with\nnewline" -> return
            ]
        }
    "#;
    let program = ParseHandler::parse_source_code(source, false).unwrap();

    if let TopLevel::NodeDef(node) = &program.body[0] {
        if let NodeStmt::Choice(choices) = &node.body[0] {
            assert_eq!(choices[0].text, "Option with\nnewline");
        } else {
            panic!("Expected Choice statement");
        }
    } else {
        panic!("Expected NodeDef");
    }
}

#[test]
fn test_parse_triple_quoted_string() {
    let source = r#"
        node multiline_test {
            text: """
                Hello, world!
                This is line 2.
                And line 3.
            """
        }
    "#;
    let program = ParseHandler::parse_source_code(source, false).unwrap();

    if let TopLevel::NodeDef(node) = &program.body[0] {
        assert_eq!(node.name, "multiline_test");

        if let NodeStmt::Text(text) = &node.body[0] {
            // Should be dedented and joined with newlines
            assert_eq!(text, "Hello, world!\nThis is line 2.\nAnd line 3.");
        } else {
            panic!("Expected Text statement");
        }
    } else {
        panic!("Expected NodeDef");
    }
}

#[test]
fn test_parse_triple_quoted_string_with_escapes() {
    let source = r#"
        node escape_multiline {
            text: """
                Line with\ttab
                Line with "quotes"
            """
        }
    "#;
    let program = ParseHandler::parse_source_code(source, false).unwrap();

    if let TopLevel::NodeDef(node) = &program.body[0] {
        if let NodeStmt::Text(text) = &node.body[0] {
            // Tab escape should work, quotes should be literal
            assert_eq!(text, "Line with\ttab\nLine with \"quotes\"");
        } else {
            panic!("Expected Text statement");
        }
    } else {
        panic!("Expected NodeDef");
    }
}

#[test]
fn test_parse_line_stmt() {
    let source = r#"
        node eat_food {
            line: "You ate the food."
            line: "You recovered 10 HP!"
        }
    "#;
    let expected = Program {
        body: vec![TopLevel::NodeDef(NodeDef {
            name: "eat_food".to_string(),
            name_span: Some((14, 22)),
            body: vec![
                NodeStmt::Line("You ate the food.".to_string()),
                NodeStmt::Line("You recovered 10 HP!".to_string()),
            ],
            jump: None,
        })],
    };
    check_parsing(source, expected);
}

#[test]
fn test_parse_line_and_text_mixed() {
    let source = r#"
        node mixed {
            text: "Press Z..."
            line: "Line A"
            line: "Line B"
            text: "Done."
        }
    "#;
    let program = ParseHandler::parse_source_code(source, false).unwrap();
    let TopLevel::NodeDef(node) = &program.body[0] else {
        panic!("Expected NodeDef");
    };
    assert_eq!(node.body.len(), 4);
    assert!(matches!(&node.body[0], NodeStmt::Text(s) if s == "Press Z..."));
    assert!(matches!(&node.body[1], NodeStmt::Line(s) if s == "Line A"));
    assert!(matches!(&node.body[2], NodeStmt::Line(s) if s == "Line B"));
    assert!(matches!(&node.body[3], NodeStmt::Text(s) if s == "Done."));
}

#[test]
fn test_parse_line_triple_quoted() {
    let source = r#"
        node multi {
            line: """
                First paragraph
                with multiple lines.
            """
        }
    "#;
    let program = ParseHandler::parse_source_code(source, false).unwrap();
    let TopLevel::NodeDef(node) = &program.body[0] else {
        panic!("Expected NodeDef");
    };
    let NodeStmt::Line(text) = &node.body[0] else {
        panic!("Expected Line statement, got {:?}", node.body[0]);
    };
    assert_eq!(text, "First paragraph\nwith multiple lines.");
}