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

---
 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 c8fbd34..08a4981 100644
--- a/storage/src/merkle_tree_public/merkle_tree.rs
+++ b/storage/src/merkle_tree_public/merkle_tree.rs
@@ -131,4 +131,15 @@ mod tests {
         assert_eq!(tree.leaves.len(), 2);
         assert!(tree.get_root().is_some());
     }
+
+    #[test]
+    fn test_get_tx() {
+        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()]);
+
+        assert_eq!(tree.get_tx(tx1.hash()), Some(&tx1));
+        assert_eq!(tree.get_tx(tx2.hash()), Some(&tx2));
+    }
 }