test: match sema + codegen 单元测试，总断言 145→158

test/test_codegen.c

@@ -1,5 +1,6 @@
 #include "test_utils.h"
 #include "codegen.h"
+#include "sema.h"
 #include "ast.h"
 #include "arena.h"
 #include <llvm-c/Analysis.h>
@@ -418,6 +419,97 @@ void test_codegen_method_call() {
     arena_destroy(&a);
 }
 
+/* === match 代码生成测试 === */
+
+void test_codegen_match() {
+    Arena a = arena_create(1);
+
+    /* enum Color { Red, Green, Blue } */
+    const char* variants[] = {"Red", "Green", "Blue"};
+    AstNode* enum_decl = ast_make_enum_decl(&a, "Color", variants, 3, loc_at(1, 1));
+    AstNode* enums[] = { enum_decl };
+
+    /* fn main() -> i64 {
+         let c = Color::Green;
+         match c {
+           Color::Red => { return 1; }
+           Color::Green => { return 2; }
+           _ => { return 0; }
+         }
+         return 0;
+       }
+       但 match 在 parser 去糖, 这里直接构造去糖后的 AST:
+       { let __match_val = Color::Green;
+         if __match_val == Color::Red { return 1; }
+         else if __match_val == Color::Green { return 2; }
+         else { return 0; }
+       }
+    */
+    AstNode* match_val = ast_make_enum_variant(&a, "Color", "Green", loc_at(1, 1));
+    match_val->type.kind = TYPE_ENUM;
+
+    AstNode* let_stmt = ast_make_let(&a, "__match_val", TYPE_UNKNOWN, false, false,
+                                     match_val, NULL, 0, NULL, 0, loc_at(1, 1));
+    let_stmt->type.kind = TYPE_ENUM;
+
+    AstNode* match_ident = ast_make_ident(&a, "__match_val", loc_at(1, 1));
+    match_ident->type.kind = TYPE_ENUM;
+
+    // if __match_val == Color::Red { return 1; }
+    AstNode* red_variant = ast_make_enum_variant(&a, "Color", "Red", loc_at(1, 1));
+    red_variant->type.kind = TYPE_ENUM;
+    AstNode* red_cond = ast_make_binary(&a, OP_EQ, match_ident, red_variant, loc_at(1, 1));
+    red_cond->type.kind = TYPE_BOOL;
+    AstNode* red_ret = ast_make_return(&a, ast_make_literal_i64(&a, 1, loc_at(1, 1)), loc_at(1, 1));
+    AstNode* red_stmts[] = { red_ret };
+    AstNode* red_block = ast_make_block(&a, red_stmts, 1, loc_at(1, 1));
+
+    // if __match_val == Color::Green { return 2; }
+    AstNode* match_ident2 = ast_make_ident(&a, "__match_val", loc_at(1, 1));
+    match_ident2->type.kind = TYPE_ENUM;
+    AstNode* green_variant = ast_make_enum_variant(&a, "Color", "Green", loc_at(1, 1));
+    green_variant->type.kind = TYPE_ENUM;
+    AstNode* green_cond = ast_make_binary(&a, OP_EQ, match_ident2, green_variant, loc_at(1, 1));
+    green_cond->type.kind = TYPE_BOOL;
+    AstNode* green_ret = ast_make_return(&a, ast_make_literal_i64(&a, 2, loc_at(1, 1)), loc_at(1, 1));
+    AstNode* green_stmts[] = { green_ret };
+    AstNode* green_block = ast_make_block(&a, green_stmts, 1, loc_at(1, 1));
+
+    // else { return 0; }
+    AstNode* else_ret = ast_make_return(&a, ast_make_literal_i64(&a, 0, loc_at(1, 1)), loc_at(1, 1));
+    AstNode* else_stmts[] = { else_ret };
+    AstNode* else_block = ast_make_block(&a, else_stmts, 1, loc_at(1, 1));
+
+    // if-else 链: if red_cond { red_block } else if green_cond { green_block } else { else_block }
+    AstNode* inner_if = ast_make_if(&a, green_cond, green_block, else_block, loc_at(1, 1));
+    AstNode* outer_if = ast_make_if(&a, red_cond, red_block, inner_if, loc_at(1, 1));
+
+    AstNode* main_stmts[] = { let_stmt, outer_if };
+    AstNode* main_body = ast_make_block(&a, main_stmts, 2, loc_at(1, 1));
+    AstNode* main_fn = ast_make_function(&a, "main", NULL, 0, TYPE_I64, NULL, main_body, loc_at(1, 1));
+
+    AstNode* fns[] = { main_fn };
+    AstNode* prog = ast_make_program(&a, fns, 1, NULL, 0, NULL, 0, enums, 1, NULL, 0, loc_at(1, 1));
+
+    ErrorList errors; error_init(&errors);
+    sema_analyze(prog, &errors, &a);
+    ASSERT(errors.count == 0);
+
+    const char* err = NULL;
+    LLVMContextRef ctx = NULL;
+    LLVMModuleRef mod = codegen_module(prog, &a, "test_match", &err, &ctx);
+    ASSERT(mod != NULL);
+    ASSERT(err == NULL);
+
+    char* verify_err = NULL;
+    int failed = LLVMVerifyModule(mod, LLVMReturnStatusAction, &verify_err);
+    ASSERT(!failed);
+
+    LLVMDisposeModule(mod);
+    LLVMContextDispose(ctx);
+    arena_destroy(&a);
+}
+
 int main(void) {
     TEST_RUN(test_codegen_simple_function);
     TEST_RUN(test_codegen_if_else);
@@ -428,5 +520,6 @@ int main(void) {
     TEST_RUN(test_codegen_enum);
     TEST_RUN(test_codegen_array);
     TEST_RUN(test_codegen_method_call);
+    TEST_RUN(test_codegen_match);
     return test_summary();
 }