1 files changed, 17 insertions, 0 deletions
diff --git a/build.c b/build.c
index 56c4d74..ec698c8 100644
--- a/build.c
+++ b/build.c
@@ -36,6 +36,23 @@ int main(int argc, char** argv) {
     }
   }
 
+  if (argc > 1) {
+    if (strcmp(argv[1], "test") == 0) {
+      printf("Test mode\n");
+      if (fork() == 0) {
+        char* args[] = {"gcc", "tests/main.c", "-o", "test.o", NULL};
+        execvp(args[0], args);
+        return 0;
+      } else {
+        wait(NULL);
+      }
+
+      char* args[] = {"./test.o", NULL};
+      execvp(args[0], args);
+      return 0;
+    }
+  }
+
   if (fork() == 0) {
     char* args[] = {"gcc", "src/main.c", "-o", "main.o", NULL};
     execvp(args[0], args);