Return NOT_FOUND if identity is missing (#130)

* Return NOT_FOUND if identity is missing * Fix test
worldcoin · Oct 7, 2024 · cbb2798 · cbb2798
1 parent af8ffe6
commit cbb2798
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/src/tree/service.rs b/src/tree/service.rs
@@ -128,7 +128,12 @@ pub async fn inclusion_proof(
         .inclusion_proof(req.identity_commitment, chain_id)
         .await?;
 
-    Ok((StatusCode::OK, Json(inclusion_proof)))
+    match inclusion_proof {
+        Some(inclusion_proof) => {
+            Ok((StatusCode::OK, Json(Some(inclusion_proof))))
+        }
+        None => Ok((StatusCode::NOT_FOUND, Json(None))),
+    }
 }
 
 #[derive(Serialize, Deserialize, Debug)]

diff --git a/tests/common/test_client.rs b/tests/common/test_client.rs
@@ -59,7 +59,11 @@ impl TestClient {
             .send()
             .await?;
 
-        response.error_for_status_ref()?;
+        if response.status() == StatusCode::NOT_FOUND {
+            return Ok(None);
+        } else {
+            response.error_for_status_ref()?;
+        }
 
         Ok(response.json().await?)
     }