Add test for GetNoteDecryptor()

2016-08-31 18:57:32 +12:00
parent 1551db870a
commit be86b6c332
2 changed files with 13 additions and 0 deletions
--- a/src/gtest/test_keystore.cpp
+++ b/src/gtest/test_keystore.cpp
@@ -27,3 +27,15 @@ TEST(keystore_tests, store_and_retrieve_spending_key) {
    EXPECT_EQ(1, addrs.size());
    EXPECT_EQ(1, addrs.count(addr));
 }
+
+TEST(keystore_tests, store_and_retrieve_note_decryptor) {
+    CBasicKeyStore keyStore;
+    ZCNoteDecryption decOut;
+
+    auto sk = libzcash::SpendingKey::random();
+    auto addr = sk.address();
+    keyStore.AddSpendingKey(sk);
+
+    keyStore.GetNoteDecryptor(addr, decOut);
+    EXPECT_EQ(ZCNoteDecryption(sk.viewing_key()), decOut);
+}