From dd755f4ba3a0d13ab86d05cc35a056ce41b4619f Mon Sep 17 00:00:00 2001
From: Rostyslav Tyshko <tyshko.rostyslav@gmail.com>
Date: Fri, 25 Oct 2024 00:39:45 +0200
Subject: [PATCH] add test_get_membership_proof

---
 utxo/src/utxo_tree.rs | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/utxo/src/utxo_tree.rs b/utxo/src/utxo_tree.rs
index 462ed9f..b696411 100644
--- a/utxo/src/utxo_tree.rs
+++ b/utxo/src/utxo_tree.rs
@@ -161,4 +161,19 @@ mod tests {
         assert!(result.is_none());
     }
 
+    #[test]
+    fn test_get_membership_proof() {
+        let mut smt = UTXOSparseMerkleTree::new();
+        let utxo = sample_utxo();
+
+        smt.insert_item(utxo.clone()).unwrap();
+
+        // Fetch membership proof for the inserted UTXO
+        let proof = smt.get_membership_proof(utxo.hash).unwrap();
+
+        // Test proof is generated successfully
+        assert!(proof.is_some());
+    }
+
+
 }