1 files changed, 96 insertions, 0 deletions
diff --git a/source/kit_test/test.h b/source/kit_test/test.h
new file mode 100644
index 0000000..da6c59b
--- /dev/null
+++ b/source/kit_test/test.h
@@ -0,0 +1,96 @@
+#ifndef KIT_TEST_TEST_H
+#define KIT_TEST_TEST_H
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+#include <stdbool.h>
+#include <stddef.h>
+#include <string.h>
+
+#ifndef KIT_TEST_FILE
+#  define kit_test
+#endif
+
+#ifndef KIT_TESTS_SIZE_LIMIT
+#  define KIT_TESTS_SIZE_LIMIT 0x1000
+#endif
+
+#ifndef KIT_TEST_ASSERTIONS_LIMIT
+#  define KIT_TEST_ASSERTIONS_LIMIT 0x50
+#endif
+
+#ifndef KIT_TEST_STRING_SIZE
+#  define KIT_TEST_STRING_SIZE 0x100
+#endif
+
+typedef void (*kit_test_report)(int, char const *file, int line,
+                                bool);
+typedef void (*kit_test_function)(int, kit_test_report);
+
+struct kit_test_case {
+  char              test_name[KIT_TEST_STRING_SIZE];
+  kit_test_function test_fn;
+  int               assertions;
+  char const       *file[KIT_TEST_ASSERTIONS_LIMIT];
+  int               line[KIT_TEST_ASSERTIONS_LIMIT];
+  bool              status[KIT_TEST_ASSERTIONS_LIMIT];
+};
+
+struct kit_tests_list {
+  int                  size;
+  struct kit_test_case tests[KIT_TESTS_SIZE_LIMIT];
+};
+
+extern struct kit_tests_list kit_tests_list;
+
+#ifdef _MSC_VER
+#  pragma section(".CRT$XCU", read)
+#  define KIT_TEST_ON_START_2(f, p)                                \
+    static void f(void);                                           \
+    __declspec(allocate(".CRT$XCU")) void (*f##_)(void) = f;       \
+    __pragma(comment(linker, "/include:" p #f "_")) static void f( \
+        void)
+#  ifdef _WIN64
+#    define KIT_TEST_ON_START(f) KIT_TEST_ON_START_2(f, "")
+#  else
+#    define KIT_TEST_ON_START(f) KIT_TEST_ON_START_2(f, "_")
+#  endif
+#else
+#  define KIT_TEST_ON_START(f)                        \
+    static void f(void) __attribute__((constructor)); \
+    static void f(void)
+#endif
+
+#define KIT_TEST_CONCAT4(a, b, c, d) a##b##c##d
+#define KIT_TEST_CONCAT3(a, b, c) KIT_TEST_CONCAT4(a, b, _, c)
+
+#define TEST(name)                                                   \
+  static void KIT_TEST_CONCAT3(kit_test_run_, __LINE__,              \
+                               KIT_TEST_FILE)(int, kit_test_report); \
+  KIT_TEST_ON_START(                                                 \
+      KIT_TEST_CONCAT3(kit_test_case_, __LINE__, KIT_TEST_FILE)) {   \
+    int n = kit_tests_list.size;                                     \
+    if (n < KIT_TESTS_SIZE_LIMIT) {                                  \
+      kit_tests_list.size++;                                         \
+      kit_tests_list.tests[n].test_fn = KIT_TEST_CONCAT3(            \
+          kit_test_run_, __LINE__, KIT_TEST_FILE);                   \
+      strcpy(kit_tests_list.tests[n].test_name, name);               \
+      kit_tests_list.tests[n].assertions = 0;                        \
+    }                                                                \
+  }                                                                  \
+  static void KIT_TEST_CONCAT3(kit_test_run_, __LINE__,              \
+                               KIT_TEST_FILE)(                       \
+      int kit_test_index_, kit_test_report kit_test_report_)
+
+#define REQUIRE(ok) \
+  kit_test_report_(kit_test_index_, __FILE__, __LINE__, (ok))
+
+int kit_run_tests(int argc, char **argv);
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif