Adapt unit tests to support ContractInfo query

confio · Nov 30, 2021 · bbbd3ce · bbbd3ce
1 parent ae227ef
commit bbbd3ce
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/contracts/tgrade-trusted-circle/src/tests/unit_tests.rs b/contracts/tgrade-trusted-circle/src/tests/unit_tests.rs
@@ -4,8 +4,8 @@ use std::marker::PhantomData;
 
 use cosmwasm_std::testing::{MockApi, MockStorage};
 use cosmwasm_std::{
-    Addr, Binary, ContractResult, Deps, Empty, QuerierResult, QueryRequest, StdError, SubMsg,
-    SystemError, SystemResult, WasmQuery,
+    to_binary, Addr, Binary, ContractResult, Deps, Empty, QuerierResult, QueryRequest, StdError,
+    SubMsg, SystemError, SystemResult, WasmQuery,
 };
 use cw_storage_plus::Item;
 
@@ -45,6 +45,10 @@ impl TokenQuerier {
                     kind: "WasmQuery::Smart".to_string(),
                 })
             }
+            QueryRequest::Wasm(WasmQuery::ContractInfo { contract_addr }) => {
+                // FIXME: Implement a proper `ContractInfo` mock query
+                self.query_wasm(contract_addr, to_binary(b"contract_info").unwrap())
+            }
             _ => SystemResult::Err(SystemError::UnsupportedRequest {
                 kind: "not wasm".to_string(),
             }),