From 40de3bee0278d6584561a827123bc342934adccb Mon Sep 17 00:00:00 2001
From: Andrei Karas <akaras@inbox.ru>
Date: Thu, 18 May 2017 22:51:52 +0300
Subject: Add support for additional unit test library doctest.

---
 src/maingui.cpp | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'src/maingui.cpp')

diff --git a/src/maingui.cpp b/src/maingui.cpp
index 37536470a..374158183 100644
--- a/src/maingui.cpp
+++ b/src/maingui.cpp
@@ -49,8 +49,14 @@
 #endif  // SDL_VERSIONNUM
 
 #ifdef UNITTESTS
+#ifdef UNITTESTS_CATCH
 #define CATCH_CONFIG_RUNNER
 #include "test/catch.hpp"
+#endif  // UNITTESTS_CATCH
+#ifdef UNITTESTS_DOCTEST
+#define DOCTEST_CONFIG_IMPLEMENT
+#include "test/doctest.h"
+#endif  // UNITTESTS_DOCTEST
 #else  // UNITTESTS
 #include "utils/xml.h"
 #endif  // UNITTESTS
@@ -140,7 +146,15 @@ int mainGui(int argc, char *argv[])
 int main(int argc, char *argv[])
 {
     VirtFs::init(argv[0]);
+#ifdef UNITTESTS_CATCH
     return Catch::Session().run(argc, argv);
+#elif defined(UNITTESTS_DOCTEST)
+    doctest::Context context;
+    context.applyCommandLine(argc, argv);
+    return context.run();
+#else // UNITTESTS_CATCH
+    return 1;
+#endif  // UNITTESTS_CATCH
 }
 
 #endif  // UNITTESTS
-- 
cgit v1.2.3-60-g2f50