X-Git-Url: https://git.novaco.in/?a=blobdiff_plain;f=src%2Ftest%2Ftest_bitcoin.cpp;h=c7e4bdd55663bb88928643fc8d59fa4f1236f7d9;hb=6ec9d30905e8df77e60d0195074920271abb977a;hp=0230bb6ecadc9df860d27c7d5143a7e1e520405f;hpb=2c3691d53d3f527638d4d9923c8a04488f459de0;p=novacoin.git

diff --git a/src/test/test_bitcoin.cpp b/src/test/test_bitcoin.cpp
index 0230bb6..c7e4bdd 100644
--- a/src/test/test_bitcoin.cpp
+++ b/src/test/test_bitcoin.cpp
@@ -1,18 +1,34 @@
 #define BOOST_TEST_MODULE Bitcoin Test Suite
 #include <boost/test/unit_test.hpp>
 
-#include "../main.h"
-#include "../wallet.h"
+#include "main.h"
+#include "wallet.h"
 
-#include "uint160_tests.cpp"
-#include "uint256_tests.cpp"
-#include "script_tests.cpp"
-#include "transaction_tests.cpp"
+CWallet* pwalletMain;
 
+extern bool fPrintToConsole;
+struct TestingSetup {
+    TestingSetup() {
+        fPrintToConsole = true; // don't want to write to debug.log file
+        pwalletMain = new CWallet();
+        RegisterWallet(pwalletMain);
+    }
+    ~TestingSetup()
+    {
+        delete pwalletMain;
+        pwalletMain = NULL;
+    }
+};
 
-CWallet* pwalletMain;
+BOOST_GLOBAL_FIXTURE(TestingSetup);
 
 void Shutdown(void* parg)
 {
-	exit(0);
+  exit(0);
+}
+
+void StartShutdown()
+{
+  exit(0);
 }
+