2 files changed, 66 insertions, 0 deletions

diff --git a/tests/codegen_test.c b/tests/codegen_test.c
index 5236268..ab771f7 100644
--- a/tests/codegen_test.c
+++ b/tests/codegen_test.c
@@ -275,8 +275,56 @@ static void test_generate_no_entry(struct TestContext* context)
 	free_program(&program);
 }
 
+static void test_generate_logical_registers(struct TestContext* context)
+{
+	struct Lexer lexer = create_lexer(
+		"[enable: logical_registers]\nproc main\n{\nr1 = 5\nr4 = r10\nr6 = r1.8\n}\n");
+	struct Program program;
+	check(context, parse_program(&lexer, &program));
+
+	FILE* out = tmpfile();
+	generate_nasm(&program, out);
+	fflush(out);
+	rewind(out);
+
+	char buffer[1024];
+	size_t read = fread(buffer, 1, sizeof(buffer) - 1, out);
+	buffer[read] = '\0';
+	fclose(out);
+
+	check(context, strstr(buffer, "mov rax, 5") != NULL);    // r1 -> rax
+	check(context, strstr(buffer, "mov rdx, r11") != NULL);  // r4 -> rdx, r10 -> r11
+	check(context, strstr(buffer, "mov rdi, al") != NULL);   // r6 -> rdi, r1.8 -> al
+
+	free_program(&program);
+}
+
+static void test_generate_logical_disabled(struct TestContext* context)
+{
+	struct Lexer lexer = create_lexer("proc main\n{\nr1 = 5\n}\n");
+	struct Program program;
+	check(context, parse_program(&lexer, &program));
+
+	FILE* out = tmpfile();
+	generate_nasm(&program, out);
+	fflush(out);
+	rewind(out);
+
+	char buffer[1024];
+	size_t read = fread(buffer, 1, sizeof(buffer) - 1, out);
+	buffer[read] = '\0';
+	fclose(out);
+
+	// without the extension, r1 is passed through untouched
+	check(context, strstr(buffer, "mov r1, 5") != NULL);
+
+	free_program(&program);
+}
+
 void run_codegen_tests(struct TestContext* context)
 {
+	test_generate_logical_registers(context);
+	test_generate_logical_disabled(context);
 	test_generate_entry_and_bits(context);
 	test_generate_no_entry(context);
 	test_generate_consts_and_data(context);
diff --git a/tests/parser_test.c b/tests/parser_test.c
index 3b60261..3fa3342 100644
--- a/tests/parser_test.c
+++ b/tests/parser_test.c
@@ -234,6 +234,23 @@ static void test_parse_bad_directive(struct TestContext* context)
 	free_program(&program);
 }
 
+static void test_parse_register_size_suffix(struct TestContext* context)
+{
+	struct Lexer lexer = create_lexer("proc main\n{\nr1 = r2.8\n}\n");
+	struct Program program;
+
+	check(context, parse_program(&lexer, &program));
+
+	struct Expr* value = program.procs[0].body[0].assign.value;
+	check(context, value->kind == EXPR_MEMBER);
+	check(context, value->member.member.type == TOKEN_INTEGER);
+	check(context, text_is(value->member.member, "8"));
+	check(context, value->member.object->kind == EXPR_PRIMARY);
+	check(context, text_is(value->member.object->primary.token, "r2"));
+
+	free_program(&program);
+}
+
 static void test_parse_errors(struct TestContext* context)
 {
 	struct Program program;
@@ -265,5 +282,6 @@ void run_parser_tests(struct TestContext* context)
 	test_parse_sized_deref(context);
 	test_parse_directives(context);
 	test_parse_bad_directive(context);
+	test_parse_register_size_suffix(context);
 	test_parse_errors(context);
 }