From cc0f05946f718d04238c0fca34dd5e04b78dbbe0 Mon Sep 17 00:00:00 2001
From: Rostyslav Tyshko <tyshko.rostyslav@gmail.com>
Date: Fri, 18 Oct 2024 02:39:11 +0200
Subject: [PATCH] add test_get_proof_multiple

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

diff --git a/storage/src/merkle_tree_public/merkle_tree.rs b/storage/src/merkle_tree_public/merkle_tree.rs
index 0f63f6c..73e9e75 100644
--- a/storage/src/merkle_tree_public/merkle_tree.rs
+++ b/storage/src/merkle_tree_public/merkle_tree.rs
@@ -179,4 +179,16 @@ mod tests {
         assert_eq!(tree.get_tx(tx2.hash()), Some(&tx2));
         assert_eq!(tree.get_tx(tx3.hash()), Some(&tx3));
     }
+
+    #[test]
+    fn test_get_proof_multiple() {
+        let tx1 = MockTransaction { hash: get_first_32_bytes("tx1") };
+        let tx2 = MockTransaction { hash: get_first_32_bytes("tx2") };
+        let tx3 = MockTransaction { hash: get_first_32_bytes("tx3") };
+
+        let tree = HashStorageMerkleTree::new(vec![tx1.clone(), tx2.clone(), tx3.clone()]);
+        let proof = tree.get_proof_multiple(&[tx1.hash(), tx2.hash()]);
+
+        assert!(proof.is_some());
+    }
 }