move test.h to test/

3 files changed, 17 insertions, 2 deletions

diff --git a/tests/db.cpp b/tests/db.cpp
index 4ef6ff9..468a4cc 100644
--- a/tests/db.cpp
+++ b/tests/db.cpp
@@ -8,7 +8,7 @@
 #include "kms++.h"
 #include "utils/color.h"
 
-#include "../test.h"
+#include "test.h"
 
 using namespace std;
 using namespace kms;
diff --git a/tests/test.h b/tests/test.h
new file mode 100644
index 0000000..090ff38
--- /dev/null
+++ b/tests/test.h
@@ -0,0 +1,15 @@
+#pragma once
+
+#define unlikely(x) __builtin_expect(!!(x), 0)
+
+static void ASSERT_FAIL(const char *cond, const char *file,
+			   unsigned int line, const char *func) __attribute__ ((__noreturn__));
+
+static void ASSERT_FAIL(const char *cond, const char *file,
+			   unsigned int line, const char *func)
+{
+	fprintf(stderr, "%s:%d: %s: ASSERT(%s) failed\n", file, line, func, cond);
+	abort();
+}
+
+#define ASSERT(x) if (unlikely(!(x))) { ASSERT_FAIL( __STRING(x), __FILE__, __LINE__, __PRETTY_FUNCTION__); }
diff --git a/tests/testpat.cpp b/tests/testpat.cpp
index 5632224..27c1bc3 100644
--- a/tests/testpat.cpp
+++ b/tests/testpat.cpp
@@ -4,7 +4,7 @@
 #include "kms++.h"
 #include "utils/testpat.h"
 
-#include "../test.h"
+#include "test.h"
 
 using namespace std;
 using namespace kms;