Adding more unit tests

Author: fjtrujy

GNU-gasp.pdf

@@ -71,7 +71,6 @@ static void print_diff_snippet(const char *out_path, const char *exp_path) {
       break;
     }
     if (strcmp(le, lo) != 0) {
-      // Trim trailing newlines for nicer printing
       size_t len_e = strlen(le); if (len_e && (le[len_e-1] == '\n' || le[len_e-1] == '\r')) le[len_e-1] = '\0';
       size_t len_o = strlen(lo); if (len_o && (lo[len_o-1] == '\n' || lo[len_o-1] == '\r')) lo[len_o-1] = '\0';
       fprintf(stderr, "L%05d | EXPECTED: %s\n", line, le);
@@ -81,9 +80,7 @@ static void print_diff_snippet(const char *out_path, const char *exp_path) {
     }
     line++;
   }
-  // If actual has extra lines
   if (shown == 0) {
-    // Check for extra lines when content so far matched
     if (fgets(lo, sizeof(lo), fo)) {
       size_t len_o = strlen(lo); if (len_o && (lo[len_o-1] == '\n' || lo[len_o-1] == '\r')) lo[len_o-1] = '\0';
       fprintf(stderr, "L%05d | EXPECTED: <EOF>\n", line);
@@ -94,6 +91,96 @@ static void print_diff_snippet(const char *out_path, const char *exp_path) {
   fclose(fe);
 }
 
+static int write_text_file(const char *path, const char *text) {
+  FILE *f = fopen(path, "wb");
+  if (!f) return -1;
+  size_t n = fwrite(text, 1, strlen(text), f);
+  fclose(f);
+  return (n == strlen(text)) ? 0 : -1;
+}
+
+static int read_file_to_buf(const char *path, char **out, size_t *out_len) {
+  *out = NULL; *out_len = 0;
+  FILE *f = fopen(path, "rb");
+  if (!f) return -1;
+  fseek(f, 0, SEEK_END);
+  long sz = ftell(f);
+  fseek(f, 0, SEEK_SET);
+  if (sz < 0) { fclose(f); return -1; }
+  char *buf = (char*)malloc((size_t)sz + 1);
+  if (!buf) { fclose(f); return -1; }
+  size_t n = fread(buf, 1, (size_t)sz, f);
+  fclose(f);
+  buf[n] = '\0';
+  *out = buf; *out_len = n;
+  return 0;
+}
+
+static int run_case(const char *name, const char *src_text, const char *must_contain) {
+  char src_path[1024];
+  char out_path[1024];
+#ifdef _WIN32
+  snprintf(src_path, sizeof(src_path), "%s\\test_outputs\\%s.vcl", BUILD_DIR, name);
+  snprintf(out_path, sizeof(out_path), "%s\\test_outputs\\%s.out", BUILD_DIR, name);
+#else
+  snprintf(src_path, sizeof(src_path), "%s/test_outputs/%s.vcl", BUILD_DIR, name);
+  snprintf(out_path, sizeof(out_path), "%s/test_outputs/%s.out", BUILD_DIR, name);
+#endif
+  if (write_text_file(src_path, src_text) != 0) {
+    fprintf(stderr, "Failed to write source %s\n", src_path);
+    return 1;
+  }
+
+  // Build argv
+  int rc;
+#if defined(_WIN32)
+  char masp_path[1024];
+  snprintf(masp_path, sizeof(masp_path), "%s\\src\\masp.exe", BUILD_DIR);
+  const char *argvp[] = { masp_path, "-p", "-s", "-c", ";", "-o", out_path, "--", src_path, NULL };
+  rc = _spawnv(_P_WAIT, masp_path, (const char* const*)argvp);
+  if (rc == -1) {
+    fprintf(stderr, "spawn failed for %s (errno=%d)\n", masp_path, errno);
+    return 1;
+  }
+#else
+  {
+    char masp_path[1024];
+    snprintf(masp_path, sizeof(masp_path), "%s/src/masp", BUILD_DIR);
+    pid_t pid = fork();
+    if (pid == 0) {
+      const char *argvp[] = { masp_path, "-p", "-s", "-c", ";", "-o", out_path, "--", src_path, NULL };
+      execv(masp_path, (char* const*)argvp);
+      _exit(127);
+    } else if (pid > 0) {
+      int status = 0;
+      if (waitpid(pid, &status, 0) < 0) { perror("waitpid"); return 1; }
+      if (WIFEXITED(status)) rc = WEXITSTATUS(status);
+      else { fprintf(stderr, "subprocess terminated abnormally\n"); return 1; }
+    } else { perror("fork"); return 1; }
+  }
+#endif
+  if (rc != 0) {
+    fprintf(stderr, "masp returned %d for %s\n", rc, name);
+    return 1;
+  }
+  struct stat st; if (stat(out_path, &st) != 0 || st.st_size <= 0) {
+    fprintf(stderr, "Output empty for %s: %s\n", name, out_path);
+    return 1;
+  }
+  if (must_contain && *must_contain) {
+    char *buf = NULL; size_t blen = 0;
+    if (read_file_to_buf(out_path, &buf, &blen) != 0) { fprintf(stderr, "Read failed %s\n", out_path); return 1; }
+    int ok = strstr(buf, must_contain) != NULL;
+    if (!ok) {
+      fprintf(stderr, "Expected substring not found for %s: '%s'\n", name, must_contain);
+      free(buf);
+      return 1;
+    }
+    free(buf);
+  }
+  return 0;
+}
+
 static int run_vu1Triangle(void) {
   char out_path[1024];
   char src_path[1024];
@@ -103,7 +190,6 @@ static int run_vu1Triangle(void) {
   snprintf(src_path, sizeof(src_path), "%s/test/vu1Triangle.vcl", SRC_DIR);
   snprintf(expected_path, sizeof(expected_path), "%s/test/vu1Triangle.vcl_masp", SRC_DIR);
 #ifdef _WIN32
-  // Normalize to backslashes for Windows APIs
   for (char *p = out_path; *p; ++p) if (*p == '/') *p = '\\';
   for (char *p = src_path; *p; ++p) if (*p == '/') *p = '\\';
   for (char *p = expected_path; *p; ++p) if (*p == '/') *p = '\\';
@@ -131,7 +217,6 @@ static int run_vu1Triangle(void) {
   } else
 #elif defined(__unix__)
   {
-    // Prefer a clean subprocess exec on UNIX to avoid sanitizer/locale clashes on CI
     char masp_path[1024];
     snprintf(masp_path, sizeof(masp_path), "%s/src/masp", BUILD_DIR);
     pid_t pid = fork();
@@ -150,7 +235,6 @@ static int run_vu1Triangle(void) {
       } else if (WIFSIGNALED(status)) {
         int sig = WTERMSIG(status);
         fprintf(stderr, "masp subprocess terminated by signal %d\n", sig);
-        // Fallback to in-process to surface better diagnostics
         const char *argv0[] = { "masp", "-p", "-s", "-c", ";", "-o", out_path, "--", src_path, NULL };
         int argc = 9;
         rc = masp_program_main(argc, (char**)argv0);
@@ -174,11 +258,7 @@ static int run_vu1Triangle(void) {
     fprintf(stderr, "masp_program_main returned %d\n", rc);
     return 1;
   }
-  struct stat st;
-  if (stat(out_path, &st) != 0) {
-    fprintf(stderr, "Output file not created: %s\n", out_path);
-    return 1;
-  }
+  struct stat st; if (stat(out_path, &st) != 0) { fprintf(stderr, "Output file not created: %s\n", out_path); return 1; }
   if (!files_equal(out_path, expected_path)) {
     struct stat st2; st2.st_size = -1; (void)stat(out_path, &st2);
     fprintf(stderr, "Output differs from expected: %s (size=%ld)\n", out_path, (long)st2.st_size);
@@ -188,9 +268,90 @@ static int run_vu1Triangle(void) {
   return 0;
 }
 
+static int run_basic_suite(void) {
+  int failed = 0;
+  // Ensure output dir exists
+  char out_dir[1024];
+#ifdef _WIN32
+  snprintf(out_dir, sizeof(out_dir), "%s\\test_outputs", BUILD_DIR);
+#else
+  snprintf(out_dir, sizeof(out_dir), "%s/test_outputs", BUILD_DIR);
+#endif
+  if (MKDIR(out_dir) != 0 && errno != EEXIST) {
+    fprintf(stderr, "Failed to create: %s (errno=%d)\n", out_dir, errno);
+    return 1;
+  }
+
+  // 1) .db outputs .byte
+  failed += run_case("db_bytes", ".db 1,2,3\n", ".byte\t1,2,3");
+  // 2) .dw outputs .short
+  failed += run_case("dw_short", ".dw 258\n", ".short\t258");
+  // 3) .dl outputs .long
+  failed += run_case("dl_long", ".dl 65539\n", ".long\t65539");
+  // 4) .assign substitution in data line
+  failed += run_case("assign_subst", "X .assign 3\n.db X\n", ".byte\t3");
+  // 5) trivial line is copied with comment when -s is used (first char ';')
+  failed += run_case("copysource", ".db 7\n", ";");
+  // 6) Macro define and expand: emits byte 5
+  failed += run_case(
+    "macro_expand",
+    ".macro M x\n .db \\x\n .endm\n M 5\n",
+    ".byte\t5");
+  // 7) Include file: create an included file that emits .db 42
+  {
+    char inc_path[1024];
+#ifdef _WIN32
+    snprintf(inc_path, sizeof(inc_path), "%s\\test_outputs\\inc_simple.vcl", BUILD_DIR);
+#else
+    snprintf(inc_path, sizeof(inc_path), "%s/test_outputs/inc_simple.vcl", BUILD_DIR);
+#endif
+    if (write_text_file(inc_path, ".db 42\n") != 0) {
+      fprintf(stderr, "Failed to write include file: %s\n", inc_path);
+      return failed + 1;
+    }
+    char src_text[1400];
+    snprintf(src_text, sizeof(src_text), ".include \"%s\"\n", inc_path);
+    failed += run_case("include_simple", src_text, ".byte\t42");
+  }
+  // 8) Multiple data on one line
+  failed += run_case("db_multi", ".db 10,11,12,13\n", ".byte\t10,11,12,13");
+  // 9) Base conversions: binary, hex, octal, decimal
+  failed += run_case("base_bin", ".db b'1010'\n", ".byte\t10");
+  failed += run_case("base_hex", ".db h'FF'\n", ".byte\t255");
+  failed += run_case("base_oct", ".db q'10'\n", ".byte\t8");
+  failed += run_case("base_dec", ".db d'12'\n", ".byte\t12");
+  // 10) Align directive
+  failed += run_case("align4", ".align 4\n", ".align\t4");
+  // 11) Conditional true branch
+  failed += run_case("aif_true", ".AIF 1\n.DB 9\n.AENDI\n", ".byte\t9");
+  // 12) Conditional else branch
+  failed += run_case("aif_false_else", ".AIF 0\n.DB 1\n.AELSE\n.DB 2\n.AENDI\n", ".byte\t2");
+  // 13) PRINT outputs string literal
+  failed += run_case("print_str", ".PRINT \"HELLO\"\n", "HELLO");
+  // 14) HEADING emits a .title line with the string
+  failed += run_case("heading", ".HEADING \"TITLE\"\n", ".title\t\"TITLE\"");
+  // 15) String in data
+  failed += run_case("db_string", ".db \"ABC\"\n", "ABC");
+  // 16) Page eject
+  failed += run_case("page", ".PAGE\n", ".eject");
+  // 17) Export emits .global
+  failed += run_case("export_global", ".EXPORT foo\n", ".global\tfoo");
+  // 18) Repeat block
+  failed += run_case("arepeat", ".AREPEAT 3\n.DB 1\n.AENDR\n", ".byte\t1");
+  // 19) Reserve space
+  failed += run_case("res_space", ".RES 4\n", ".space");
+  failed += run_case("sres_space", ".SRES 2\n", ".space");
+  // 20) Print listing toggles
+  failed += run_case("print_list_toggle", ".PRINT LIST\n", ".list");
+  failed += run_case("print_nolist_toggle", ".PRINT NOLIST\n", ".nolist");
+
+  return failed;
+}
+
 int main(void) {
   int failures = 0;
   failures += run_vu1Triangle();
+  failures += run_basic_suite();
   if (failures) {
     fprintf(stderr, "Unit tests failed: %d\n", failures);
     return 1;