-
Notifications
You must be signed in to change notification settings - Fork 32
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
25bdfbd
commit abebb3e
Showing
6 changed files
with
1,273 additions
and
1,257 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
231 changes: 111 additions & 120 deletions
231
contracts/neutron_interchain_queries/src/testing/mock_querier.rs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,120 +1,111 @@ | ||
| // use std::collections::HashMap; | ||
| // use std::marker::PhantomData; | ||
| // | ||
| // use cosmwasm_std::testing::{MockApi, MockQuerier, MockStorage}; | ||
| // use cosmwasm_std::{ | ||
| // from_json, Binary, Coin, ContractResult, CustomQuery, FullDelegation, OwnedDeps, Querier, | ||
| // QuerierResult, QueryRequest, SystemError, SystemResult, Uint128, Validator, | ||
| // }; | ||
| // use schemars::JsonSchema; | ||
| // use serde::{Deserialize, Serialize}; | ||
| // | ||
| // pub const MOCK_CONTRACT_ADDR: &str = "cosmos2contract"; | ||
| // | ||
| // #[derive(Serialize, Deserialize, Clone, Debug, PartialEq, Eq, JsonSchema)] | ||
| // #[serde(rename_all = "snake_case")] | ||
| // pub struct CustomQueryWrapper {} | ||
| // | ||
| // // implement custom query | ||
| // impl CustomQuery for CustomQueryWrapper {} | ||
| // | ||
| // pub fn mock_dependencies( | ||
| // contract_balance: &[Coin], | ||
| // ) -> OwnedDeps<MockStorage, MockApi, WasmMockQuerier> { | ||
| // let contract_addr = MOCK_CONTRACT_ADDR; | ||
| // let custom_querier: WasmMockQuerier = | ||
| // WasmMockQuerier::new(MockQuerier::new(&[(contract_addr, contract_balance)])); | ||
| // | ||
| // OwnedDeps { | ||
| // storage: MockStorage::default(), | ||
| // api: MockApi::default(), | ||
| // querier: custom_querier, | ||
| // custom_query_type: PhantomData, | ||
| // } | ||
| // } | ||
| // | ||
| // pub struct WasmMockQuerier { | ||
| // base: MockQuerier, | ||
| // query_responses: HashMap<u64, Binary>, | ||
| // registered_queries: HashMap<u64, Binary>, | ||
| // } | ||
| // | ||
| // impl Querier for WasmMockQuerier { | ||
| // fn raw_query(&self, bin_request: &[u8]) -> QuerierResult { | ||
| // let request: QueryRequest = match from_json(bin_request) { | ||
| // Ok(v) => v, | ||
| // Err(e) => { | ||
| // return QuerierResult::Err(SystemError::InvalidRequest { | ||
| // error: format!("Parsing query request: {}", e), | ||
| // request: bin_request.into(), | ||
| // }); | ||
| // } | ||
| // }; | ||
| // self.handle_query(&request) | ||
| // } | ||
| // } | ||
| // | ||
| // // TODO: fix | ||
| // impl WasmMockQuerier { | ||
| // pub fn handle_query(&self, request: &QueryRequest) -> QuerierResult { | ||
| // match &request { | ||
| // QueryRequest::Custom(NeutronQuery::InterchainQueryResult { query_id }) => { | ||
| // SystemResult::Ok(ContractResult::Ok( | ||
| // (*self.query_responses.get(query_id).unwrap()).clone(), | ||
| // )) | ||
| // } | ||
| // QueryRequest::Custom(NeutronQuery::RegisteredInterchainQuery { query_id }) => { | ||
| // SystemResult::Ok(ContractResult::Ok( | ||
| // (*self.registered_queries.get(query_id).unwrap()).clone(), | ||
| // )) | ||
| // } | ||
| // QueryRequest::Custom(NeutronQuery::RegisteredInterchainQueries { | ||
| // owners: _owners, | ||
| // connection_id: _connection_id, | ||
| // pagination: _pagination, | ||
| // }) => { | ||
| // todo!() | ||
| // } | ||
| // QueryRequest::Custom(NeutronQuery::InterchainAccountAddress { .. }) => { | ||
| // todo!() | ||
| // } | ||
| // _ => self.base.handle_query(request), | ||
| // } | ||
| // } | ||
| // | ||
| // pub fn _update_staking( | ||
| // &mut self, | ||
| // denom: &str, | ||
| // validators: &[Validator], | ||
| // delegations: &[FullDelegation], | ||
| // ) { | ||
| // self.base.staking.update(denom, validators, delegations); | ||
| // } | ||
| // | ||
| // pub fn add_query_response(&mut self, query_id: u64, response: Binary) { | ||
| // self.query_responses.insert(query_id, response); | ||
| // } | ||
| // pub fn add_registered_queries(&mut self, query_id: u64, response: Binary) { | ||
| // self.registered_queries.insert(query_id, response); | ||
| // } | ||
| // } | ||
| // | ||
| // #[derive(Clone, Default)] | ||
| // pub struct BalanceQuerier { | ||
| // _balances: HashMap<String, Coin>, | ||
| // } | ||
| // | ||
| // #[derive(Clone, Default)] | ||
| // pub struct TokenQuerier { | ||
| // _balances: HashMap<String, HashMap<String, Uint128>>, | ||
| // } | ||
| // | ||
| // impl WasmMockQuerier { | ||
| // pub fn new(base: MockQuerier) -> Self { | ||
| // WasmMockQuerier { | ||
| // base, | ||
| // query_responses: HashMap::new(), | ||
| // registered_queries: HashMap::new(), | ||
| // } | ||
| // } | ||
| // } | ||
| use cosmwasm_std::testing::{MockApi, MockQuerier, MockStorage}; | ||
| use cosmwasm_std::{ | ||
| from_json, Binary, Coin, ContractResult, CustomQuery, FullDelegation, GrpcQuery, OwnedDeps, | ||
| Querier, QuerierResult, QueryRequest, SystemError, SystemResult, Uint128, Validator, | ||
| }; | ||
| use neutron_std::types::neutron::interchainqueries::{ | ||
| QueryRegisteredQueryRequest, QueryRegisteredQueryResultRequest, | ||
| }; | ||
| use schemars::JsonSchema; | ||
| use serde::{Deserialize, Serialize}; | ||
| use std::collections::HashMap; | ||
| use std::marker::PhantomData; | ||
|
|
||
| pub const MOCK_CONTRACT_ADDR: &str = "cosmos2contract"; | ||
|
|
||
| #[derive(Serialize, Deserialize, Clone, Debug, PartialEq, Eq, JsonSchema)] | ||
| #[serde(rename_all = "snake_case")] | ||
| pub struct CustomQueryWrapper {} | ||
|
|
||
| // implement custom query | ||
| impl CustomQuery for CustomQueryWrapper {} | ||
|
|
||
| pub fn mock_dependencies( | ||
| contract_balance: &[Coin], | ||
| ) -> OwnedDeps<MockStorage, MockApi, WasmMockQuerier> { | ||
| let contract_addr = MOCK_CONTRACT_ADDR; | ||
| let custom_querier: WasmMockQuerier = | ||
| WasmMockQuerier::new(MockQuerier::new(&[(contract_addr, contract_balance)])); | ||
|
|
||
| OwnedDeps { | ||
| storage: MockStorage::default(), | ||
| api: MockApi::default(), | ||
| querier: custom_querier, | ||
| custom_query_type: PhantomData, | ||
| } | ||
| } | ||
|
|
||
| pub struct WasmMockQuerier { | ||
| base: MockQuerier, | ||
| query_responses: HashMap<u64, Binary>, | ||
| registered_queries: HashMap<u64, Binary>, | ||
| } | ||
|
|
||
| impl Querier for WasmMockQuerier { | ||
| fn raw_query(&self, bin_request: &[u8]) -> QuerierResult { | ||
| let request: QueryRequest = match from_json(bin_request) { | ||
| Ok(v) => v, | ||
| Err(e) => { | ||
| return QuerierResult::Err(SystemError::InvalidRequest { | ||
| error: format!("Parsing query request: {}", e), | ||
| request: bin_request.into(), | ||
| }); | ||
| } | ||
| }; | ||
| self.handle_query(&request) | ||
| } | ||
| } | ||
|
|
||
| impl WasmMockQuerier { | ||
| pub fn handle_query(&self, request: &QueryRequest) -> QuerierResult { | ||
| match &request { | ||
| QueryRequest::Grpc(GrpcQuery { path, data }) => { | ||
| let quoted_path = path.trim_matches('"').to_string(); | ||
| if "ed_path == "/neutron.interchainqueries.Query/QueryResult" { | ||
| let request: QueryRegisteredQueryResultRequest = | ||
| ::prost::Message::decode(&data[..]).unwrap(); | ||
| SystemResult::Ok(ContractResult::Ok( | ||
| (*self.query_responses.get(&request.query_id).unwrap()).clone(), | ||
| )) | ||
| } else if "ed_path == "/neutron.interchainqueries.Query/RegisteredQuery" { | ||
| let request: QueryRegisteredQueryRequest = | ||
| ::prost::Message::decode(&data[..]).unwrap(); | ||
| SystemResult::Ok(ContractResult::Ok( | ||
| (*self.registered_queries.get(&request.query_id).unwrap()).clone(), | ||
| )) | ||
| } else { | ||
| println!("PATH: {}", quoted_path); | ||
| self.base.handle_query(request) | ||
| } | ||
| } | ||
| _ => self.base.handle_query(request), | ||
| } | ||
| } | ||
|
|
||
| pub fn add_query_response(&mut self, query_id: u64, response: Binary) { | ||
| self.query_responses.insert(query_id, response); | ||
| } | ||
| pub fn add_registered_queries(&mut self, query_id: u64, response: Binary) { | ||
| self.registered_queries.insert(query_id, response); | ||
| } | ||
| } | ||
|
|
||
| #[derive(Clone, Default)] | ||
| pub struct BalanceQuerier { | ||
| _balances: HashMap<String, Coin>, | ||
| } | ||
|
|
||
| #[derive(Clone, Default)] | ||
| pub struct TokenQuerier { | ||
| _balances: HashMap<String, HashMap<String, Uint128>>, | ||
| } | ||
|
|
||
| impl WasmMockQuerier { | ||
| pub fn new(base: MockQuerier) -> Self { | ||
| WasmMockQuerier { | ||
| base, | ||
| query_responses: HashMap::new(), | ||
| registered_queries: HashMap::new(), | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.