From 125fb26ceae09da92d17505071cad881ad31ba88 Mon Sep 17 00:00:00 2001
From: Daniel Schemmel <daniel@schemmel.net>
Date: Wed, 28 Jun 2023 23:21:39 +0000
Subject: Simplify KDAlloc tests

---
 unittests/KDAlloc/allocate.cpp | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

(limited to 'unittests/KDAlloc/allocate.cpp')

diff --git a/unittests/KDAlloc/allocate.cpp b/unittests/KDAlloc/allocate.cpp
index c691e114..eb13d5e1 100644
--- a/unittests/KDAlloc/allocate.cpp
+++ b/unittests/KDAlloc/allocate.cpp
@@ -18,11 +18,7 @@
 #include <iomanip>
 #include <iostream>
 
-#if defined(USE_GTEST_INSTEAD_OF_MAIN)
-int allocate_sample_test() {
-#else
-int main() {
-#endif
+void allocate_sample_test() {
   // initialize a factory and an associated allocator (1 MiB and no quarantine)
   klee::kdalloc::AllocatorFactory factory(static_cast<std::size_t>(1) << 20, 0);
   klee::kdalloc::Allocator allocator = factory.makeAllocator();
@@ -62,4 +58,6 @@ int main() {
 TEST(KDAllocDeathTest, AllocateSample) {
   ASSERT_EXIT(allocate_sample_test(), ::testing::ExitedWithCode(0), "");
 }
+#else
+int main() { allocate_sample_test(); }
 #endif
-- 
cgit 1.4.1