cw-asset 4.0.0

Helper library for interacting with Cosmos assets (native coins and CW20 tokens)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

use cosmwasm_std::{
    from_json, testing::MockQuerier, Addr, Coin, Empty, Querier, QuerierResult, QueryRequest,
    StdResult, SystemError, WasmQuery,
};
use cw20::Cw20QueryMsg;

use super::cw20_querier::Cw20Querier;

pub struct CustomMockQuerier {
    base: MockQuerier<Empty>,
    cw20_querier: Cw20Querier,
}

impl Default for CustomMockQuerier {
    fn default() -> Self {
        CustomMockQuerier {
            base: MockQuerier::<Empty>::new(&[]),
            cw20_querier: Cw20Querier::default(),
        }
    }
}

impl Querier for CustomMockQuerier {
    fn raw_query(&self, bin_request: &[u8]) -> QuerierResult {
        let request: QueryRequest<Empty> = match from_json(bin_request) {
            Ok(v) => v,
            Err(e) => {
                return Err(SystemError::InvalidRequest {
                    error: format!("[mock]: failed to parse query request {e}"),
                    request: bin_request.into(),
                })
                .into()
            },
        };
        self.handle_query(&request)
    }
}

impl CustomMockQuerier {
    pub fn handle_query(&self, request: &QueryRequest<Empty>) -> QuerierResult {
        match request {
            QueryRequest::Wasm(WasmQuery::Smart {
                contract_addr,
                msg,
            }) => {
                let contract_addr = Addr::unchecked(contract_addr);

                let parse_cw20_query: StdResult<Cw20QueryMsg> = from_json(msg);
                if let Ok(cw20_query) = parse_cw20_query {
                    return self.cw20_querier.handle_query(&contract_addr, cw20_query);
                }

                panic!("[mock]: unsupported wasm query {msg:?}");
            },

            _ => self.base.handle_query(request),
        }
    }

    pub fn set_base_balances(&mut self, address: &str, balances: &[Coin]) {
        self.base.bank.update_balance(address, balances.to_vec());
    }

    pub fn set_cw20_balance(&mut self, contract: &str, user: &str, balance: u128) {
        self.cw20_querier.set_balance(contract, user, balance);
    }
}