[Testing] Support --only in harness (#460)

Author: ZippeyKeys12

runtime/libcn/include/cn-smt/from_smt.h

@@ -15,7 +15,6 @@
 extern "C" {
 #endif
 
-// Forward declarations and types
 typedef const char* const_char_ptr;
 
 // Hash table for string -> sexp mapping (used in let-bindings)

runtime/libcn/src/cn-testing/test.c

@@ -9,6 +9,8 @@
 
 #include <bennet/prelude.h>
 #include <bennet/state/rand_alloc.h>
+#include <bennet/utils.h>
+#include <bennet/utils/hash_table.h>
 #include <cn-executable/utils.h>
 #include <cn-testing/result.h>
 #include <cn-testing/test.h>
@@ -138,6 +140,10 @@ void cn_trap(void) {
   _cn_trap();
 }
 
+typedef const char* const_char_ptr;
+BENNET_HASH_TABLE_DECL(const_char_ptr, uint8_t)
+BENNET_HASH_TABLE_IMPL(const_char_ptr, uint8_t)
+
 struct cn_test_reproduction {
   size_t size;
   bennet_rand_checkpoint checkpoint;
@@ -153,6 +159,12 @@ int cn_test_main(int argc, char* argv[]) {
   set_cn_logging_level(CN_LOGGING_NONE);
 
   bennet_srand(bennet_get_milliseconds());
+
+  // Initialize test filter hash table
+  bennet_hash_table(const_char_ptr, uint8_t) test_filter;
+  bennet_hash_table_init(const_char_ptr, uint8_t)(
+      &test_filter, string_hash, string_equal);
+
   enum cn_test_gen_progress progress_level = CN_TEST_GEN_PROGRESS_ALL;
   uint64_t seed = bennet_rand();
   enum cn_logging_level logging_level = CN_LOGGING_ERROR;
@@ -314,6 +326,32 @@ int cn_test_main(int argc, char* argv[]) {
     } else if (strcmp("--max-input-alloc", arg) == 0) {
       bennet_rand_alloc_set_mem_size(strtoul(argv[i + 1], NULL, 10));
       i++;
+    } else if (strcmp("--only", arg) == 0) {
+      char* test_names = argv[i + 1];
+      char* test_names_copy = strdup(test_names);
+      assert(test_names_copy != NULL);
+
+      // Parse comma-separated test names
+      char* token = strtok(test_names_copy, ",");
+      while (token != NULL) {
+        // Trim leading/trailing whitespace
+        while (*token == ' ' || *token == '\t')
+          token++;
+        char* end = token + strlen(token) - 1;
+        while (end > token && (*end == ' ' || *end == '\t'))
+          end--;
+        *(end + 1) = '\0';
+
+        // Add to filter (store the original token pointer)
+        char* stored_name = strdup(token);
+        assert(stored_name != NULL);
+        bennet_hash_table_set(const_char_ptr, uint8_t)(&test_filter, stored_name, true);
+
+        token = strtok(NULL, ",");
+      }
+
+      free(test_names_copy);
+      i++;
     }
   }
 
@@ -360,6 +398,14 @@ int cn_test_main(int argc, char* argv[]) {
         continue;
       }
 
+      // Skip tests not in the filter if filter is non-empty
+      if (bennet_hash_table_size(const_char_ptr, uint8_t)(&test_filter) > 0) {
+        if (!bennet_hash_table_contains(const_char_ptr, uint8_t)(
+                &test_filter, test_cases[i].name)) {
+          continue;
+        }
+      }
+
       if (output_tyche || print_size_info) {
         bennet_info_sizes_set_function_under_test(test_cases[i].name);
       }
@@ -524,5 +570,8 @@ outside_loop:;
     bennet_info_timing_print_info();
   }
 
+  // Clean up test filter hash table
+  bennet_hash_table_free(const_char_ptr, uint8_t)(&test_filter);
+
   return !(failed == 0 && errored == 0);
 }