From 994e8ff802e98344bf711170ef98465ad1129cdb Mon Sep 17 00:00:00 2001
From: Rostyslav Tyshko <tyshko.rostyslav@gmail.com>
Date: Fri, 18 Oct 2024 02:38:35 +0200
Subject: [PATCH] add test_get_proof

---
 storage/src/merkle_tree_public/merkle_tree.rs | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/storage/src/merkle_tree_public/merkle_tree.rs b/storage/src/merkle_tree_public/merkle_tree.rs
index 08a4981..e30928b 100644
--- a/storage/src/merkle_tree_public/merkle_tree.rs
+++ b/storage/src/merkle_tree_public/merkle_tree.rs
@@ -142,4 +142,15 @@ mod tests {
         assert_eq!(tree.get_tx(tx1.hash()), Some(&tx1));
         assert_eq!(tree.get_tx(tx2.hash()), Some(&tx2));
     }
+
+    #[test]
+    fn test_get_proof() {
+        let tx1 = MockTransaction { hash: get_first_32_bytes("tx1") };
+        let tx2 = MockTransaction { hash: get_first_32_bytes("tx2") };
+
+        let tree = HashStorageMerkleTree::new(vec![tx1.clone(), tx2.clone()]);
+
+        let proof = tree.get_proof(tx1.hash());
+        assert!(proof.is_some());
+    }
 }