Rework test to check for failure to return a spending key

This commit is contained in:
Jack Grigg 2016-08-18 12:22:30 +12:00
parent 0d7386916d
commit 04dfc3c569
1 changed files with 8 additions and 5 deletions

View File

@ -5,19 +5,22 @@
TEST(keystore_tests, store_and_retrieve_spending_key) {
CBasicKeyStore keyStore;
libzcash::SpendingKey skOut;
std::set<libzcash::PaymentAddress> addrs;
keyStore.GetPaymentAddresses(addrs);
ASSERT_EQ(0, addrs.size());
auto sk = libzcash::SpendingKey::random();
keyStore.AddSpendingKey(sk);
auto addr = sk.address();
ASSERT_TRUE(keyStore.HaveSpendingKey(addr));
libzcash::SpendingKey skOut;
keyStore.GetSpendingKey(addr, skOut);
// Sanity-check: we can't get a key we haven't added
ASSERT_FALSE(keyStore.HaveSpendingKey(addr));
ASSERT_FALSE(keyStore.GetSpendingKey(addr, skOut));
keyStore.AddSpendingKey(sk);
ASSERT_TRUE(keyStore.HaveSpendingKey(addr));
ASSERT_TRUE(keyStore.GetSpendingKey(addr, skOut));
ASSERT_EQ(sk, skOut);
keyStore.GetPaymentAddresses(addrs);