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/sample.cpp | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

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

diff --git a/unittests/KDAlloc/sample.cpp b/unittests/KDAlloc/sample.cpp
index 365e7e21..ddf8b75d 100644
--- a/unittests/KDAlloc/sample.cpp
+++ b/unittests/KDAlloc/sample.cpp
@@ -15,11 +15,7 @@
 
 #include <cassert>
 
-#if defined(USE_GTEST_INSTEAD_OF_MAIN)
-int sample_test() {
-#else
-int main() {
-#endif
+void sample_test() {
   // initialize a factory and an associated allocator (1 TiB and no quarantine)
   klee::kdalloc::AllocatorFactory factory(static_cast<std::size_t>(1) << 40, 0);
   klee::kdalloc::Allocator allocator = factory.makeAllocator();
@@ -64,4 +60,6 @@ int main() {
 TEST(KDAllocDeathTest, Sample) {
   ASSERT_EXIT(sample_test(), ::testing::ExitedWithCode(0), "");
 }
+#else
+int main() { sample_test(); }
 #endif
-- 
cgit 1.4.1