diff --git a/README.md b/README.md
index 1b74415..07b4432 100644
--- a/README.md
+++ b/README.md
@@ -8,7 +8,7 @@
- 
+ 
@@ -159,8 +159,8 @@ cd build && mingw32-make -j4
# 运行全部测试 (145 单元 + 23 集成)
./l_lang_lexer_test.exe # 词法分析 (41 tests)
./l_lang_test.exe # 语法分析 (15 tests)
-./l_lang_sema_test.exe # 语义分析 (65 tests)
-./l_lang_codegen_test.exe # 代码生成 (24 tests)
+./l_lang_sema_test.exe # 语义分析 (74 tests)
+./l_lang_codegen_test.exe # 代码生成 (28 tests)
# 集成测试
for f in ../test/programs/*.l; do
diff --git a/test/test_codegen.c b/test/test_codegen.c
index 7c6e4f4..0d36beb 100644
--- a/test/test_codegen.c
+++ b/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
@@ -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();
}
diff --git a/test/test_sema.c b/test/test_sema.c
index 6c3ed17..3215097 100644
--- a/test/test_sema.c
+++ b/test/test_sema.c
@@ -365,6 +365,59 @@ void test_method_undefined() {
arena_destroy(&a);
}
+/* === match 语义分析测试 === */
+
+void test_match_enum_sema_ok() {
+ Arena a = arena_create(1);
+ size_t tc; ErrorInfo lex_err = {0};
+ Token* toks = lex(&a,
+ "enum Color { Red, Green, Blue } fn main() -> i64 { let c = Color::Green; match c { Color::Red => { return 1; } Color::Green => { return 2; } _ => { return 0; } } return 0; }",
+ "test", &tc, &lex_err);
+ ASSERT(toks != NULL);
+ ErrorInfo parse_err = {0};
+ AstNode* ast = parse(&a, toks, tc, "test", &parse_err);
+ ASSERT(ast != NULL);
+
+ ErrorList errors; error_init(&errors);
+ sema_analyze(ast, &errors, &a);
+ ASSERT(errors.count == 0); // match 去糖后应通过类型检查
+ arena_destroy(&a);
+}
+
+void test_match_int_sema_ok() {
+ Arena a = arena_create(1);
+ size_t tc; ErrorInfo lex_err = {0};
+ Token* toks = lex(&a,
+ "fn main() -> i64 { let x: i64 = 42; match x { 1 => { return 10; } 42 => { return 20; } _ => { return 0; } } return 0; }",
+ "test", &tc, &lex_err);
+ ASSERT(toks != NULL);
+ ErrorInfo parse_err = {0};
+ AstNode* ast = parse(&a, toks, tc, "test", &parse_err);
+ ASSERT(ast != NULL);
+
+ ErrorList errors; error_init(&errors);
+ sema_analyze(ast, &errors, &a);
+ ASSERT(errors.count == 0);
+ arena_destroy(&a);
+}
+
+void test_match_wildcard_only_sema_ok() {
+ Arena a = arena_create(1);
+ size_t tc; ErrorInfo lex_err = {0};
+ Token* toks = lex(&a,
+ "fn main() -> i64 { let x: i64 = 42; match x { _ => { return 99; } } return 0; }",
+ "test", &tc, &lex_err);
+ ASSERT(toks != NULL);
+ ErrorInfo parse_err = {0};
+ AstNode* ast = parse(&a, toks, tc, "test", &parse_err);
+ ASSERT(ast != NULL);
+
+ ErrorList errors; error_init(&errors);
+ sema_analyze(ast, &errors, &a);
+ ASSERT(errors.count == 0); // 纯通配符 match 应通过
+ arena_destroy(&a);
+}
+
int main(void) {
TEST_RUN(test_type_error);
TEST_RUN(test_undefined_var);
@@ -387,5 +440,8 @@ int main(void) {
TEST_RUN(test_array_assign_ok);
TEST_RUN(test_method_call_ok);
TEST_RUN(test_method_undefined);
+ TEST_RUN(test_match_enum_sema_ok);
+ TEST_RUN(test_match_int_sema_ok);
+ TEST_RUN(test_match_wildcard_only_sema_ok);
return test_summary();
}