fix: state sync panics on no data (0xPolygonMiden#52)

lambdaclass · Dec 12, 2023 · bb7b26b · bb7b26b
1 parent 3d82958
commit bb7b26b
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/src/mock.rs b/src/mock.rs
@@ -31,12 +31,15 @@ impl MockRpcApi {
         request: impl tonic::IntoRequest<SyncStateRequest>,
     ) -> std::result::Result<tonic::Response<SyncStateResponse>, tonic::Status> {
         let request = request.into_request().into_inner();
-        let response = self
-            .sync_state_requests
-            .get(&request)
-            .expect("no response for sync state request")
-            .clone();
-        Ok(tonic::Response::new(response))
+        match self.sync_state_requests.get(&request) {
+            Some(response) => {
+                let response = response.clone();
+                Ok(tonic::Response::new(response))
+            }
+            None => Err(tonic::Status::not_found(
+                "no response for sync state request",
+            )),
+        }
     }
 }