2 files changed, 57 insertions, 0 deletions

diff --git a/tests/codegen_test.c b/tests/codegen_test.c
index 2b5fdb7..0a4a841 100644
--- a/tests/codegen_test.c
+++ b/tests/codegen_test.c
@@ -104,6 +104,41 @@ static void test_generate_fasm(struct TestContext* context)
 	free_program(&program);
 }
 
+static void test_generate_select(struct TestContext* context)
+{
+	// destination distinct from both sources: default-move else, then cmov
+	struct Lexer lexer = create_lexer("proc main\n{\nrax = rbx if rcx < rdx else rsi\n}\n");
+	struct Program program;
+	check(context, parse_program(&lexer, &program));
+
+	char buffer[1024];
+	generate_to_buffer(&program, buffer, sizeof(buffer));
+
+	check(context, strstr(buffer, "mov rax, rsi") != NULL);
+	check(context, strstr(buffer, "cmp rcx, rdx") != NULL);
+	check(context, strstr(buffer, "cmovl rax, rbx") != NULL);
+	check(context, strstr(buffer, "; TODO") == NULL);
+
+	free_program(&program);
+}
+
+static void test_generate_select_aarch64(struct TestContext* context)
+{
+	struct Lexer lexer = create_lexer(
+		"[enable: logical_registers]\nproc main\n{\nr1 = r2 if r3 < r4 else r5\n}\n");
+	struct Program program;
+	check(context, parse_program(&lexer, &program));
+
+	char buffer[1024];
+	generate_aarch64_to_buffer(&program, buffer, sizeof(buffer));
+
+	check(context, strstr(buffer, "cmp x2, x3") != NULL);
+	check(context, strstr(buffer, "csel x0, x1, x4, lt") != NULL);
+	check(context, strstr(buffer, "; TODO") == NULL);
+
+	free_program(&program);
+}
+
 static void test_generate_aarch64(struct TestContext* context)
 {
 	// logical registers map r1 -> x0, r2 -> x1; three-operand arithmetic and svc
@@ -633,6 +668,8 @@ void run_codegen_tests(struct TestContext* context)
 	test_generate_consts_and_data(context);
 	test_generate_fasm(context);
 	test_generate_aarch64(context);
+	test_generate_select(context);
+	test_generate_select_aarch64(context);
 	test_generate_text(context);
 	test_generate_if(context);
 	test_generate_if_else(context);
diff --git a/tests/parser_test.c b/tests/parser_test.c
index 9db85ec..eecc9b9 100644
--- a/tests/parser_test.c
+++ b/tests/parser_test.c
@@ -246,6 +246,25 @@ static void test_parse_while(struct TestContext* context)
 	free_program(&program);
 }
 
+static void test_parse_select(struct TestContext* context)
+{
+	struct Lexer lexer = create_lexer("proc main\n{\nr1 = r2 if r3 < r4 else r5\n}\n");
+	struct Program program;
+
+	check(context, parse_program(&lexer, &program));
+
+	struct Statement s = program.procs[0].body[0];
+	check(context, s.kind == STATEMENT_SELECT);
+	check(context, text_is(s.select.target, "r1"));
+	check(context, primary_is(s.select.if_value, "r2"));
+	check(context, primary_is(s.select.left, "r3"));
+	check(context, text_is(s.select.comparison, "<"));
+	check(context, primary_is(s.select.right, "r4"));
+	check(context, primary_is(s.select.else_value, "r5"));
+
+	free_program(&program);
+}
+
 static void test_parse_while_named(struct TestContext* context)
 {
 	struct Lexer lexer = create_lexer("proc main\n{\nwhile .drain rcx > 0\n{\nrcx -= 1\n}\n}\n");
@@ -435,6 +454,7 @@ void run_parser_tests(struct TestContext* context)
 	test_parse_else_if(context);
 	test_parse_while(context);
 	test_parse_while_named(context);
+	test_parse_select(context);
 	test_parse_call(context);
 	test_parse_stack(context);
 	test_parse_sized_deref(context);