Page MenuHomePhabricator
Differential D5075 Diff 15806 src/wallet/test/wallet_tests.cpp

Changeset View

Standalone View

View Options

src/wallet/test/wallet_tests.cpp

// Copyright (c) 2012-2016 The Bitcoin Core developers // Copyright (c) 2012-2016 The Bitcoin Core developers
// Distributed under the MIT software license, see the accompanying // Distributed under the MIT software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php. // file COPYING or http://www.opensource.org/licenses/mit-license.php.
#include <chain.h> #include <chain.h>
#include <chainparams.h> #include <chainparams.h>
#include <config.h> #include <config.h>
#include <consensus/validation.h> #include <consensus/validation.h>
#include <interfaces/chain.h>
#include <rpc/server.h> #include <rpc/server.h>
#include <validation.h> #include <validation.h>
#include <wallet/coincontrol.h> #include <wallet/coincontrol.h>
#include <wallet/rpcdump.h> #include <wallet/rpcdump.h>
#include <wallet/wallet.h> #include <wallet/wallet.h>
#include <test/test_bitcoin.h> #include <test/test_bitcoin.h>
#include <wallet/test/wallet_test_fixture.h> #include <wallet/test/wallet_test_fixture.h>
Show All 20 LinesBOOST_FIXTURE_TEST_CASE(rescan, TestChain100Setup) {
// Cap last block file size, and mine new block in a new block file. // Cap last block file size, and mine new block in a new block file.
CBlockIndex *const nullBlock = nullptr; CBlockIndex *const nullBlock = nullptr;
CBlockIndex *oldTip = chainActive.Tip(); CBlockIndex *oldTip = chainActive.Tip();
GetBlockFileInfo(oldTip->GetBlockPos().nFile)->nSize = MAX_BLOCKFILE_SIZE; GetBlockFileInfo(oldTip->GetBlockPos().nFile)->nSize = MAX_BLOCKFILE_SIZE;
CreateAndProcessBlock({}, GetScriptForRawPubKey(coinbaseKey.GetPubKey())); CreateAndProcessBlock({}, GetScriptForRawPubKey(coinbaseKey.GetPubKey()));
CBlockIndex *newTip = chainActive.Tip(); CBlockIndex *newTip = chainActive.Tip();
LOCK(cs_main); auto locked_chain = chain->lock();
// Verify ScanForWalletTransactions picks up transactions in both the old // Verify ScanForWalletTransactions picks up transactions in both the old
// and new block files. // and new block files.
{ {
CWallet wallet(Params(), *chain, WalletLocation(), CWallet wallet(Params(), *chain, WalletLocation(),
WalletDatabase::CreateDummy()); WalletDatabase::CreateDummy());
AddKey(wallet, coinbaseKey); AddKey(wallet, coinbaseKey);
WalletRescanReserver reserver(&wallet); WalletRescanReserver reserver(&wallet);
▲ Show 20 Lines • Show All 93 Lines • ▼ Show 20 LinesBOOST_FIXTURE_TEST_CASE(importwallet_rescan, TestChain100Setup) {
// rescan will start at the block time. // rescan will start at the block time.
const int64_t KEY_TIME = BLOCK_TIME + TIMESTAMP_WINDOW; const int64_t KEY_TIME = BLOCK_TIME + TIMESTAMP_WINDOW;
SetMockTime(KEY_TIME); SetMockTime(KEY_TIME);
m_coinbase_txns.emplace_back( m_coinbase_txns.emplace_back(
CreateAndProcessBlock({}, CreateAndProcessBlock({},
GetScriptForRawPubKey(coinbaseKey.GetPubKey())) GetScriptForRawPubKey(coinbaseKey.GetPubKey()))
.vtx[0]); .vtx[0]);
LOCK(cs_main); auto locked_chain = chain->lock();
std::string backup_file = std::string backup_file =
(SetDataDir("importwallet_rescan") / "wallet.backup").string(); (SetDataDir("importwallet_rescan") / "wallet.backup").string();
// Import key into wallet and call dumpwallet to create backup file. // Import key into wallet and call dumpwallet to create backup file.
{ {
std::shared_ptr<CWallet> wallet = std::make_shared<CWallet>( std::shared_ptr<CWallet> wallet = std::make_shared<CWallet>(
Params(), *chain, WalletLocation(), WalletDatabase::CreateDummy()); Params(), *chain, WalletLocation(), WalletDatabase::CreateDummy());
▲ Show 20 Lines • Show All 42 Lines • ▼ Show 20 Lines
// This is a regression test written to verify a bugfix for the immature credit // This is a regression test written to verify a bugfix for the immature credit
// function. Similar tests probably should be written for the other credit and // function. Similar tests probably should be written for the other credit and
// debit functions. // debit functions.
BOOST_FIXTURE_TEST_CASE(coin_mark_dirty_immature_credit, TestChain100Setup) { BOOST_FIXTURE_TEST_CASE(coin_mark_dirty_immature_credit, TestChain100Setup) {
auto chain = interfaces::MakeChain(); auto chain = interfaces::MakeChain();
CWallet wallet(Params(), *chain, WalletLocation(), CWallet wallet(Params(), *chain, WalletLocation(),
WalletDatabase::CreateDummy()); WalletDatabase::CreateDummy());
CWalletTx wtx(&wallet, m_coinbase_txns.back()); CWalletTx wtx(&wallet, m_coinbase_txns.back());
LOCK2(cs_main, wallet.cs_wallet); auto locked_chain = chain->lock();
LOCK(wallet.cs_wallet);
wtx.hashBlock = chainActive.Tip()->GetBlockHash(); wtx.hashBlock = chainActive.Tip()->GetBlockHash();
wtx.nIndex = 0; wtx.nIndex = 0;
// Call GetImmatureCredit() once before adding the key to the wallet to // Call GetImmatureCredit() once before adding the key to the wallet to
// cache the current immature credit amount, which is 0. // cache the current immature credit amount, which is 0.
BOOST_CHECK_EQUAL(wtx.GetImmatureCredit(), Amount::zero()); BOOST_CHECK_EQUAL(wtx.GetImmatureCredit(), Amount::zero());
// Invalidate the cached value, add the key, and make sure a new immature // Invalidate the cached value, add the key, and make sure a new immature
// credit amount is calculated. // credit amount is calculated.
wtx.MarkDirty(); wtx.MarkDirty();
wallet.AddKeyPubKey(coinbaseKey, coinbaseKey.GetPubKey()); wallet.AddKeyPubKey(coinbaseKey, coinbaseKey.GetPubKey());
BOOST_CHECK_EQUAL(wtx.GetImmatureCredit(), 50 * COIN); BOOST_CHECK_EQUAL(wtx.GetImmatureCredit(), 50 * COIN);
} }
static int64_t AddTx(CWallet &wallet, uint32_t lockTime, int64_t mockTime, static int64_t AddTx(CWallet &wallet, uint32_t lockTime, int64_t mockTime,
int64_t blockTime) { int64_t blockTime) {
CMutableTransaction tx; CMutableTransaction tx;
tx.nLockTime = lockTime; tx.nLockTime = lockTime;
SetMockTime(mockTime); SetMockTime(mockTime);
CBlockIndex *block = nullptr; CBlockIndex *block = nullptr;
if (blockTime > 0) { if (blockTime > 0) {
LOCK(cs_main); auto locked_chain = wallet.chain().lock();
auto inserted = auto inserted =
mapBlockIndex.emplace(BlockHash(GetRandHash()), new CBlockIndex); mapBlockIndex.emplace(BlockHash(GetRandHash()), new CBlockIndex);
assert(inserted.second); assert(inserted.second);
const BlockHash &hash = inserted.first->first; const BlockHash &hash = inserted.first->first;
block = inserted.first->second; block = inserted.first->second;
block->nTime = blockTime; block->nTime = blockTime;
block->phashBlock = &hash; block->phashBlock = &hash;
} }
▲ Show 20 Lines • Show All 91 Lines • ▼ Show 20 Lines CWalletTx &AddTx(CRecipient recipient) {
LOCK(wallet->cs_wallet); LOCK(wallet->cs_wallet);
auto it = wallet->mapWallet.find(tx->GetId()); auto it = wallet->mapWallet.find(tx->GetId());
BOOST_CHECK(it != wallet->mapWallet.end()); BOOST_CHECK(it != wallet->mapWallet.end());
it->second.SetMerkleBranch(chainActive.Tip(), 1); it->second.SetMerkleBranch(chainActive.Tip(), 1);
return it->second; return it->second;
} }
std::unique_ptr<interfaces::Chain> m_chain = interfaces::MakeChain(); std::unique_ptr<interfaces::Chain> m_chain = interfaces::MakeChain();
// Temporary. Removed in upcoming lock cleanup
std::unique_ptr<interfaces::Chain::Lock> m_locked_chain =
m_chain->assumeLocked();
std::unique_ptr<CWallet> wallet; std::unique_ptr<CWallet> wallet;
}; };
BOOST_FIXTURE_TEST_CASE(ListCoins, ListCoinsTestingSetup) { BOOST_FIXTURE_TEST_CASE(ListCoins, ListCoinsTestingSetup) {
std::string coinbaseAddress = coinbaseKey.GetPubKey().GetID().ToString(); std::string coinbaseAddress = coinbaseKey.GetPubKey().GetID().ToString();
// Confirm ListCoins initially returns 1 coin grouped under coinbaseKey // Confirm ListCoins initially returns 1 coin grouped under coinbaseKey
// address. // address.
▲ Show 20 Lines • Show All 71 Lines • Show Last 20 Lines