From d3ad79e98125fa1c9d15b6b85e4da9efb76f7682 Mon Sep 17 00:00:00 2001
From: Aditya Kulkarni <zcash@adityapk.com>
Date: Wed, 2 Oct 2019 17:45:10 -0700
Subject: [PATCH] Note sapling test

---
 src/lightwallet/mod.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/lightwallet/mod.rs b/src/lightwallet/mod.rs
index a2a9eb4..7df7a3b 100644
--- a/src/lightwallet/mod.rs
+++ b/src/lightwallet/mod.rs
@@ -2055,7 +2055,6 @@ pub mod tests {
 
             // Note
             assert_eq!(txs[&txid1].notes[0].note.value, AMOUNT_Z);
-            assert_eq!(wallet.note_address(&txs[&txid1].notes[0]), Some(ext_address));
             assert_eq!(txs[&txid1].notes[0].spent, None);
             assert_eq!(txs[&txid1].notes[0].unconfirmed_spent, Some(sent_txid));
         }
@@ -2124,6 +2123,7 @@ pub mod tests {
 
             assert_eq!(txs[&sent_txid].notes[0].extfvk, wallet.extfvks[0]);
             assert_eq!(txs[&sent_txid].notes[0].note.value, AMOUNT1 - fee);
+            assert_eq!(wallet.note_address(&txs[&sent_txid].notes[0]), Some(my_address));
             assert_eq!(LightWallet::memo_str(&txs[&sent_txid].notes[0].memo), Some(memo));
         }
     }