added tree history size to inclusion proof test

worldcoin · Oct 27, 2023 · 788e563 · 788e563
1 parent 7f944a1
commit 788e563
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/crates/tree_availability/src/world_tree/tree_data.rs b/crates/tree_availability/src/world_tree/tree_data.rs
@@ -53,7 +53,7 @@ impl TreeData {
     }
 
     pub async fn cache_tree_history(&self) {
-        if self.tree_history_size > 0 {
+        if self.tree_history_size != 0 {
             let mut tree_history = self.tree_history.write().await;
 
             if tree_history.len() == self.tree_history_size {
@@ -199,7 +199,7 @@ mod tests {
         // Then you can apply the remaining updates
         tree_data.insert_many_at(5, &identities[5..]).await;
 
-        for (i, identity) in identities.iter().enumerate().take(5) {
+        for (i, _identity) in identities.iter().enumerate().take(5) {
             let proof_from_world_tree = tree_data
                 .get_inclusion_proof(identities[i], Some(root))
                 .await
@@ -227,7 +227,7 @@ mod tests {
     }
 
     #[tokio::test]
-    async fn test_inclusion_proof_after_deletions() {
+    async fn test_get_inclusion_proof_after_deletions() {
         let (tree_data, mut ref_tree, identities) =
             initialize_tree_data(TREE_DEPTH, TREE_HISTORY_SIZE, NUM_IDENTITIES);
 

diff --git a/crates/tree_availability/tests/inclusion_proof.rs b/crates/tree_availability/tests/inclusion_proof.rs
@@ -26,7 +26,7 @@ async fn test_inclusion_proof() -> eyre::Result<()> {
         vec![U256::from(1), U256::from(2), U256::from(3)];
 
     let world_tree_creation_block =
-        middleware.get_block_number().await?.as_u64();
+        middleware.get_block_number().await?.as_u64() - 1;
 
     mock_world_id
         .register_identities(
@@ -57,7 +57,7 @@ async fn test_inclusion_proof() -> eyre::Result<()> {
     let tree_availability_service = TreeAvailabilityService::new(
         3,
         1,
-        0,
+        5,
         world_tree_address,
         world_tree_creation_block,
         middleware,