From ef8ca179ef007eba5bed497be18f5bc90a4194a6 Mon Sep 17 00:00:00 2001 From: Russell Yanofsky Date: Fri, 28 Apr 2017 13:59:25 -0400 Subject: [PATCH] [test] Add tests for some walletmodel functions Add unit tests for some walletmodel functions that will be refactored & moved in the next commit. --- src/Makefile.qttest.include | 3 +- src/qt/test/wallettests.cpp | 127 +++++++++++++++++++++++++++++++++++- 2 files changed, 127 insertions(+), 3 deletions(-) diff --git a/src/Makefile.qttest.include b/src/Makefile.qttest.include index 948e13a9e1..391b9ebdf6 100644 --- a/src/Makefile.qttest.include +++ b/src/Makefile.qttest.include @@ -46,7 +46,8 @@ qt_test_test_bitcoin_qt_SOURCES = \ if ENABLE_WALLET qt_test_test_bitcoin_qt_SOURCES += \ qt/test/paymentservertests.cpp \ - qt/test/wallettests.cpp + qt/test/wallettests.cpp \ + wallet/test/wallet_test_fixture.cpp endif nodist_qt_test_test_bitcoin_qt_SOURCES = $(TEST_QT_MOC_CPP) diff --git a/src/qt/test/wallettests.cpp b/src/qt/test/wallettests.cpp index a0dce3d997..867711309e 100644 --- a/src/qt/test/wallettests.cpp +++ b/src/qt/test/wallettests.cpp @@ -1,5 +1,6 @@ #include "wallettests.h" +#include "consensus/validation.h" #include "qt/bitcoinamountfield.h" #include "qt/callback.h" #include "qt/optionsmodel.h" @@ -11,6 +12,8 @@ #include "qt/walletmodel.h" #include "test/test_bitcoin.h" #include "validation.h" +#include "wallet/test/wallet_test_fixture.h" +#include "wallet/coincontrol.h" #include "wallet/wallet.h" #include @@ -20,6 +23,118 @@ namespace { + +void TestLoadReceiveRequests() +{ + WalletTestingSetup test; + OptionsModel optionsModel; + WalletModel walletModel(nullptr, pwalletMain, &optionsModel); + + CTxDestination dest = CKeyID(); + pwalletMain->AddDestData(dest, "misc", "val_misc"); + pwalletMain->AddDestData(dest, "rr0", "val_rr0"); + pwalletMain->AddDestData(dest, "rr1", "val_rr1"); + + std::vector values; + walletModel.loadReceiveRequests(values); + QCOMPARE((int)values.size(), 2); + QCOMPARE(QString::fromStdString(values[0]), QString("val_rr0")); + QCOMPARE(QString::fromStdString(values[1]), QString("val_rr1")); +} + +class ListCoinsTestingSetup : public TestChain100Setup +{ +public: + ListCoinsTestingSetup() + { + CreateAndProcessBlock({}, GetScriptForRawPubKey(coinbaseKey.GetPubKey())); + ::bitdb.MakeMock(); + wallet.reset(new CWallet(std::unique_ptr(new CWalletDBWrapper(&bitdb, "wallet_test.dat")))); + bool firstRun; + wallet->LoadWallet(firstRun); + LOCK(wallet->cs_wallet); + wallet->AddKeyPubKey(coinbaseKey, coinbaseKey.GetPubKey()); + wallet->ScanForWalletTransactions(chainActive.Genesis()); + } + + ~ListCoinsTestingSetup() + { + ::bitdb.Flush(true); + ::bitdb.Reset(); + } + + CWalletTx& AddTx(CRecipient recipient) + { + CWalletTx wtx; + CReserveKey reservekey(wallet.get()); + CAmount fee; + int changePos = -1; + std::string error; + wallet->CreateTransaction({recipient}, wtx, reservekey, fee, changePos, error); + CValidationState state; + wallet->CommitTransaction(wtx, reservekey, nullptr, state); + auto it = wallet->mapWallet.find(wtx.GetHash()); + CreateAndProcessBlock({CMutableTransaction(*it->second.tx)}, GetScriptForRawPubKey(coinbaseKey.GetPubKey())); + it->second.SetMerkleBranch(chainActive.Tip(), 1); + return it->second; + } + + std::unique_ptr wallet; +}; + +void TestListCoins() +{ + ListCoinsTestingSetup test; + OptionsModel optionsModel; + WalletModel walletModel(nullptr, test.wallet.get(), &optionsModel); + QString coinbaseAddress = QString::fromStdString(CBitcoinAddress(test.coinbaseKey.GetPubKey().GetID()).ToString()); + + LOCK(test.wallet->cs_wallet); + + // Confirm ListCoins initially returns 1 coin grouped under coinbaseKey + // address. + std::map> list; + walletModel.listCoins(list); + QCOMPARE((int)list.size(), 1); + QCOMPARE(list.begin()->first, coinbaseAddress); + QCOMPARE((int)list.begin()->second.size(), 1); + + // Check initial balance from one mature coinbase transaction. + CCoinControl coinControl; + QCOMPARE(50 * COIN, walletModel.getBalance(&coinControl)); + + // Add a transaction creating a change address, and confirm ListCoins still + // returns the coin associated with the change address underneath the + // coinbaseKey pubkey, even though the change address has a different + // pubkey. + test.AddTx(CRecipient{GetScriptForRawPubKey({}), 1 * COIN, false /* subtract fee */}); + list.clear(); + walletModel.listCoins(list); + QCOMPARE((int)list.size(), 1); + QCOMPARE(list.begin()->first, coinbaseAddress); + QCOMPARE((int)list.begin()->second.size(), 2); + + // Lock both coins. Confirm number of available coins drops to 0. + std::vector available; + test.wallet->AvailableCoins(available); + QCOMPARE((int)available.size(), 2); + for (const auto& group : list) { + for (const auto& coin : group.second) { + test.wallet->LockCoin(COutPoint(coin.tx->GetHash(), coin.i)); + } + } + test.wallet->AvailableCoins(available); + QCOMPARE((int)available.size(), 0); + + // Confirm ListCoins still returns same result as before, despite coins + // being locked. + list.clear(); + walletModel.listCoins(list); + QCOMPARE((int)list.size(), 1); + QCOMPARE(list.begin()->first, coinbaseAddress); + QCOMPARE((int)list.begin()->second.size(), 2); +} + //! Press "Yes" button in modal send confirmation dialog. void ConfirmSend() { @@ -65,7 +180,6 @@ QModelIndex FindTx(const QAbstractItemModel& model, const uint256& txid) } return {}; } -} //! Simple qt wallet tests. // @@ -80,7 +194,7 @@ QModelIndex FindTx(const QAbstractItemModel& model, const uint256& txid) // src/qt/test/test_bitcoin-qt -platform xcb # Linux // src/qt/test/test_bitcoin-qt -platform windows # Windows // src/qt/test/test_bitcoin-qt -platform cocoa # macOS -void WalletTests::walletTests() +void TestSendCoins() { // Set up wallet and chain with 101 blocks (1 mature block for spending). TestChain100Setup test; @@ -117,3 +231,12 @@ void WalletTests::walletTests() bitdb.Flush(true); bitdb.Reset(); } + +} + +void WalletTests::walletTests() +{ + TestLoadReceiveRequests(); + TestListCoins(); + TestSendCoins(); +}