about summary refs log tree commit diff homepage
path: root/unittests/KDAlloc/reuse.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'unittests/KDAlloc/reuse.cpp')
-rw-r--r--unittests/KDAlloc/reuse.cpp8
1 files changed, 3 insertions, 5 deletions
diff --git a/unittests/KDAlloc/reuse.cpp b/unittests/KDAlloc/reuse.cpp
index d886a9d6..e6c79ddc 100644
--- a/unittests/KDAlloc/reuse.cpp
+++ b/unittests/KDAlloc/reuse.cpp
@@ -19,11 +19,7 @@
 #include <iostream>
 #include <unordered_set>
 
-#if defined(USE_GTEST_INSTEAD_OF_MAIN)
-int reuse_test() {
-#else
-int main() {
-#endif
+void reuse_test() {
   {
     static const std::size_t size = 8;
     static const std::uint32_t quarantine = 0;
@@ -131,4 +127,6 @@ int main() {
 TEST(KDAllocDeathTest, Reuse) {
   ASSERT_EXIT(reuse_test(), ::testing::ExitedWithCode(0), "");
 }
+#else
+int main() { reuse_test(); }
 #endif