diff options
Diffstat (limited to 'tests/codegen_test.c')
| -rw-r--r-- | tests/codegen_test.c | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/tests/codegen_test.c b/tests/codegen_test.c index 1da711c..c92dfc2 100644 --- a/tests/codegen_test.c +++ b/tests/codegen_test.c @@ -100,6 +100,27 @@ static void test_generate_if_else(struct TestContext* context) free_program(&program); } +static void test_generate_while(struct TestContext* context) +{ + struct Lexer lexer = create_lexer( + "proc main\n{\nwhile rcx > 0\n{\nrbx += rcx\nrcx -= 1\n}\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, ".while_0:") != NULL); + check(context, strstr(buffer, "cmp rcx, 0") != NULL); + check(context, strstr(buffer, "jle .while_end_0") != NULL); + check(context, strstr(buffer, "add rbx, rcx") != NULL); + check(context, strstr(buffer, "jmp .while_0") != NULL); + check(context, strstr(buffer, ".while_end_0:") != NULL); + check(context, strstr(buffer, "; TODO") == NULL); + + free_program(&program); +} + static void test_generate_negative(struct TestContext* context) { struct Lexer lexer = create_lexer( @@ -485,6 +506,7 @@ void run_codegen_tests(struct TestContext* context) test_generate_text(context); test_generate_if(context); test_generate_if_else(context); + test_generate_while(context); test_generate_negative(context); test_generate_call(context); test_generate_param_substitution(context); |