Page MenuHomePhabricator
Differential D3882 Diff 10810 src/wallet/test/wallet_tests.cpp

Changeset View

Standalone View

View Options

src/wallet/test/wallet_tests.cpp

Show First 20 Lines • Show All 271 Lines • ▼ Show 20 Lines
BOOST_AUTO_TEST_CASE(LoadReceiveRequests) { BOOST_AUTO_TEST_CASE(LoadReceiveRequests) {
CTxDestination dest = CKeyID(); CTxDestination dest = CKeyID();
LOCK(m_wallet.cs_wallet); LOCK(m_wallet.cs_wallet);
m_wallet.AddDestData(dest, "misc", "val_misc"); m_wallet.AddDestData(dest, "misc", "val_misc");
m_wallet.AddDestData(dest, "rr0", "val_rr0"); m_wallet.AddDestData(dest, "rr0", "val_rr0");
m_wallet.AddDestData(dest, "rr1", "val_rr1"); m_wallet.AddDestData(dest, "rr1", "val_rr1");
auto values = m_wallet.GetDestValues("rr"); auto values = m_wallet.GetDestValues("rr");
BOOST_CHECK_EQUAL(values.size(), 2); BOOST_CHECK_EQUAL(values.size(), 2U);
BOOST_CHECK_EQUAL(values[0], "val_rr0"); BOOST_CHECK_EQUAL(values[0], "val_rr0");
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({},
▲ Show 20 Lines • Show All 42 Lines • ▼ Show 20 Lines
}; };
BOOST_FIXTURE_TEST_CASE(ListCoins, ListCoinsTestingSetup) { BOOST_FIXTURE_TEST_CASE(ListCoins, ListCoinsTestingSetup) {
std::string coinbaseAddress = coinbaseKey.GetPubKey().GetID().ToString(); std::string coinbaseAddress = coinbaseKey.GetPubKey().GetID().ToString();
// Confirm ListCoins initially returns 1 coin grouped under coinbaseKey // Confirm ListCoins initially returns 1 coin grouped under coinbaseKey
// address. // address.
auto list = wallet->ListCoins(); auto list = wallet->ListCoins();
BOOST_CHECK_EQUAL(list.size(), 1); 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(), 1); BOOST_CHECK_EQUAL(list.begin()->second.size(), 1U);
// Check initial balance from one mature coinbase transaction. // Check initial balance from one mature coinbase transaction.
BOOST_CHECK_EQUAL(50 * COIN, wallet->GetAvailableBalance()); BOOST_CHECK_EQUAL(50 * COIN, wallet->GetAvailableBalance());
// Add a transaction creating a change address, and confirm ListCoins still // Add a transaction creating a change address, and confirm ListCoins still
// returns the coin associated with the change address underneath the // returns the coin associated with the change address underneath the
// coinbaseKey pubkey, even though the change address has a different // coinbaseKey pubkey, even though the change address has a different
// pubkey. // pubkey.
AddTx(CRecipient{GetScriptForRawPubKey({}), 1 * COIN, AddTx(CRecipient{GetScriptForRawPubKey({}), 1 * COIN,
false /* subtract fee */}); false /* subtract fee */});
list = wallet->ListCoins(); list = wallet->ListCoins();
BOOST_CHECK_EQUAL(list.size(), 1); 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(), 2); BOOST_CHECK_EQUAL(list.begin()->second.size(), 2U);
// Lock both coins. Confirm number of available coins drops to 0. // Lock both coins. Confirm number of available coins drops to 0.
{ {
LOCK2(cs_main, wallet->cs_wallet); LOCK2(cs_main, wallet->cs_wallet);
std::vector<COutput> available; std::vector<COutput> available;
wallet->AvailableCoins(available); wallet->AvailableCoins(available);
BOOST_CHECK_EQUAL(available.size(), 2); BOOST_CHECK_EQUAL(available.size(), 2U);
} }
for (const auto &group : list) { for (const auto &group : list) {
for (const auto &coin : group.second) { for (const auto &coin : group.second) {
LOCK(wallet->cs_wallet); LOCK(wallet->cs_wallet);
wallet->LockCoin(COutPoint(coin.tx->GetId(), coin.i)); wallet->LockCoin(COutPoint(coin.tx->GetId(), coin.i));
} }
} }
{ {
LOCK2(cs_main, wallet->cs_wallet); LOCK2(cs_main, wallet->cs_wallet);
std::vector<COutput> available; std::vector<COutput> available;
wallet->AvailableCoins(available); wallet->AvailableCoins(available);
BOOST_CHECK_EQUAL(available.size(), 0); BOOST_CHECK_EQUAL(available.size(), 0U);
} }
// Confirm ListCoins still returns same result as before, despite coins // Confirm ListCoins still returns same result as before, despite coins
// being locked. // being locked.
list = wallet->ListCoins(); list = wallet->ListCoins();
BOOST_CHECK_EQUAL(list.size(), 1); 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(), 2); BOOST_CHECK_EQUAL(list.begin()->second.size(), 2U);
} }
BOOST_AUTO_TEST_SUITE_END() BOOST_AUTO_TEST_SUITE_END()