Skip to content

Commit

Permalink
add finalize tests
Browse files Browse the repository at this point in the history
  • Loading branch information
@Ninjatosba
Ninjatosba committed Dec 4, 2024
1 parent 05369f2 commit 866a0d4
Show file tree
Hide file tree
Showing 2 changed files with 274 additions and 4 deletions.
5 changes: 2 additions & 3 deletions contracts/stream/src/contract.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -5,9 +5,8 @@ use crate::stream::{compute_shares_amount, sync_stream, sync_stream_status};
use crate::{circuit_ops, ContractError};
use core::str;
use cosmwasm_std::{
attr, entry_point, to_json_binary, Addr, Attribute, BankMsg, Binary, Coin, CosmosMsg,
Decimal256, Deps, DepsMut, Env, MessageInfo, Order, Response, StdResult, Timestamp, Uint128,
Uint256,
attr, entry_point, to_json_binary, Addr, BankMsg, Binary, Coin, CosmosMsg, Decimal256, Deps,
DepsMut, Env, MessageInfo, Order, Response, StdResult, Timestamp, Uint128, Uint256,
};
use cw2::{ensure_from_older_version, set_contract_version};
use cw_storage_plus::Bound;
Expand Down
273 changes: 272 additions & 1 deletion tests/src/tests/streamswap_tests/finalize_stream.rs
View file
Original file line number Diff line number Diff line change
@@ -1,10 +1,12 @@
#[cfg(test)]
mod finalize_stream_tests {
use std::str::FromStr;

use crate::helpers::mock_messages::CreateStreamMsgBuilder;
use crate::helpers::suite::SuiteBuilder;
use crate::helpers::utils::{get_contract_address_from_res, get_funds_from_res};
use crate::helpers::{mock_messages::get_controller_inst_msg, suite::Suite};
use cosmwasm_std::{coin, Addr, BlockInfo, Coin, Uint128};
use cosmwasm_std::{coin, Addr, BlockInfo, Coin, Uint128, Uint256};
use cw_multi_test::Executor;
use streamswap_types::stream::{
ExecuteMsg as StreamSwapExecuteMsg, FinalizedStatus, QueryMsg as StreamSwapQueryMsg,
Expand Down Expand Up @@ -381,4 +383,273 @@ mod finalize_stream_tests {
Status::Finalized(FinalizedStatus::ThresholdReached)
);
}

#[test]
fn finalize_stream_threshold_not_reached() {
let Suite {
mut app,
test_accounts,
stream_swap_code_id,
stream_swap_controller_code_id,
vesting_code_id,
} = SuiteBuilder::default().build();
let start_time = app.block_info().time.plus_seconds(100);
let end_time = app.block_info().time.plus_seconds(200);
let bootstrapping_start_time = app.block_info().time.plus_seconds(50);
let threshold = Uint256::from(1_000u128);
let msg = get_controller_inst_msg(stream_swap_code_id, vesting_code_id, &test_accounts);
let controller_address = app
.instantiate_contract(
stream_swap_controller_code_id,
test_accounts.admin.clone(),
&msg,
&[],
"Controller".to_string(),
None,
)
.unwrap();
let create_stream_msg = CreateStreamMsgBuilder::new(
"Stream Swap tests",
test_accounts.creator_1.as_ref(),
coin(1_000_000, "out_denom"),
"in_denom",
bootstrapping_start_time,
start_time,
end_time,
)
.threshold(threshold)
.build();
let res = app
.execute_contract(
test_accounts.creator_1.clone(),
controller_address.clone(),
&create_stream_msg,
&[coin(100, "fee_denom"), coin(1_000_000, "out_denom")],
)
.unwrap();
let stream_swap_contract_address: String = get_contract_address_from_res(res);
let subscribe_msg = StreamSwapExecuteMsg::Subscribe {};
app.set_block(BlockInfo {
height: 1_100,
time: start_time,
chain_id: "test".to_string(),
});
// First subscription
let _res = app
.execute_contract(
test_accounts.subscriber_1.clone(),
Addr::unchecked(stream_swap_contract_address.clone()),
&subscribe_msg,
&[coin(200, "in_denom")],
)
.unwrap();
//Update environment time to end_time
app.set_block(BlockInfo {
height: 1_100,
time: end_time,
chain_id: "test".to_string(),
});

// Finalize stream with threshold not reached
let finalized_msg = StreamSwapExecuteMsg::FinalizeStream {
new_treasury: None,
create_pool: None,
salt: None,
};
let res = app
.execute_contract(
test_accounts.creator_1.clone(),
Addr::unchecked(stream_swap_contract_address.clone()),
&finalized_msg,
&[],
)
.unwrap();
let stream_swap_funds = get_funds_from_res(res);
assert_eq!(
stream_swap_funds,
vec![(
String::from(test_accounts.creator_1.clone()),
Coin {
denom: "out_denom".to_string(),
amount: Uint128::new(1_000_000)
}
),]
);
assert_eq!(stream_swap_funds.len(), 1);
}

#[test]
fn finalize_stream_threshold_not_reached_pool_refund() {
let Suite {
mut app,
test_accounts,
stream_swap_code_id,
stream_swap_controller_code_id,
vesting_code_id,
} = SuiteBuilder::default().build();
let start_time = app.block_info().time.plus_seconds(100);
let end_time = app.block_info().time.plus_seconds(200);
let bootstrapping_start_time = app.block_info().time.plus_seconds(50);
let threshold = Uint256::from(1_000u128);
let pool_creation_fee: Coin = coin(1000000, "fee_denom");
let pool_out_amount_clp: Uint256 = Uint256::from(500_000u128);
let msg = get_controller_inst_msg(stream_swap_code_id, vesting_code_id, &test_accounts);
let controller_address = app
.instantiate_contract(
stream_swap_controller_code_id,
test_accounts.admin.clone(),
&msg,
&[],
"Controller".to_string(),
None,
)
.unwrap();
let create_stream_msg = CreateStreamMsgBuilder::new(
"Stream Swap tests",
test_accounts.creator_1.as_ref(),
coin(1_000_000, "out_denom"),
"in_denom",
bootstrapping_start_time,
start_time,
end_time,
)
.threshold(threshold)
.pool_config(
streamswap_types::controller::PoolConfig::ConcentratedLiquidity {
out_amount_clp: pool_out_amount_clp,
},
)
.build();

let res = app
.execute_contract(
test_accounts.creator_1.clone(),
controller_address.clone(),
&create_stream_msg,
&[
coin(100, "fee_denom"),
coin(1_000_000, "out_denom"),
pool_creation_fee.clone(),
coin(
Uint128::from_str(pool_out_amount_clp.to_string().as_str())
.unwrap()
.u128(),
"out_denom",
),
],
)
.unwrap();
let stream_swap_contract_address: String = get_contract_address_from_res(res);
let subscribe_msg = StreamSwapExecuteMsg::Subscribe {};
app.set_block(BlockInfo {
height: 1_100,
time: start_time,
chain_id: "test".to_string(),
});
// First subscription
let _res = app
.execute_contract(
test_accounts.subscriber_1.clone(),
Addr::unchecked(stream_swap_contract_address.clone()),
&subscribe_msg,
&[coin(200, "in_denom")],
)
.unwrap();
//Update environment time to end_time
app.set_block(BlockInfo {
height: 1_100,
time: end_time,
chain_id: "test".to_string(),
});

// Finalize stream with threshold not reached
let finalized_msg = StreamSwapExecuteMsg::FinalizeStream {
new_treasury: None,
create_pool: None,
salt: None,
};
let res = app
.execute_contract(
test_accounts.creator_1.clone(),
Addr::unchecked(stream_swap_contract_address.clone()),
&finalized_msg,
&[],
)
.unwrap();
let stream_swap_funds = get_funds_from_res(res);

assert_eq!(
stream_swap_funds,
vec![
(
String::from(test_accounts.creator_1.clone()),
pool_creation_fee.clone()
),
(
String::from(test_accounts.creator_1.clone()),
Coin {
denom: "out_denom".to_string(),
amount: Uint128::new(1_000_000 + 500_000)
}
),
]
);
// Contract only has the subscription amount left in the contract
let contract_balance = app
.wrap()
.query_all_balances(Addr::unchecked(stream_swap_contract_address.clone()))
.unwrap();
assert_eq!(contract_balance.len(), 1);
assert_eq!(
contract_balance[0],
Coin {
denom: "in_denom".to_string(),
amount: Uint128::new(200)
}
);

// Query the stream status(Check stream status)
let stream: StreamResponse = app
.wrap()
.query_wasm_smart(
Addr::unchecked(stream_swap_contract_address.clone()),
&StreamSwapQueryMsg::Stream {},
)
.unwrap();

assert_eq!(
stream.status,
Status::Finalized(FinalizedStatus::ThresholdNotReached)
);

// Subscriber 1 exits the stream
let exit_stream_msg = StreamSwapExecuteMsg::ExitStream { salt: None };

let res = app
.execute_contract(
test_accounts.subscriber_1.clone(),
Addr::unchecked(stream_swap_contract_address.clone()),
&exit_stream_msg,
&[],
)
.unwrap();
let stream_swap_funds = get_funds_from_res(res);
assert_eq!(
stream_swap_funds,
vec![(
String::from(test_accounts.subscriber_1.clone()),
Coin {
denom: "in_denom".to_string(),
amount: Uint128::new(200)
}
),]
);

// Contract has no funds left
let contract_balance = app
.wrap()
.query_all_balances(Addr::unchecked(stream_swap_contract_address.clone()))
.unwrap();
assert_eq!(contract_balance.len(), 0);
}
}

0 comments on commit 866a0d4

Please sign in to comment.