1 files changed, 37 insertions, 0 deletions

diff --git a/tests/codegen_test.c b/tests/codegen_test.c
index db23791..e2969db 100644
--- a/tests/codegen_test.c
+++ b/tests/codegen_test.c
@@ -100,6 +100,23 @@ static void test_generate_if_else(struct TestContext* context)
 	free_program(&program);
 }
 
+static void test_generate_negative(struct TestContext* context)
+{
+	struct Lexer lexer = create_lexer(
+		"const N = -8\nproc main\n{\nrax = -5\nrbx = rax + -3\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, "%define N (-8)") != NULL);
+	check(context, strstr(buffer, "mov rax, -5") != NULL);
+	check(context, strstr(buffer, "add rbx, -3") != NULL);
+
+	free_program(&program);
+}
+
 static void test_generate_call(struct TestContext* context)
 {
 	struct Lexer lexer = create_lexer(
@@ -344,6 +361,24 @@ static void test_generate_load(struct TestContext* context)
 	free_program(&program);
 }
 
+static void test_generate_signed_load(struct TestContext* context)
+{
+	struct Lexer lexer = create_lexer(
+		"proc main\n{\nrax = ^signed byte rsi\nrbx = ^signed word rsi\nrcx = ^signed dword 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, "movsx rax, byte [rsi]") != NULL);
+	check(context, strstr(buffer, "movsx rbx, word [rsi]") != NULL);
+	check(context, strstr(buffer, "movsxd rcx, dword [rsi]") != NULL);
+	check(context, strstr(buffer, "; TODO") == NULL);
+
+	free_program(&program);
+}
+
 static void test_generate_enum_struct(struct TestContext* context)
 {
 	struct Lexer lexer = create_lexer(
@@ -441,6 +476,7 @@ void run_codegen_tests(struct TestContext* context)
 	test_generate_add_zero_peephole(context);
 	test_generate_enum_struct(context);
 	test_generate_load(context);
+	test_generate_signed_load(context);
 	test_generate_logical_registers(context);
 	test_generate_logical_disabled(context);
 	test_generate_entry_and_bits(context);
@@ -449,6 +485,7 @@ void run_codegen_tests(struct TestContext* context)
 	test_generate_text(context);
 	test_generate_if(context);
 	test_generate_if_else(context);
+	test_generate_negative(context);
 	test_generate_call(context);
 	test_generate_param_substitution(context);
 	test_generate_divide(context);