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

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

diff --git a/storage/src/nullifier_sparse_merkle_tree.rs b/storage/src/nullifier_sparse_merkle_tree.rs
index 64ab39a..57c3e55 100644
--- a/storage/src/nullifier_sparse_merkle_tree.rs
+++ b/storage/src/nullifier_sparse_merkle_tree.rs
@@ -179,4 +179,22 @@ mod tests {
         assert_eq!(non_existing.unwrap(), false);
     }
 
+    #[test]
+    fn test_search_multiple_item_inclusions() {
+        let mut tree = NullifierSparseMerkleTree::new();
+        let nullifiers = vec![
+            create_nullifier([1u8; 32]),
+            create_nullifier([2u8; 32]),
+            create_nullifier([3u8; 32]),
+        ];
+
+        tree.insert_items(nullifiers).unwrap();
+
+        let search_hashes = vec![[1u8; 32], [2u8; 32], [99u8; 32]];
+        let result = tree.search_item_inclusions(&search_hashes);
+        assert!(result.is_ok());
+
+        let expected_results = vec![true, true, false];
+        assert_eq!(result.unwrap(), expected_results);
+    }
 }