src/wallet/test/wallet_test_fixture.cpp

   1 // Copyright (c) 2016 The Bitcoin Core developers
   2 // Distributed under the MIT software license, see the accompanying
   3 // file COPYING or http://www.opensource.org/licenses/mit-license.php.
   4
   5 #include <wallet/test/wallet_test_fixture.h>
   6
   7 #include <rpc/server.h>
   8 #include <wallet/db.h>
   9
  10 WalletTestingSetup::WalletTestingSetup(const std::string& chainName):
  11     TestingSetup(chainName)
  12 {
  13     bitdb.MakeMock();
  14
  15     bool fFirstRun;
  16     std::unique_ptr<CWalletDBWrapper> dbw(new CWalletDBWrapper(&bitdb, "wallet_test.dat"));
  17     pwalletMain = MakeUnique<CWallet>(std::move(dbw));
  18     pwalletMain->LoadWallet(fFirstRun);
  19     RegisterValidationInterface(pwalletMain.get());
  20
  21     RegisterWalletRPCCommands(tableRPC);
  22 }
  23
  24 WalletTestingSetup::~WalletTestingSetup()
  25 {
  26     UnregisterValidationInterface(pwalletMain.get());
  27
  28     bitdb.Flush(true);
  29     bitdb.Reset();
  30 }