You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
1.8 KiB
80 lines
1.8 KiB
#define BOOST_TEST_MODULE Bitcoin Test Suite |
|
|
|
|
|
|
|
#include "db.h" |
|
#include "main.h" |
|
#include "txdb.h" |
|
#include "ui_interface.h" |
|
#include "util.h" |
|
#ifdef ENABLE_WALLET |
|
#include "wallet.h" |
|
#endif |
|
|
|
#include <boost/filesystem.hpp> |
|
#include <boost/test/unit_test.hpp> |
|
|
|
|
|
CWallet* pwalletMain; |
|
|
|
extern bool fPrintToConsole; |
|
extern void noui_connect(); |
|
|
|
struct TestingSetup { |
|
CCoinsViewDB *pcoinsdbview; |
|
boost::filesystem::path pathTemp; |
|
boost::thread_group threadGroup; |
|
|
|
TestingSetup() { |
|
fPrintToDebugger = true; // don't want to write to debug.log file |
|
noui_connect(); |
|
#ifdef ENABLE_WALLET |
|
bitdb.MakeMock(); |
|
#endif |
|
pathTemp = GetTempPath() / strprintf("test_bitcoin_%lu_%i", (unsigned long)GetTime(), (int)(GetRand(100000))); |
|
boost::filesystem::create_directories(pathTemp); |
|
mapArgs["-datadir"] = pathTemp.string(); |
|
pblocktree = new CBlockTreeDB(1 << 20, true); |
|
pcoinsdbview = new CCoinsViewDB(1 << 23, true); |
|
pcoinsTip = new CCoinsViewCache(*pcoinsdbview); |
|
InitBlockIndex(); |
|
#ifdef ENABLE_WALLET |
|
bool fFirstRun; |
|
pwalletMain = new CWallet("wallet.dat"); |
|
pwalletMain->LoadWallet(fFirstRun); |
|
RegisterWallet(pwalletMain); |
|
#endif |
|
nScriptCheckThreads = 3; |
|
for (int i=0; i < nScriptCheckThreads-1; i++) |
|
threadGroup.create_thread(&ThreadScriptCheck); |
|
} |
|
~TestingSetup() |
|
{ |
|
threadGroup.interrupt_all(); |
|
threadGroup.join_all(); |
|
#ifdef ENABLE_WALLET |
|
delete pwalletMain; |
|
pwalletMain = NULL; |
|
#endif |
|
delete pcoinsTip; |
|
delete pcoinsdbview; |
|
delete pblocktree; |
|
#ifdef ENABLE_WALLET |
|
bitdb.Flush(true); |
|
#endif |
|
boost::filesystem::remove_all(pathTemp); |
|
} |
|
}; |
|
|
|
BOOST_GLOBAL_FIXTURE(TestingSetup); |
|
|
|
void Shutdown(void* parg) |
|
{ |
|
exit(0); |
|
} |
|
|
|
void StartShutdown() |
|
{ |
|
exit(0); |
|
} |
|
|
|
|