Move all tests next to the code they're testingHEADmaster

1 files changed, 22 insertions, 0 deletions

diff --git a/lib/libk/stdlib/test_allocator.hh b/lib/libk/stdlib/test_allocator.hh
new file mode 100644
index 0000000..3bc1715
--- /dev/null
+++ b/lib/libk/stdlib/test_allocator.hh
@@ -0,0 +1,22 @@
+#pragma once
+
+class TestAllocator : public ::testing::Test {
+protected:
+  void
+  SetUp() override
+  {
+    memory = malloc(memory_size);
+    libk::alloc_init(memory, memory_size);
+    ASSERT_EQ(libk::begin, memory);
+  }
+
+  void
+  TearDown() override
+  {
+    free(memory);
+    libk::begin = nullptr;
+  }
+
+  const size_t memory_size = 4096;
+  void *memory = nullptr;
+};