cw-denom 2.4.2

A package for validation and handling of cw20 and native Cosmos SDK denominations.
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use cosmwasm_std::{coins, Addr, Empty, Uint128};
use cw20::Cw20Coin;
use cw_multi_test::{App, BankSudo, Contract, ContractWrapper, Executor};

use crate::CheckedDenom;

fn cw20_contract() -> Box<dyn Contract<Empty>> {
    let contract = ContractWrapper::new(
        cw20_base::contract::execute,
        cw20_base::contract::instantiate,
        cw20_base::contract::query,
    );
    Box::new(contract)
}

#[test]
fn test_cw20_denom_send() {
    let mut app = App::default();

    let cw20_id = app.store_code(cw20_contract());
    let cw20 = app
        .instantiate_contract(
            cw20_id,
            Addr::unchecked("ekez"),
            &cw20_base::msg::InstantiateMsg {
                name: "token".to_string(),
                symbol: "symbol".to_string(),
                decimals: 6,
                initial_balances: vec![Cw20Coin {
                    amount: Uint128::new(10),
                    address: "ekez".to_string(),
                }],
                mint: None,
                marketing: None,
            },
            &[],
            "token contract",
            None,
        )
        .unwrap();

    let denom = CheckedDenom::Cw20(cw20);

    let start_balance = denom
        .query_balance(&app.wrap(), &Addr::unchecked("ekez"))
        .unwrap();
    let send_message = denom
        .get_transfer_to_message(&Addr::unchecked("dao"), Uint128::new(9))
        .unwrap();
    app.execute(Addr::unchecked("ekez"), send_message).unwrap();
    let end_balance = denom
        .query_balance(&app.wrap(), &Addr::unchecked("ekez"))
        .unwrap();

    assert_eq!(start_balance, Uint128::new(10));
    assert_eq!(end_balance, Uint128::new(1));

    let dao_balance = denom
        .query_balance(&app.wrap(), &Addr::unchecked("dao"))
        .unwrap();
    assert_eq!(dao_balance, Uint128::new(9))
}

#[test]
fn test_native_denom_send() {
    let mut app = App::default();
    app.sudo(cw_multi_test::SudoMsg::Bank(BankSudo::Mint {
        to_address: "ekez".to_string(),
        amount: coins(10, "ujuno"),
    }))
    .unwrap();

    let denom = CheckedDenom::Native("ujuno".to_string());

    let start_balance = denom
        .query_balance(&app.wrap(), &Addr::unchecked("ekez"))
        .unwrap();
    let send_message = denom
        .get_transfer_to_message(&Addr::unchecked("dao"), Uint128::new(9))
        .unwrap();
    app.execute(Addr::unchecked("ekez"), send_message).unwrap();
    let end_balance = denom
        .query_balance(&app.wrap(), &Addr::unchecked("ekez"))
        .unwrap();

    assert_eq!(start_balance, Uint128::new(10));
    assert_eq!(end_balance, Uint128::new(1));

    let dao_balance = denom
        .query_balance(&app.wrap(), &Addr::unchecked("dao"))
        .unwrap();
    assert_eq!(dao_balance, Uint128::new(9))
}