diff --git a/contracts/fee_splitter/src/testing.rs b/contracts/fee_splitter/src/testing.rs index 763d702..4c8bbd6 100644 --- a/contracts/fee_splitter/src/testing.rs +++ b/contracts/fee_splitter/src/testing.rs @@ -13,18 +13,26 @@ use crate::{ state::Config, }; +const SENDER: &str = "addr0000"; +const FIRST_RECIPIENT: &str = "address0000"; +const SECOND_RECIPIENT: &str = "address0001"; +const ATOM: &str = "ATOM"; +const TIA: &str = "TIA"; +const USDT: &str = "USDT"; +const CW20_ASSET_ONE: &str = "asset0000"; +const CW20_ASSET_TWO: &str = "asset0001"; + #[test] fn init_works() { let mut deps = mock_coreum_deps(&[]); let env = mock_env(); - let sender = "addr0000"; - let info = mock_info(sender, &[]); + let info = mock_info(SENDER, &[]); - let first_addr_pecnt = ("address0000".to_string(), Decimal::percent(50u64)); - let second_addr_pecnt = ("address0001".to_string(), Decimal::percent(50u64)); + let first_addr_percent = (FIRST_RECIPIENT.to_string(), Decimal::percent(50u64)); + let second_addr_percent = (SECOND_RECIPIENT.to_string(), Decimal::percent(50u64)); let msg = InstantiateMsg { - addresses: vec![first_addr_pecnt.clone(), second_addr_pecnt.clone()], - cw20_contracts: vec!["USDT".to_string()], + addresses: vec![first_addr_percent.clone(), second_addr_percent.clone()], + cw20_contracts: vec![USDT.to_string()], }; let res = instantiate(deps.as_mut(), env, info, msg).unwrap(); @@ -42,14 +50,13 @@ fn init_works() { fn fails_to_init_because_weights_not_correct() { let mut deps = mock_coreum_deps(&[]); let env = mock_env(); - let sender = "addr0000"; - let info = mock_info(sender, &[]); + let info = mock_info(SENDER, &[]); - let first_tupple = ("ATOM".to_string(), Decimal::percent(50u64)); - let second_tuple = ("TIA".to_string(), Decimal::percent(60u64)); + let first_addr_percent = (FIRST_RECIPIENT.to_string(), Decimal::percent(50u64)); + let second_addr_percent = (SECOND_RECIPIENT.to_string(), Decimal::percent(60u64)); let msg = InstantiateMsg { - addresses: vec![first_tupple.clone(), second_tuple.clone()], - cw20_contracts: vec!["USDT".to_string()], + addresses: vec![first_addr_percent.clone(), second_addr_percent.clone()], + cw20_contracts: vec![USDT.to_string()], }; let res = instantiate(deps.as_mut(), env, info, msg).unwrap_err(); @@ -61,7 +68,7 @@ fn should_send_tokens_in_correct_amount() { let mut deps = mock_coreum_deps(&[]); deps.querier.with_token_balances(&[( - &String::from("asset0000"), + &String::from(CW20_ASSET_ONE), &[(&String::from(MOCK_CONTRACT_ADDR), &Uint128::new(100_000))], )]); @@ -69,11 +76,11 @@ fn should_send_tokens_in_correct_amount() { &String::from(MOCK_CONTRACT_ADDR), &[ Coin { - denom: "ATOM".to_string(), + denom: ATOM.to_string(), amount: Uint128::new(100_000), }, Coin { - denom: "TIA".to_string(), + denom: TIA.to_string(), amount: Uint128::new(100_000), }, ], @@ -81,15 +88,13 @@ fn should_send_tokens_in_correct_amount() { let env = mock_env(); - let sender = "addr0000"; - - let info = mock_info(sender, &[]); + let info = mock_info(SENDER, &[]); let msg = InstantiateMsg { addresses: vec![ - ("address0000".to_string(), Decimal::percent(60u64)), - ("address0001".to_string(), Decimal::percent(40u64)), + (FIRST_RECIPIENT.to_string(), Decimal::percent(60u64)), + (SECOND_RECIPIENT.to_string(), Decimal::percent(40u64)), ], - cw20_contracts: vec!["asset0000".to_string(), "asset0001".to_string()], + cw20_contracts: vec![CW20_ASSET_ONE.to_string(), CW20_ASSET_TWO.to_string()], }; let fee_splitter_instance = instantiate(deps.as_mut(), env.clone(), info.clone(), msg).unwrap(); @@ -103,8 +108,8 @@ fn should_send_tokens_in_correct_amount() { ); let msg = ExecuteMsg::SendTokens { - native_denoms: vec!["ATOM".to_string(), "TIA".to_string()], - cw20_addresses: vec!["asset0000".to_string()], + native_denoms: vec![ATOM.to_string(), TIA.to_string()], + cw20_addresses: vec![CW20_ASSET_ONE.to_string()], }; let res = execute(deps.as_mut(), env.clone(), info, msg).unwrap(); @@ -115,14 +120,14 @@ fn should_send_tokens_in_correct_amount() { SubMsg { id: 0, msg: CosmosMsg::Bank(BankMsg::Send { - to_address: "address0000".to_string(), + to_address: FIRST_RECIPIENT.to_string(), amount: vec![ Coin { - denom: "ATOM".to_string(), + denom: ATOM.to_string(), amount: Uint128::new(60_000), }, Coin { - denom: "TIA".to_string(), + denom: TIA.to_string(), amount: Uint128::new(60_000), } ] @@ -133,9 +138,9 @@ fn should_send_tokens_in_correct_amount() { SubMsg { id: 0, msg: CosmosMsg::Wasm(WasmMsg::Execute { - contract_addr: "asset0000".to_string(), + contract_addr: CW20_ASSET_ONE.to_string(), msg: to_json_binary(&Cw20ExecuteMsg::Transfer { - recipient: "address0000".to_string(), + recipient: FIRST_RECIPIENT.to_string(), amount: Uint128::new(60_000), }) .unwrap(), @@ -147,14 +152,14 @@ fn should_send_tokens_in_correct_amount() { SubMsg { id: 0, msg: CosmosMsg::Bank(BankMsg::Send { - to_address: "address0001".to_string(), + to_address: SECOND_RECIPIENT.to_string(), amount: vec![ Coin { - denom: "ATOM".to_string(), + denom: ATOM.to_string(), amount: Uint128::new(40_000), }, Coin { - denom: "TIA".to_string(), + denom: TIA.to_string(), amount: Uint128::new(40_000), } ] @@ -165,9 +170,9 @@ fn should_send_tokens_in_correct_amount() { SubMsg { id: 0, msg: CosmosMsg::Wasm(WasmMsg::Execute { - contract_addr: "asset0000".to_string(), + contract_addr: CW20_ASSET_ONE.to_string(), msg: to_json_binary(&Cw20ExecuteMsg::Transfer { - recipient: "address0001".to_string(), + recipient: SECOND_RECIPIENT.to_string(), amount: Uint128::new(40_000), }) .unwrap(), @@ -188,8 +193,8 @@ fn should_send_tokens_in_correct_amount() { config_res, Config { addresses: vec![ - ("address0000".to_string(), Decimal::percent(60)), - ("address0001".to_string(), Decimal::percent(40)) + (FIRST_RECIPIENT.to_string(), Decimal::percent(60)), + (SECOND_RECIPIENT.to_string(), Decimal::percent(40)) ], } );