From 23878e45b5b75defb21b104c731969185246faf0 Mon Sep 17 00:00:00 2001
From: Rostyslav Tyshko <tyshko.rostyslav@gmail.com>
Date: Wed, 23 Oct 2024 02:01:31 +0200
Subject: [PATCH] add test_search_item_inclusion

---
 storage/src/nullifier_sparse_merkle_tree.rs | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/storage/src/nullifier_sparse_merkle_tree.rs b/storage/src/nullifier_sparse_merkle_tree.rs
index d616ab2..64ab39a 100644
--- a/storage/src/nullifier_sparse_merkle_tree.rs
+++ b/storage/src/nullifier_sparse_merkle_tree.rs
@@ -162,4 +162,21 @@ mod tests {
         assert!(result.is_ok());
         assert!(tree.curr_root.is_some());
     }
+
+    #[test]
+    fn test_search_item_inclusion() {
+        let mut tree = NullifierSparseMerkleTree::new();
+        let nullifier = create_nullifier([1u8; 32]);
+
+        tree.insert_item(nullifier.clone()).unwrap();
+
+        let result = tree.search_item_inclusion([1u8; 32]);
+        assert!(result.is_ok());
+        assert_eq!(result.unwrap(), true);
+
+        let non_existing = tree.search_item_inclusion([99u8; 32]);
+        assert!(non_existing.is_ok());
+        assert_eq!(non_existing.unwrap(), false);
+    }
+
 }