update mock querier for grpc

provenance-io · Aug 30, 2024 · 0573bde · 0573bde
1 parent 9a8f7b7
commit 0573bde
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/packages/provwasm-mocks/src/querier.rs b/packages/provwasm-mocks/src/querier.rs
@@ -37,6 +37,14 @@ impl MockProvenanceQuerier {
                 }
                 SystemResult::Err(SystemError::UnsupportedRequest { kind: path.into() })
             }
+            QueryRequest::Grpc(grpc_query) => {
+                if let Some(response_fn) = self.registered_custom_queries.get(&grpc_query.path) {
+                    return response_fn(&grpc_query.data);
+                }
+                SystemResult::Err(SystemError::UnsupportedRequest {
+                    kind: grpc_query.path.to_string(),
+                })
+            }
             _ => self.mock_querier.handle_query(request),
         }
     }