Page MenuHomePhabricator
Differential D4739 Diff 14933 src/wallet/test/wallet_tests.cpp

Changeset View

Standalone View

View Options

src/wallet/test/wallet_tests.cpp

Show All 39 LinesBOOST_FIXTURE_TEST_CASE(rescan, TestChain100Setup) {
CreateAndProcessBlock({}, GetScriptForRawPubKey(coinbaseKey.GetPubKey())); CreateAndProcessBlock({}, GetScriptForRawPubKey(coinbaseKey.GetPubKey()));
CBlockIndex *newTip = chainActive.Tip(); CBlockIndex *newTip = chainActive.Tip();
LOCK(cs_main); LOCK(cs_main);
// 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(), "dummy", WalletDatabase::CreateDummy()); CWallet wallet(Params(), WalletLocation(),
WalletDatabase::CreateDummy());
AddKey(wallet, coinbaseKey); AddKey(wallet, coinbaseKey);
WalletRescanReserver reserver(&wallet); WalletRescanReserver reserver(&wallet);
reserver.reserve(); reserver.reserve();
BOOST_CHECK_EQUAL(nullBlock, wallet.ScanForWalletTransactions( BOOST_CHECK_EQUAL(nullBlock, wallet.ScanForWalletTransactions(
oldTip, nullptr, reserver)); oldTip, nullptr, reserver));
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(), "dummy", WalletDatabase::CreateDummy()); CWallet wallet(Params(), WalletLocation(),
WalletDatabase::CreateDummy());
AddKey(wallet, coinbaseKey); AddKey(wallet, coinbaseKey);
WalletRescanReserver reserver(&wallet); WalletRescanReserver reserver(&wallet);
reserver.reserve(); reserver.reserve();
BOOST_CHECK_EQUAL(oldTip, wallet.ScanForWalletTransactions( BOOST_CHECK_EQUAL(oldTip, wallet.ScanForWalletTransactions(
oldTip, nullptr, reserver)); oldTip, nullptr, reserver));
BOOST_CHECK_EQUAL(wallet.GetImmatureBalance(), 50 * COIN); BOOST_CHECK_EQUAL(wallet.GetImmatureBalance(), 50 * COIN);
} }
// 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(), "dummy", WalletDatabase::CreateDummy()); Params(), WalletLocation(), WalletDatabase::CreateDummy());
AddWallet(wallet); AddWallet(wallet);
UniValue keys; UniValue keys;
keys.setArray(); keys.setArray();
UniValue key; UniValue key;
key.setObject(); key.setObject();
key.pushKV("scriptPubKey", key.pushKV("scriptPubKey",
HexStr(GetScriptForRawPubKey(coinbaseKey.GetPubKey()))); HexStr(GetScriptForRawPubKey(coinbaseKey.GetPubKey())));
key.pushKV("timestamp", 0); key.pushKV("timestamp", 0);
▲ Show 20 Lines • Show All 62 Lines • ▼ Show 20 LinesBOOST_FIXTURE_TEST_CASE(importwallet_rescan, TestChain100Setup) {
LOCK(cs_main); LOCK(cs_main);
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(), "dummy", WalletDatabase::CreateDummy()); Params(), WalletLocation(), WalletDatabase::CreateDummy());
LOCK(wallet->cs_wallet); LOCK(wallet->cs_wallet);
wallet->mapKeyMetadata[coinbaseKey.GetPubKey().GetID()].nCreateTime = wallet->mapKeyMetadata[coinbaseKey.GetPubKey().GetID()].nCreateTime =
KEY_TIME; KEY_TIME;
wallet->AddKeyPubKey(coinbaseKey, coinbaseKey.GetPubKey()); wallet->AddKeyPubKey(coinbaseKey, coinbaseKey.GetPubKey());
JSONRPCRequest request; JSONRPCRequest request;
request.params.setArray(); request.params.setArray();
request.params.push_back(backup_file); request.params.push_back(backup_file);
AddWallet(wallet); AddWallet(wallet);
::dumpwallet(GetConfig(), request); ::dumpwallet(GetConfig(), request);
RemoveWallet(wallet); RemoveWallet(wallet);
} }
// Call importwallet RPC and verify all blocks with timestamps >= BLOCK_TIME // Call importwallet RPC and verify all blocks with timestamps >= BLOCK_TIME
// were scanned, and no prior blocks were scanned. // were scanned, and no prior blocks were scanned.
{ {
std::shared_ptr<CWallet> wallet = std::make_shared<CWallet>( std::shared_ptr<CWallet> wallet = std::make_shared<CWallet>(
Params(), "dummy", WalletDatabase::CreateDummy()); Params(), WalletLocation(), WalletDatabase::CreateDummy());
JSONRPCRequest request; JSONRPCRequest request;
request.params.setArray(); request.params.setArray();
request.params.push_back(backup_file); request.params.push_back(backup_file);
AddWallet(wallet); AddWallet(wallet);
::importwallet(GetConfig(), request); ::importwallet(GetConfig(), request);
RemoveWallet(wallet); RemoveWallet(wallet);
Show All 12 Lines
// Check that GetImmatureCredit() returns a newly calculated value instead of // Check that GetImmatureCredit() returns a newly calculated value instead of
// the cached value after a MarkDirty() call. // the cached value after a MarkDirty() call.
// //
// 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) {
CWallet wallet(Params(), "dummy", WalletDatabase::CreateDummy()); CWallet wallet(Params(), WalletLocation(), WalletDatabase::CreateDummy());
CWalletTx wtx(&wallet, m_coinbase_txns.back()); CWalletTx wtx(&wallet, m_coinbase_txns.back());
LOCK2(cs_main, wallet.cs_wallet); LOCK2(cs_main, 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());
▲ Show 20 Lines • Show All 74 Lines • ▼ Show 20 LinesBOOST_AUTO_TEST_CASE(LoadReceiveRequests) {
BOOST_CHECK_EQUAL(values[1], "val_rr1"); BOOST_CHECK_EQUAL(values[1], "val_rr1");
} }
class ListCoinsTestingSetup : public TestChain100Setup { class ListCoinsTestingSetup : public TestChain100Setup {
public: public:
ListCoinsTestingSetup() { ListCoinsTestingSetup() {
CreateAndProcessBlock({}, CreateAndProcessBlock({},
GetScriptForRawPubKey(coinbaseKey.GetPubKey())); GetScriptForRawPubKey(coinbaseKey.GetPubKey()));
wallet = std::make_unique<CWallet>(Params(), "mock", wallet = std::make_unique<CWallet>(Params(), 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);
▲ Show 20 Lines • Show All 92 Lines • ▼ Show 20 LinesBOOST_FIXTURE_TEST_CASE(ListCoins, ListCoinsTestingSetup) {
BOOST_CHECK_EQUAL(list.size(), 1U); BOOST_CHECK_EQUAL(list.size(), 1U);
BOOST_CHECK_EQUAL(boost::get<CKeyID>(list.begin()->first).ToString(), BOOST_CHECK_EQUAL(boost::get<CKeyID>(list.begin()->first).ToString(),
coinbaseAddress); coinbaseAddress);
BOOST_CHECK_EQUAL(list.begin()->second.size(), 2U); BOOST_CHECK_EQUAL(list.begin()->second.size(), 2U);
} }
BOOST_FIXTURE_TEST_CASE(wallet_disableprivkeys, TestChain100Setup) { BOOST_FIXTURE_TEST_CASE(wallet_disableprivkeys, TestChain100Setup) {
std::shared_ptr<CWallet> wallet = std::make_shared<CWallet>( std::shared_ptr<CWallet> wallet = std::make_shared<CWallet>(
Params(), "dummy", WalletDatabase::CreateDummy()); Params(), WalletLocation(), WalletDatabase::CreateDummy());
wallet->SetWalletFlag(WALLET_FLAG_DISABLE_PRIVATE_KEYS); wallet->SetWalletFlag(WALLET_FLAG_DISABLE_PRIVATE_KEYS);
BOOST_CHECK(!wallet->TopUpKeyPool(1000)); BOOST_CHECK(!wallet->TopUpKeyPool(1000));
CPubKey pubkey; CPubKey pubkey;
BOOST_CHECK(!wallet->GetKeyFromPool(pubkey, false)); BOOST_CHECK(!wallet->GetKeyFromPool(pubkey, false));
} }
BOOST_AUTO_TEST_SUITE_END() BOOST_AUTO_TEST_SUITE_END()