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

---
 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 e30928b..6b7d35f 100644
--- a/storage/src/merkle_tree_public/merkle_tree.rs
+++ b/storage/src/merkle_tree_public/merkle_tree.rs
@@ -153,4 +153,16 @@ mod tests {
         let proof = tree.get_proof(tx1.hash());
         assert!(proof.is_some());
     }
+
+    #[test]
+    fn test_add_tx() {
+        let tx1 = MockTransaction { hash: get_first_32_bytes("tx1") };
+        let tx2 = MockTransaction { hash: get_first_32_bytes("tx2") };
+
+        let mut tree = HashStorageMerkleTree::new(vec![tx1.clone()]);
+
+        tree.add_tx(tx2.clone());
+        assert_eq!(tree.leaves.len(), 2);
+        assert_eq!(tree.get_tx(tx2.hash()), Some(&tx2));
+    }
 }