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

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

diff --git a/storage/src/merkle_tree_public/merkle_tree.rs b/storage/src/merkle_tree_public/merkle_tree.rs
index 6b7d35f..0f63f6c 100644
--- a/storage/src/merkle_tree_public/merkle_tree.rs
+++ b/storage/src/merkle_tree_public/merkle_tree.rs
@@ -165,4 +165,18 @@ mod tests {
         assert_eq!(tree.leaves.len(), 2);
         assert_eq!(tree.get_tx(tx2.hash()), Some(&tx2));
     }
+
+    #[test]
+    fn test_add_tx_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 mut tree = HashStorageMerkleTree::new(vec![tx1.clone()]);
+        tree.add_tx_multiple(vec![tx2.clone(), tx3.clone()]);
+
+        assert_eq!(tree.leaves.len(), 3);
+        assert_eq!(tree.get_tx(tx2.hash()), Some(&tx2));
+        assert_eq!(tree.get_tx(tx3.hash()), Some(&tx3));
+    }
 }