mock-cosmwasm-contract 0.1.2

Mock CosmWasm contract that can set predetermined responses to queries and keep track of execution for testing purposes.
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

use cosmwasm_schema::cw_serde;
use cosmwasm_std::{to_json_string, StdResult};
use serde::Serialize;

#[cw_serde]
pub struct InstantiateMsg {}

#[cw_serde]
pub enum ExecuteMsg {
    #[serde(rename = "@set_stub_query")]
    SetStubQuery { query: QueryMsg, response: String },

    #[serde(rename = "@remove_stub_query")]
    RemoveStubQuery { query: QueryMsg },
}

pub fn set_stub_query_msg<Q, R>(query: &Q, response: &R) -> StdResult<ExecuteMsg>
where
    Q: Serialize + ?Sized,
    R: Serialize + ?Sized,
{
    Ok(ExecuteMsg::SetStubQuery {
        query: query_msg(query)?,
        response: to_json_string(&response)?,
    })
}

pub fn remove_stub_query_msg<Q>(query: &Q) -> StdResult<ExecuteMsg>
where
    Q: Serialize + ?Sized,
{
    Ok(ExecuteMsg::RemoveStubQuery {
        query: query_msg(query)?,
    })
}

#[cw_serde]
#[serde(transparent)]
pub struct QueryMsg(pub String);

pub fn query_msg<T>(msg: &T) -> StdResult<QueryMsg>
where
    T: Serialize + ?Sized,
{
    Ok(QueryMsg(to_json_string(&msg)?))
}