zecfoundation
/
zcashd
mirror of https://github.com/ZcashFoundation/zcashd.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add test for GetUnspentNotes() in wallet.

This commit is contained in:
Simon 2016-09-07 11:53:22 -07:00
parent a5ac2e25a0
commit fe2f198277
1 changed files with 78 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

78
src/wallet/gtest/test_wallet.cpp
View File

@ -175,6 +175,84 @@ TEST(wallet_tests, note_data_serialisation) {
EXPECT_EQ(noteData[jsoutpt].witnesses, noteData2[jsoutpt].witnesses);
}
TEST(wallet_tests, find_unspent_notes) {
SelectParams(CBaseChainParams::TESTNET);
CWallet wallet;
auto sk = libzcash::SpendingKey::random();
wallet.AddSpendingKey(sk);
auto wtx = GetValidReceive(sk, 10, true);
auto note = GetNote(sk, wtx, 0, 1);
auto nullifier = note.nullifier(sk);
auto noteMap = wallet.FindMyNotes(wtx);
EXPECT_EQ(2, noteMap.size());
wallet.AddToWallet(wtx, true, NULL);
EXPECT_FALSE(wallet.IsSpent(nullifier));
auto wtx2 = GetValidSpend(sk, note, 5);
wallet.AddToWallet(wtx2, true, NULL);
// two notes, one is spent but wallet doesn't see it as spent yet until mined
std::vector<CNotePlaintextEntry> entries;
wallet.GetUnspentNotes(entries, "", 0);
EXPECT_EQ(2, entries.size());
// Create new payment address, add new note, and filter
auto user2_sk = libzcash::SpendingKey::random();
wallet.AddSpendingKey(user2_sk);
auto user2_pa = user2_sk.address();
auto user2_payment_address = CZCPaymentAddress(user2_pa).ToString();
auto wtx3 = GetValidReceive(user2_sk, 15, true); // adds 2 more notes
auto note2 = GetNote(user2_sk, wtx3, 0, 1);
auto nullifier2 = note2.nullifier(user2_sk);
wallet.AddToWallet(wtx3, true, NULL);
EXPECT_FALSE(wallet.IsSpent(nullifier2));
// 4 notes in wallet, 1 spent (not seen), 1 is the new payment address
entries.clear();
wallet.GetUnspentNotes(entries, "", 0);
EXPECT_EQ(4, entries.size());
entries.clear();
wallet.GetUnspentNotes(entries, user2_payment_address, 0);
EXPECT_EQ(2, entries.size());
// Fake-mine the transaction
EXPECT_EQ(-1, chainActive.Height());
CBlock block;
block.vtx.push_back(wtx2);
block.hashMerkleRoot = block.BuildMerkleTree();
auto blockHash = block.GetHash();
CBlockIndex fakeIndex {block};
mapBlockIndex.insert(std::make_pair(blockHash, &fakeIndex));
chainActive.SetTip(&fakeIndex);
EXPECT_TRUE(chainActive.Contains(&fakeIndex));
EXPECT_EQ(0, chainActive.Height());
wtx2.SetMerkleBranch(block);
wallet.AddToWallet(wtx2, true, NULL);
EXPECT_TRUE(wallet.IsSpent(nullifier));
// 4 notes, 1 spent (now seen)
entries.clear();
wallet.GetUnspentNotes(entries, "", 0);
EXPECT_EQ(3, entries.size());
entries.clear();
// no change to user2 and their two notes.
wallet.GetUnspentNotes(entries, user2_payment_address, 0);
EXPECT_EQ(2, entries.size());
// Tear down
chainActive.SetTip(NULL);
mapBlockIndex.erase(blockHash);
}
TEST(wallet_tests, set_note_addrs_in_cwallettx) {
auto sk = libzcash::SpendingKey::random();
auto wtx = GetValidReceive(sk, 10, true);