spoofed_criterion: support init functions

immunant · Jul 24, 2024 · 97a8651 · 97a8651
1 parent 50d4fdf
commit 97a8651
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/misc/spoofed_criterion/include/criterion/criterion.h b/misc/spoofed_criterion/include/criterion/criterion.h
@@ -3,6 +3,7 @@
 
 struct fake_criterion_test {
   void (*test)(void);
+  void (*init)(void);
   int exit_code;
 };
 

diff --git a/misc/spoofed_criterion/test_runner.c b/misc/spoofed_criterion/test_runner.c
@@ -16,6 +16,9 @@ int main() {
     pid_t pid = fork();
     bool in_child = pid == 0;
     if (in_child) {
+      if (test_info->init) {
+        (*test_info->init)();
+      }
       (*test_info->test)();
       return 0;
     }