Page MenuHomePhabricator
Differential D5564 Diff 17200 src/wallet/test/wallet_tests.cpp

Changeset View

Standalone View

View Options

src/wallet/test/wallet_tests.cpp

Show All 25 Lines
BOOST_FIXTURE_TEST_SUITE(wallet_tests, WalletTestingSetup) BOOST_FIXTURE_TEST_SUITE(wallet_tests, WalletTestingSetup)
static void AddKey(CWallet &wallet, const CKey &key) { static void AddKey(CWallet &wallet, const CKey &key) {
LOCK(wallet.cs_wallet); LOCK(wallet.cs_wallet);
wallet.AddKeyPubKey(key, key.GetPubKey()); wallet.AddKeyPubKey(key, key.GetPubKey());
} }
// Note: when backporting PR14957, see PR15321 and ensure both LockAnnotations BOOST_FIXTURE_TEST_CASE(scan_for_wallet_transactions, TestChain100Setup) {
// are in place.
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.
const CBlockIndex *const null_block = nullptr; const CBlockIndex *const null_block = 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 accommodates a null start block.
{
CWallet wallet(Params(), *chain, WalletLocation(),
WalletDatabase::CreateDummy());
AddKey(wallet, coinbaseKey);
WalletRescanReserver reserver(&wallet);
reserver.reserve();
const CBlockIndex *stop_block = null_block + 1,
*failed_block = null_block + 1;
BOOST_CHECK_EQUAL(
wallet.ScanForWalletTransactions(nullptr, nullptr, reserver,
failed_block, stop_block),
CWallet::ScanResult::SUCCESS);
BOOST_CHECK_EQUAL(failed_block, null_block);
BOOST_CHECK_EQUAL(stop_block, null_block);
BOOST_CHECK_EQUAL(wallet.GetImmatureBalance(), Amount::zero());
}
// 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);
reserver.reserve(); reserver.reserve();
const CBlockIndex *stop_block, *failed_block; const CBlockIndex *stop_block = null_block + 1,
*failed_block = null_block + 1;
BOOST_CHECK_EQUAL( BOOST_CHECK_EQUAL(
wallet.ScanForWalletTransactions(oldTip, nullptr, reserver, wallet.ScanForWalletTransactions(oldTip, nullptr, reserver,
failed_block, stop_block), failed_block, stop_block),
CWallet::ScanResult::SUCCESS); CWallet::ScanResult::SUCCESS);
BOOST_CHECK_EQUAL(failed_block, null_block); BOOST_CHECK_EQUAL(failed_block, null_block);
BOOST_CHECK_EQUAL(stop_block, newTip); BOOST_CHECK_EQUAL(stop_block, newTip);
BOOST_CHECK_EQUAL(wallet.GetImmatureBalance(), 100 * COIN); BOOST_CHECK_EQUAL(wallet.GetImmatureBalance(), 100 * COIN);
} }
// 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 ScanForWalletTransactions only picks transactions in the new block // Verify ScanForWalletTransactions only picks transactions in the new block
// file. // file.
{ {
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);
reserver.reserve(); reserver.reserve();
const CBlockIndex *stop_block, *failed_block; const CBlockIndex *stop_block = null_block + 1,
*failed_block = null_block + 1;
BOOST_CHECK_EQUAL( BOOST_CHECK_EQUAL(
wallet.ScanForWalletTransactions(oldTip, nullptr, reserver, wallet.ScanForWalletTransactions(oldTip, nullptr, reserver,
failed_block, stop_block), failed_block, stop_block),
CWallet::ScanResult::FAILURE); CWallet::ScanResult::FAILURE);
BOOST_CHECK_EQUAL(failed_block, oldTip); BOOST_CHECK_EQUAL(failed_block, oldTip);
BOOST_CHECK_EQUAL(stop_block, newTip); BOOST_CHECK_EQUAL(stop_block, newTip);
BOOST_CHECK_EQUAL(wallet.GetImmatureBalance(), 50 * COIN); BOOST_CHECK_EQUAL(wallet.GetImmatureBalance(), 50 * COIN);
} }
// Prune the remaining block file.
PruneOneBlockFile(newTip->GetBlockPos().nFile);
UnlinkPrunedFiles({newTip->GetBlockPos().nFile});
// Verify ScanForWalletTransactions scans no blocks.
{
CWallet wallet(Params(), *chain, WalletLocation(),
WalletDatabase::CreateDummy());
AddKey(wallet, coinbaseKey);
WalletRescanReserver reserver(&wallet);
reserver.reserve();
const CBlockIndex *stop_block = null_block + 1,
*failed_block = null_block + 1;
BOOST_CHECK_EQUAL(
wallet.ScanForWalletTransactions(oldTip, nullptr, reserver,
failed_block, stop_block),
CWallet::ScanResult::FAILURE);
BOOST_CHECK_EQUAL(failed_block, newTip);
BOOST_CHECK_EQUAL(stop_block, null_block);
BOOST_CHECK_EQUAL(wallet.GetImmatureBalance(), Amount::zero());
}
}
BOOST_FIXTURE_TEST_CASE(importmulti_rescan, TestChain100Setup) {
auto chain = interfaces::MakeChain();
// Cap last block file size, and mine new block in a new block file.
CBlockIndex *oldTip = ::ChainActive().Tip();
GetBlockFileInfo(oldTip->GetBlockPos().nFile)->nSize = MAX_BLOCKFILE_SIZE;
CreateAndProcessBlock({}, GetScriptForRawPubKey(coinbaseKey.GetPubKey()));
CBlockIndex *newTip = ::ChainActive().Tip();
LockAnnotation lock(::cs_main);
auto locked_chain = chain->lock();
// Prune the older block file.
PruneOneBlockFile(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.
{ {
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());
AddWallet(wallet); AddWallet(wallet);
UniValue keys; UniValue keys;
▲ Show 20 Lines • Show All 219 Lines • ▼ Show 20 Lines ListCoinsTestingSetup() {
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();
const CBlockIndex *const null_block = nullptr; const CBlockIndex *const null_block = nullptr;
const CBlockIndex *stop_block, *failed_block; const CBlockIndex *stop_block = null_block + 1,
*failed_block = null_block + 1;
BOOST_CHECK_EQUAL(wallet->ScanForWalletTransactions( BOOST_CHECK_EQUAL(wallet->ScanForWalletTransactions(
ChainActive().Genesis(), nullptr, reserver, ChainActive().Genesis(), nullptr, reserver,
failed_block, stop_block), failed_block, stop_block),
CWallet::ScanResult::SUCCESS); CWallet::ScanResult::SUCCESS);
BOOST_CHECK_EQUAL(stop_block, ChainActive().Tip()); BOOST_CHECK_EQUAL(stop_block, ChainActive().Tip());
BOOST_CHECK_EQUAL(failed_block, null_block); BOOST_CHECK_EQUAL(failed_block, null_block);
} }
▲ Show 20 Lines • Show All 113 Lines • Show Last 20 Lines