Page MenuHomePhabricator
Differential D6262 Diff 20627 src/wallet/test/wallet_tests.cpp

Changeset View

Standalone View

View Options

src/wallet/test/wallet_tests.cpp

Show All 35 LinesBOOST_FIXTURE_TEST_CASE(scan_for_wallet_transactions, 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 *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();
auto chain = interfaces::MakeChain(); auto chain = interfaces::MakeChain();
auto locked_chain = chain->lock(); auto locked_chain = chain->lock();
LockAnnotation lock(::cs_main); LockAssertion lock(::cs_main);
// Verify ScanForWalletTransactions accommodates a null start block. // Verify ScanForWalletTransactions accommodates a null start block.
{ {
CWallet wallet(Params(), chain.get(), WalletLocation(), CWallet wallet(Params(), chain.get(), WalletLocation(),
WalletDatabase::CreateDummy()); WalletDatabase::CreateDummy());
AddKey(wallet, coinbaseKey); AddKey(wallet, coinbaseKey);
WalletRescanReserver reserver(&wallet); WalletRescanReserver reserver(&wallet);
reserver.reserve(); reserver.reserve();
▲ Show 20 Lines • Show All 69 Lines • ▼ Show 20 LinesBOOST_FIXTURE_TEST_CASE(importmulti_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 *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();
auto chain = interfaces::MakeChain(); auto chain = interfaces::MakeChain();
auto locked_chain = chain->lock(); auto locked_chain = chain->lock();
LockAnnotation lock(::cs_main); LockAssertion lock(::cs_main);
// Prune the older block file. // Prune the older block file.
PruneOneBlockFile(oldTip->GetBlockPos().nFile); PruneOneBlockFile(oldTip->GetBlockPos().nFile);
UnlinkPrunedFiles({oldTip->GetBlockPos().nFile}); UnlinkPrunedFiles({oldTip->GetBlockPos().nFile});
// Verify importmulti RPC returns failure for a key whose creation time is // Verify importmulti RPC returns failure for a key whose creation time is
// before the missing block, and success for a key whose creation time is // before the missing block, and success for a key whose creation time is
// after. // after.
▲ Show 20 Lines • Show All 68 Lines • ▼ Show 20 LinesBOOST_FIXTURE_TEST_CASE(importwallet_rescan, TestChain100Setup) {
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]);
auto chain = interfaces::MakeChain(); auto chain = interfaces::MakeChain();
auto locked_chain = chain->lock(); auto locked_chain = chain->lock();
LockAnnotation lock(::cs_main); LockAssertion lock(::cs_main);
std::string backup_file = (GetDataDir() / "wallet.backup").string(); std::string backup_file = (GetDataDir() / "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::shared_ptr<CWallet> wallet =
std::make_shared<CWallet>(Params(), chain.get(), WalletLocation(), std::make_shared<CWallet>(Params(), chain.get(), WalletLocation(),
WalletDatabase::CreateDummy()); WalletDatabase::CreateDummy());
▲ Show 20 Lines • Show All 43 Lines • ▼ Show 20 Lines
// 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.get(), WalletLocation(), CWallet wallet(Params(), chain.get(), WalletLocation(),
WalletDatabase::CreateDummy()); WalletDatabase::CreateDummy());
CWalletTx wtx(&wallet, m_coinbase_txns.back()); CWalletTx wtx(&wallet, m_coinbase_txns.back());
auto locked_chain = chain->lock(); auto locked_chain = chain->lock();
LockAnnotation lock(::cs_main); LockAssertion lock(::cs_main);
LOCK(wallet.cs_wallet); 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(*locked_chain), Amount::zero()); BOOST_CHECK_EQUAL(wtx.GetImmatureCredit(*locked_chain), 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(*locked_chain), 50 * COIN); BOOST_CHECK_EQUAL(wtx.GetImmatureCredit(*locked_chain), 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) {
auto locked_chain = wallet.chain().lock(); auto locked_chain = wallet.chain().lock();
LockAnnotation lock(::cs_main); LockAssertion lock(::cs_main);
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 196 Lines • Show Last 20 Lines