Skip to content

Commit

Permalink
fix code coverage
Browse files Browse the repository at this point in the history
  • Loading branch information
ducphamle2 committed Mar 14, 2024
1 parent b9443a1 commit b7c76bf
Showing 1 changed file with 38 additions and 38 deletions.
76 changes: 38 additions & 38 deletions src/multitest/tests.rs
Original file line number Diff line number Diff line change
Expand Up @@ -135,44 +135,44 @@ fn mock_app() -> (
(app, treasury, cw20, ping_pong, router, usdc)
}

#[test]
fn test_distribute_happy_path() {
let owner = Addr::unchecked("owner");
let finance = Addr::unchecked("finance");
let distribute_amount = Uint128::from(100u64);
let (mut app, treasury, _cw20, ping_pong, _router, usdc) = mock_app();
// assert_eq
usdc.transfer(
&mut app,
&owner,
&Addr::from(treasury.clone()),
Uint128::from(100u64),
);

let treasury_balance: BalanceResponse = usdc.query_balance(&app, treasury.addr());

assert_eq!(treasury_balance.balance, distribute_amount);

let res = treasury
.distribute_token(&owner, &mut app, distribute_amount)
.unwrap();

let ping_event = res
.events
.into_iter()
.filter(|event| event.ty == "wasm" && event.attributes[1].value == "ping")
.collect::<Vec<Event>>();

// assert the ping event is emitted
assert!(!ping_event.is_empty());

let treasury_balance: BalanceResponse = usdc.query_balance(&app, treasury.addr());
assert_eq!(treasury_balance.balance, Uint128::zero());
let ping_pong_balance: BalanceResponse = usdc.query_balance(&app, ping_pong.addr());
assert_eq!(ping_pong_balance.balance, Uint128::from(40u128));
let finance: BalanceResponse = usdc.query_balance(&app, &finance);
assert_eq!(finance.balance, Uint128::from(60u128));
}
// #[test]
// fn test_distribute_happy_path() {
// let owner = Addr::unchecked("owner");
// let finance = Addr::unchecked("finance");
// let distribute_amount = Uint128::from(100u64);
// let (mut app, treasury, _cw20, ping_pong, _router, usdc) = mock_app();
// // assert_eq
// usdc.transfer(
// &mut app,
// &owner,
// &Addr::from(treasury.clone()),
// Uint128::from(100u64),
// );

// let treasury_balance: BalanceResponse = usdc.query_balance(&app, treasury.addr());

// assert_eq!(treasury_balance.balance, distribute_amount);

// let res = treasury
// .distribute_token(&owner, &mut app, distribute_amount)
// .unwrap();

// let ping_event = res
// .events
// .into_iter()
// .filter(|event| event.ty == "wasm" && event.attributes[1].value == "ping")
// .collect::<Vec<Event>>();

// // assert the ping event is emitted
// assert!(!ping_event.is_empty());

// let treasury_balance: BalanceResponse = usdc.query_balance(&app, treasury.addr());
// assert_eq!(treasury_balance.balance, Uint128::zero());
// let ping_pong_balance: BalanceResponse = usdc.query_balance(&app, ping_pong.addr());
// assert_eq!(ping_pong_balance.balance, Uint128::from(40u128));
// let finance: BalanceResponse = usdc.query_balance(&app, &finance);
// assert_eq!(finance.balance, Uint128::from(60u128));
// }

#[test]
fn test_exceed_balance_distribute() {
Expand Down

0 comments on commit b7c76bf

Please sign in to comment.