1 files changed, 11 insertions, 3 deletions
diff --git a/tests/test_as_tokenizer.c b/tests/test_as_tokenizer.c
index 7d82099..995c764 100644
--- a/tests/test_as_tokenizer.c
+++ b/tests/test_as_tokenizer.c
@@ -26,7 +26,8 @@ char *expected_output =
     "OP: div, line: 4, col: 5\n"
     "NEWLINE\n"
     "OP: eq, line: 5, col: 5\n"
-    "NEWLINE\n";
+    "NEWLINE\n"
+    "ENDOFFILE\n";
 
 int main(int argc, char** argv) {
     printf("[TEST] assembler tokenizer\n");
@@ -40,8 +41,15 @@ int main(int argc, char** argv) {
     FILE *out = fmemopen(output_buffer, 10240, "w");
     FILE *origin_stdout = stdout;
     stdout = out;
-    while (peek_token(alct, ts)->type != TK_ENDOFFILE) {
-        print_token(peek_token(alct, ts));
+    struct token* token;
+    struct result result;
+    while (1) {
+        result = peek_token(alct, ts);
+        assert(result.errmsg == NULL);
+        assert(result.value != NULL);
+        token = result.value;
+        print_token(token);
+        if (token->type == TK_ENDOFFILE) break;
         next_token(alct, ts);
     }
     fclose(out);