From 8ffa7f8b610faa9f4e424f19ca6139d54a22a0c6 Mon Sep 17 00:00:00 2001
From: KunoiSayami <i@leanhe.dev>
Date: Fri, 5 Nov 2021 17:31:32 +0800
Subject: refactor(test): Move tests to standalone folder

Signed-off-by: KunoiSayami <i@leanhe.dev>
---
 CMakeLists.txt        | 25 ++++++++++++++++---------
 headers/arena_test.cu | 52 ---------------------------------------------------
 tests/arena_test.cu   | 52 +++++++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 68 insertions(+), 61 deletions(-)
 delete mode 100644 headers/arena_test.cu
 create mode 100644 tests/arena_test.cu

diff --git a/CMakeLists.txt b/CMakeLists.txt
index c5bbf68..91c7ea4 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -10,20 +10,27 @@ add_library(cleveldb "")
 target_sources(
         cleveldb
         PUBLIC
-        "headers/arena.cu"
         "headers/arena.cuh"
+        "headers/arena.cu"
         "headers/random.h"
 )
 
-add_executable(arena_test "")
-target_sources(arena_test
-        PRIVATE
-        "headers/arena_test.cu")
-target_link_libraries(arena_test cleveldb)
-add_test(arena_test "arena_test" COMMAND "arena_test")
+function(cleveldb_test test_file)
+    get_filename_component(test_target_name "${test_file}" NAME_WE)
 
-add_executable(clevelmain main.cu cleveldb)
+    add_executable("${test_target_name}" "")
+    target_sources(arena_test
+            PRIVATE
+            "${test_file}")
+    target_link_libraries("${test_target_name}" cleveldb)
+    add_test(NAME "${test_target_name}" COMMAND "${test_target_name}")
+endfunction(cleveldb_test)
 
+add_executable(clevelmain main.cu cleveldb)
 
-set_target_properties(cleveldb PROPERTIES
+set_target_properties(clevelmain PROPERTIES
         CUDA_SEPARABLE_COMPILATION ON)
+
+
+cleveldb_test(headers/arena_test.cu)
+
diff --git a/headers/arena_test.cu b/headers/arena_test.cu
deleted file mode 100644
index 68fc8ff..0000000
--- a/headers/arena_test.cu
+++ /dev/null
@@ -1,52 +0,0 @@
-#include "arena.cuh"
-#include "random.h"
-
-using namespace cleveldb;
-
-int main() {
-	std::vector<std::pair<size_t, char*>> allocated;
-	Arena arena;
-	const int N = 100000;
-	size_t bytes = 0;
-	Random rnd(301);
-	for (int i = 0; i < N; i++) {
-		size_t s;
-		if (i % (N / 10) == 0) {
-			s = i;
-		} else {
-			s = rnd.OneIn(4000)
-				? rnd.Uniform(6000)
-				: (rnd.OneIn(10) ? rnd.Uniform(100) : rnd.Uniform(20));
-		}
-		if (s == 0) {
-			// Our arena disallows size 0 allocations.
-			s = 1;
-		}
-		char* r;
-		if (rnd.OneIn(10)) {
-			r = arena.AllocateAligned(s);
-		} else {
-			r = arena.Allocate(s);
-		}
-
-		for (size_t b = 0; b < s; b++) {
-			// Fill the "i"th allocation with a known bit pattern
-			r[b] = i % 256;
-		}
-		bytes += s;
-		allocated.push_back(std::make_pair(s, r));
-		assert(arena.MemoryUsage() >= bytes);
-		if (i > N / 10) {
-			assert(arena.MemoryUsage() <= bytes * 1.10);
-		}
-	}
-	for (size_t i = 0; i < allocated.size(); i++) {
-		size_t num_bytes = allocated[i].first;
-		const char* p = allocated[i].second;
-		for (size_t b = 0; b < num_bytes; b++) {
-			// Check the "i"th allocation for the known bit pattern
-			assert((int(p[b]) & 0xff) == (i % 256));
-		}
-	}
-	return 0;
-}
diff --git a/tests/arena_test.cu b/tests/arena_test.cu
new file mode 100644
index 0000000..68fc8ff
--- /dev/null
+++ b/tests/arena_test.cu
@@ -0,0 +1,52 @@
+#include "arena.cuh"
+#include "random.h"
+
+using namespace cleveldb;
+
+int main() {
+	std::vector<std::pair<size_t, char*>> allocated;
+	Arena arena;
+	const int N = 100000;
+	size_t bytes = 0;
+	Random rnd(301);
+	for (int i = 0; i < N; i++) {
+		size_t s;
+		if (i % (N / 10) == 0) {
+			s = i;
+		} else {
+			s = rnd.OneIn(4000)
+				? rnd.Uniform(6000)
+				: (rnd.OneIn(10) ? rnd.Uniform(100) : rnd.Uniform(20));
+		}
+		if (s == 0) {
+			// Our arena disallows size 0 allocations.
+			s = 1;
+		}
+		char* r;
+		if (rnd.OneIn(10)) {
+			r = arena.AllocateAligned(s);
+		} else {
+			r = arena.Allocate(s);
+		}
+
+		for (size_t b = 0; b < s; b++) {
+			// Fill the "i"th allocation with a known bit pattern
+			r[b] = i % 256;
+		}
+		bytes += s;
+		allocated.push_back(std::make_pair(s, r));
+		assert(arena.MemoryUsage() >= bytes);
+		if (i > N / 10) {
+			assert(arena.MemoryUsage() <= bytes * 1.10);
+		}
+	}
+	for (size_t i = 0; i < allocated.size(); i++) {
+		size_t num_bytes = allocated[i].first;
+		const char* p = allocated[i].second;
+		for (size_t b = 0; b < num_bytes; b++) {
+			// Check the "i"th allocation for the known bit pattern
+			assert((int(p[b]) & 0xff) == (i % 256));
+		}
+	}
+	return 0;
+}
-- 
cgit v1.3.1