Page MenuHomePhabricator
Differential D5373 Diff 16702 src/wallet/test/wallet_tests.cpp

Changeset View

Standalone View

View Options

src/wallet/test/wallet_tests.cpp

Show All 34 Lines
// Note: when backporting PR14957, see PR15321 and ensure both LockAnnotations // Note: when backporting PR14957, see PR15321 and ensure both LockAnnotations
// are in place. // are in place.
BOOST_FIXTURE_TEST_CASE(rescan, TestChain100Setup) { BOOST_FIXTURE_TEST_CASE(rescan, TestChain100Setup) {
auto chain = interfaces::MakeChain(); auto chain = interfaces::MakeChain();
// 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();
LockAnnotation lock(::cs_main); LockAnnotation lock(::cs_main);
auto locked_chain = chain->lock(); 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(),
▲ Show 20 Lines • Show All 76 Lines • ▼ Show 20 Lines
// greater or equal than key birthday. Previously there was a bug where // greater or equal than key birthday. Previously there was a bug where
// importwallet RPC would start the scan at the latest block with timestamp less // importwallet RPC would start the scan at the latest block with timestamp less
// than or equal to key birthday. // than or equal to key birthday.
BOOST_FIXTURE_TEST_CASE(importwallet_rescan, TestChain100Setup) { BOOST_FIXTURE_TEST_CASE(importwallet_rescan, TestChain100Setup) {
auto chain = interfaces::MakeChain(); auto chain = interfaces::MakeChain();
// Create two blocks with same timestamp to verify that importwallet rescan // Create two blocks with same timestamp to verify that importwallet rescan
// will pick up both blocks, not just the first. // will pick up both blocks, not just the first.
const int64_t BLOCK_TIME = chainActive.Tip()->GetBlockTimeMax() + 5; const int64_t BLOCK_TIME = ::ChainActive().Tip()->GetBlockTimeMax() + 5;
SetMockTime(BLOCK_TIME); SetMockTime(BLOCK_TIME);
m_coinbase_txns.emplace_back( m_coinbase_txns.emplace_back(
CreateAndProcessBlock({}, CreateAndProcessBlock({},
GetScriptForRawPubKey(coinbaseKey.GetPubKey())) GetScriptForRawPubKey(coinbaseKey.GetPubKey()))
.vtx[0]); .vtx[0]);
m_coinbase_txns.emplace_back( m_coinbase_txns.emplace_back(
CreateAndProcessBlock({}, CreateAndProcessBlock({},
GetScriptForRawPubKey(coinbaseKey.GetPubKey())) GetScriptForRawPubKey(coinbaseKey.GetPubKey()))
▲ Show 20 Lines • Show All 64 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, WalletLocation(), CWallet wallet(Params(), *chain, 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();
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.
▲ Show 20 Lines • Show All 79 Lines • ▼ Show 20 Lines ListCoinsTestingSetup() {
GetScriptForRawPubKey(coinbaseKey.GetPubKey())); GetScriptForRawPubKey(coinbaseKey.GetPubKey()));
wallet = std::make_unique<CWallet>(Params(), *m_chain, WalletLocation(), wallet = std::make_unique<CWallet>(Params(), *m_chain, WalletLocation(),
WalletDatabase::CreateMock()); WalletDatabase::CreateMock());
bool firstRun; bool firstRun;
wallet->LoadWallet(firstRun); wallet->LoadWallet(firstRun);
AddKey(*wallet, coinbaseKey); AddKey(*wallet, coinbaseKey);
WalletRescanReserver reserver(wallet.get()); WalletRescanReserver reserver(wallet.get());
reserver.reserve(); reserver.reserve();
wallet->ScanForWalletTransactions(chainActive.Genesis(), nullptr, wallet->ScanForWalletTransactions(::ChainActive().Genesis(), nullptr,
reserver); reserver);
} }
~ListCoinsTestingSetup() { wallet.reset(); } ~ListCoinsTestingSetup() { wallet.reset(); }
CWalletTx &AddTx(CRecipient recipient) { CWalletTx &AddTx(CRecipient recipient) {
CTransactionRef tx; CTransactionRef tx;
CReserveKey reservekey(wallet.get()); CReserveKey reservekey(wallet.get());
Show All 13 Lines CWalletTx &AddTx(CRecipient recipient) {
blocktx = blocktx =
CMutableTransaction(*wallet->mapWallet.at(tx->GetId()).tx); CMutableTransaction(*wallet->mapWallet.at(tx->GetId()).tx);
} }
CreateAndProcessBlock({CMutableTransaction(blocktx)}, CreateAndProcessBlock({CMutableTransaction(blocktx)},
GetScriptForRawPubKey(coinbaseKey.GetPubKey())); GetScriptForRawPubKey(coinbaseKey.GetPubKey()));
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 // Temporary. Removed in upcoming lock cleanup
std::unique_ptr<interfaces::Chain::Lock> m_locked_chain = std::unique_ptr<interfaces::Chain::Lock> m_locked_chain =
m_chain->assumeLocked(); m_chain->assumeLocked();
std::unique_ptr<CWallet> wallet; std::unique_ptr<CWallet> wallet;
▲ Show 20 Lines • Show All 78 Lines • Show Last 20 Lines