Add support for spending keys to the encrypted wallet.

2016-09-21 14:02:35 -07:00
parent 71b2b66e26
commit 73699ceaf6
7 changed files with 209 additions and 6 deletions
--- a/src/gtest/test_wallet_zkeys.cpp
+++ b/src/gtest/test_wallet_zkeys.cpp
@@ -138,3 +138,81 @@ TEST(wallet_zkeys_tests, write_zkey_direct_to_db) {
    ASSERT_EQ(m.nCreateTime, now);
 }

+
+
+/**
+ * This test covers methods on CWalletDB to load/save crypted z keys.
+ */
+TEST(wallet_zkeys_tests, write_cryptedzkey_direct_to_db) {
+    ECC_Start();
+
+    SelectParams(CBaseChainParams::TESTNET);
+
+    // Get temporary and unique path for file.
+    // Note: / operator to append paths
+    boost::filesystem::path temp = boost::filesystem::temp_directory_path() /
+            boost::filesystem::unique_path();
+    const std::string path = temp.native();
+
+    bool fFirstRun;
+    CWallet wallet(path);
+    ASSERT_EQ(DB_LOAD_OK, wallet.LoadWallet(fFirstRun));
+
+    // No default CPubKey set
+    ASSERT_TRUE(fFirstRun);
+
+    // wallet should be empty
+    std::set<libzcash::PaymentAddress> addrs;
+    wallet.GetPaymentAddresses(addrs);
+    ASSERT_EQ(0, addrs.size());
+
+    // Add random key to the wallet
+    auto paymentAddress = wallet.GenerateNewZKey();
+
+    // wallet should have one key
+    wallet.GetPaymentAddresses(addrs);
+    ASSERT_EQ(1, addrs.size());
+
+    // encrypt wallet
+    SecureString strWalletPass;
+    strWalletPass.reserve(100);
+    strWalletPass = "hello";
+    ASSERT_TRUE(wallet.EncryptWallet(strWalletPass));
+    
+    // adding a new key will fail as the wallet is locked
+    EXPECT_ANY_THROW(wallet.GenerateNewZKey());
+    
+    // unlock wallet and then add
+    wallet.Unlock(strWalletPass);
+    auto paymentAddress2 = wallet.GenerateNewZKey();
+
+    // Create a new wallet from the existing wallet path
+    CWallet wallet2(path);
+    ASSERT_EQ(DB_LOAD_OK, wallet2.LoadWallet(fFirstRun));
+
+    // Confirm it's not the same as the other wallet
+    ASSERT_TRUE(&wallet != &wallet2);
+    
+    // wallet should have two keys
+    wallet2.GetPaymentAddresses(addrs);
+    ASSERT_EQ(2, addrs.size());
+    
+    // check we have entries for our payment addresses
+    ASSERT_TRUE(addrs.count(paymentAddress.Get()));
+    ASSERT_TRUE(addrs.count(paymentAddress2.Get()));
+
+    // spending key is crypted, so we can't extract valid payment address
+    libzcash::SpendingKey keyOut;
+    wallet2.GetSpendingKey(paymentAddress.Get(), keyOut);
+    ASSERT_FALSE(paymentAddress.Get() == keyOut.address());
+    
+    // unlock wallet to get spending keys and verify payment addresses
+    wallet2.Unlock(strWalletPass);
+
+    wallet2.GetSpendingKey(paymentAddress.Get(), keyOut);
+    ASSERT_EQ(paymentAddress.Get(), keyOut.address());
+    
+    wallet2.GetSpendingKey(paymentAddress2.Get(), keyOut);
+    ASSERT_EQ(paymentAddress2.Get(), keyOut.address());
+}
+