Merge pull request #7903 from Habbie/dnsdist-doc-nits

[thirdparty/pdns.git] / pdns / testrunner.cc

diff --git a/pdns/testrunner.cc b/pdns/testrunner.cc
index f27a6c3d1d82310f660060e1d6d8bef959a34a5f..0ea154593f0acec2b56d1baba40fbffd2121d981 100644 (file)
--- a/pdns/testrunner.cc
+++ b/pdns/testrunner.cc
@@ -1,8 +1,31 @@
 #define BOOST_TEST_DYN_LINK
-#define BOOST_TEST_MAIN
-#define BOOST_TEST_MODULE unit
-
-#include "packetcache.hh"
-PacketCache PC;
 
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
 #include <boost/test/unit_test.hpp>
+#include "arguments.hh"
+#include "auth-packetcache.hh"
+#include "auth-querycache.hh"
+#include "statbag.hh"
+StatBag S;
+AuthPacketCache PC;
+AuthQueryCache QC;
+
+ArgvMap &arg()
+{
+  static ArgvMap theArg;
+  return theArg;
+}
+
+
+bool init_unit_test() {
+  reportAllTypes();
+  return true;
+}
+
+// entry point:
+int main(int argc, char* argv[])
+{
+  return boost::unit_test::unit_test_main( &init_unit_test, argc, argv );
+}