2022-03-17 06:17:28 -07:00
|
|
|
#![cfg(feature = "test-bpf")]
|
|
|
|
|
2022-05-25 10:29:53 -07:00
|
|
|
use fixed::types::I80F48;
|
2022-05-25 02:27:58 -07:00
|
|
|
use mango_v4::state::*;
|
2022-03-17 06:17:28 -07:00
|
|
|
use solana_program_test::*;
|
2022-09-07 03:39:21 -07:00
|
|
|
use solana_sdk::transport::TransportError;
|
2022-03-17 06:17:28 -07:00
|
|
|
|
|
|
|
use program_test::*;
|
|
|
|
|
2022-08-26 06:59:47 -07:00
|
|
|
use mango_setup::*;
|
|
|
|
|
2022-03-17 06:17:28 -07:00
|
|
|
mod program_test;
|
|
|
|
|
|
|
|
// Try to reach compute limits in health checks by having many different tokens in an account
|
|
|
|
#[tokio::test]
|
2022-05-18 08:16:14 -07:00
|
|
|
async fn test_health_compute_tokens() -> Result<(), TransportError> {
|
2022-03-21 02:45:55 -07:00
|
|
|
let context = TestContext::new().await;
|
2022-03-17 06:17:28 -07:00
|
|
|
let solana = &context.solana.clone();
|
|
|
|
|
2022-09-07 03:39:21 -07:00
|
|
|
let admin = TestKeypair::new();
|
|
|
|
let owner = context.users[0].key;
|
|
|
|
let payer = context.users[1].key;
|
2022-03-17 06:17:28 -07:00
|
|
|
let mints = &context.mints[0..10];
|
|
|
|
|
|
|
|
//
|
|
|
|
// SETUP: Create a group and an account
|
|
|
|
//
|
|
|
|
|
2022-08-26 06:59:47 -07:00
|
|
|
let GroupWithTokens { group, .. } = GroupWithTokensConfig {
|
2022-03-20 23:49:51 -07:00
|
|
|
admin,
|
|
|
|
payer,
|
2022-09-12 06:25:50 -07:00
|
|
|
mints: mints.to_vec(),
|
|
|
|
..GroupWithTokensConfig::default()
|
2022-03-20 23:49:51 -07:00
|
|
|
}
|
|
|
|
.create(solana)
|
|
|
|
.await;
|
2022-03-17 06:17:28 -07:00
|
|
|
|
2022-08-26 06:59:47 -07:00
|
|
|
// each deposit ends with a health check
|
|
|
|
create_funded_account(&solana, group, owner, 0, &context.users[1], mints, 1000, 0).await;
|
2022-03-17 06:17:28 -07:00
|
|
|
|
|
|
|
// TODO: actual explicit CU comparisons.
|
2022-11-21 06:01:10 -08:00
|
|
|
// On 2022-11-18 the final deposit costs 45495 CU and each new token increases it by roughly 1729 CU
|
2022-03-17 06:17:28 -07:00
|
|
|
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
|
|
|
// Try to reach compute limits in health checks by having many serum markets in an account
|
|
|
|
#[tokio::test]
|
2022-05-18 08:16:14 -07:00
|
|
|
async fn test_health_compute_serum() -> Result<(), TransportError> {
|
2022-03-21 02:45:55 -07:00
|
|
|
let context = TestContext::new().await;
|
2022-03-17 06:17:28 -07:00
|
|
|
let solana = &context.solana.clone();
|
|
|
|
|
2022-09-07 03:39:21 -07:00
|
|
|
let admin = TestKeypair::new();
|
|
|
|
let owner = context.users[0].key;
|
|
|
|
let payer = context.users[1].key;
|
2022-03-17 06:17:28 -07:00
|
|
|
let mints = &context.mints[0..8];
|
|
|
|
let payer_mint_accounts = &context.users[1].token_accounts[0..mints.len()];
|
|
|
|
|
|
|
|
//
|
|
|
|
// SETUP: Create a group and an account
|
|
|
|
//
|
|
|
|
|
2022-07-04 04:13:11 -07:00
|
|
|
let mango_setup::GroupWithTokens { group, tokens, .. } = mango_setup::GroupWithTokensConfig {
|
2022-03-20 23:49:51 -07:00
|
|
|
admin,
|
|
|
|
payer,
|
2022-09-12 06:25:50 -07:00
|
|
|
mints: mints.to_vec(),
|
|
|
|
..GroupWithTokensConfig::default()
|
2022-03-20 23:49:51 -07:00
|
|
|
}
|
|
|
|
.create(solana)
|
|
|
|
.await;
|
2022-03-17 06:17:28 -07:00
|
|
|
|
|
|
|
let account = send_tx(
|
|
|
|
solana,
|
2022-07-06 05:51:15 -07:00
|
|
|
AccountCreateInstruction {
|
2022-03-17 06:17:28 -07:00
|
|
|
account_num: 0,
|
2022-08-07 05:16:06 -07:00
|
|
|
token_count: 16,
|
|
|
|
serum3_count: 8,
|
|
|
|
perp_count: 8,
|
|
|
|
perp_oo_count: 8,
|
2022-03-17 06:17:28 -07:00
|
|
|
group,
|
|
|
|
owner,
|
|
|
|
payer,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap()
|
|
|
|
.account;
|
|
|
|
|
|
|
|
//
|
|
|
|
// SETUP: Create serum markets and register them
|
|
|
|
//
|
2022-03-30 03:24:07 -07:00
|
|
|
let quote_token = &tokens[0];
|
2022-03-17 06:17:28 -07:00
|
|
|
let mut serum_market_cookies = vec![];
|
2022-03-30 03:24:07 -07:00
|
|
|
for token in tokens[1..].iter() {
|
|
|
|
serum_market_cookies.push((
|
|
|
|
token,
|
|
|
|
context
|
|
|
|
.serum
|
|
|
|
.list_spot_market(&token.mint, "e_token.mint)
|
|
|
|
.await,
|
|
|
|
));
|
2022-03-17 06:17:28 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
let mut serum_markets = vec![];
|
2022-03-30 03:24:07 -07:00
|
|
|
for (base_token, spot) in serum_market_cookies {
|
2022-03-17 06:17:28 -07:00
|
|
|
serum_markets.push(
|
|
|
|
send_tx(
|
|
|
|
solana,
|
2022-03-18 05:42:20 -07:00
|
|
|
Serum3RegisterMarketInstruction {
|
2022-03-17 06:17:28 -07:00
|
|
|
group,
|
|
|
|
admin,
|
|
|
|
serum_program: context.serum.program_id,
|
2022-03-30 03:24:07 -07:00
|
|
|
serum_market_external: spot.market,
|
|
|
|
market_index: spot.coin_mint.index as u16,
|
|
|
|
base_bank: base_token.bank,
|
|
|
|
quote_bank: quote_token.bank,
|
2022-03-17 06:17:28 -07:00
|
|
|
payer,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap()
|
|
|
|
.serum_market,
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// TEST: Create open orders and trigger a Deposit to check health
|
|
|
|
//
|
|
|
|
for (i, &serum_market) in serum_markets.iter().enumerate() {
|
|
|
|
println!("adding market {}", i);
|
|
|
|
send_tx(
|
|
|
|
solana,
|
2022-03-18 05:42:20 -07:00
|
|
|
Serum3CreateOpenOrdersInstruction {
|
2022-03-17 06:17:28 -07:00
|
|
|
account,
|
|
|
|
serum_market,
|
|
|
|
owner,
|
|
|
|
payer,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
|
|
|
|
send_tx(
|
|
|
|
solana,
|
2022-06-09 09:27:31 -07:00
|
|
|
TokenDepositInstruction {
|
2022-03-17 06:17:28 -07:00
|
|
|
amount: 10,
|
|
|
|
account,
|
2022-09-30 04:21:01 -07:00
|
|
|
owner,
|
2022-05-25 02:27:58 -07:00
|
|
|
token_account: payer_mint_accounts[0],
|
2022-06-29 02:18:59 -07:00
|
|
|
token_authority: payer.clone(),
|
|
|
|
bank_index: 0,
|
2022-03-17 06:17:28 -07:00
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: actual explicit CU comparisons.
|
2022-11-21 06:01:10 -08:00
|
|
|
// On 2022-11-18 the final deposit costs 62920 CU and each new market increases it by roughly 4820 CU
|
2022-05-25 02:27:58 -07:00
|
|
|
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
|
|
|
// Try to reach compute limits in health checks by having many perp markets in an account
|
|
|
|
#[tokio::test]
|
|
|
|
async fn test_health_compute_perp() -> Result<(), TransportError> {
|
|
|
|
let context = TestContext::new().await;
|
|
|
|
let solana = &context.solana.clone();
|
|
|
|
|
2022-09-07 03:39:21 -07:00
|
|
|
let admin = TestKeypair::new();
|
|
|
|
let owner = context.users[0].key;
|
|
|
|
let payer = context.users[1].key;
|
2022-05-25 02:27:58 -07:00
|
|
|
let mints = &context.mints[0..8];
|
|
|
|
let payer_mint_accounts = &context.users[1].token_accounts[0..mints.len()];
|
|
|
|
|
|
|
|
//
|
|
|
|
// SETUP: Create a group and an account
|
|
|
|
//
|
|
|
|
|
2022-07-04 04:13:11 -07:00
|
|
|
let mango_setup::GroupWithTokens { group, tokens, .. } = mango_setup::GroupWithTokensConfig {
|
2022-05-25 02:27:58 -07:00
|
|
|
admin,
|
|
|
|
payer,
|
2022-09-12 06:25:50 -07:00
|
|
|
mints: mints.to_vec(),
|
|
|
|
..GroupWithTokensConfig::default()
|
2022-05-25 02:27:58 -07:00
|
|
|
}
|
|
|
|
.create(solana)
|
|
|
|
.await;
|
|
|
|
|
2022-08-26 06:59:47 -07:00
|
|
|
let account = create_funded_account(
|
|
|
|
&solana,
|
|
|
|
group,
|
|
|
|
owner,
|
|
|
|
0,
|
|
|
|
&context.users[1],
|
|
|
|
&mints[..1],
|
|
|
|
1000,
|
|
|
|
0,
|
2022-05-25 02:27:58 -07:00
|
|
|
)
|
2022-08-26 06:59:47 -07:00
|
|
|
.await;
|
2022-05-25 02:27:58 -07:00
|
|
|
|
|
|
|
//
|
|
|
|
// SETUP: Create perp markets
|
|
|
|
//
|
|
|
|
let mut perp_markets = vec![];
|
|
|
|
for (perp_market_index, token) in tokens[1..].iter().enumerate() {
|
2022-11-08 06:27:56 -08:00
|
|
|
let mango_v4::accounts::PerpCreateMarket { perp_market, .. } = send_tx(
|
2022-05-25 02:27:58 -07:00
|
|
|
solana,
|
|
|
|
PerpCreateMarketInstruction {
|
|
|
|
group,
|
|
|
|
admin,
|
|
|
|
payer,
|
|
|
|
perp_market_index: perp_market_index as PerpMarketIndex,
|
|
|
|
quote_lot_size: 10,
|
|
|
|
base_lot_size: 100,
|
|
|
|
maint_asset_weight: 0.975,
|
|
|
|
init_asset_weight: 0.95,
|
|
|
|
maint_liab_weight: 1.025,
|
|
|
|
init_liab_weight: 1.05,
|
|
|
|
liquidation_fee: 0.012,
|
|
|
|
maker_fee: 0.0002,
|
|
|
|
taker_fee: 0.000,
|
2022-09-09 01:03:49 -07:00
|
|
|
..PerpCreateMarketInstruction::with_new_book_and_queue(&solana, &token).await
|
2022-05-25 02:27:58 -07:00
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
|
2022-11-08 06:27:56 -08:00
|
|
|
perp_markets.push(perp_market);
|
2022-05-25 02:27:58 -07:00
|
|
|
}
|
|
|
|
|
2022-05-25 10:29:53 -07:00
|
|
|
let price_lots = {
|
2022-11-08 06:27:56 -08:00
|
|
|
let perp_market = solana.get_account::<PerpMarket>(perp_markets[0]).await;
|
2022-05-25 10:29:53 -07:00
|
|
|
perp_market.native_price_to_lot(I80F48::from(1))
|
|
|
|
};
|
|
|
|
|
2022-05-25 02:27:58 -07:00
|
|
|
//
|
|
|
|
// TEST: Create a perp order for each market
|
|
|
|
//
|
2022-11-08 06:27:56 -08:00
|
|
|
for (i, &perp_market) in perp_markets.iter().enumerate() {
|
2022-05-25 02:27:58 -07:00
|
|
|
println!("adding market {}", i);
|
|
|
|
send_tx(
|
|
|
|
solana,
|
|
|
|
PerpPlaceOrderInstruction {
|
|
|
|
account,
|
|
|
|
perp_market,
|
|
|
|
owner,
|
|
|
|
side: Side::Bid,
|
2022-05-25 10:29:53 -07:00
|
|
|
price_lots,
|
2022-05-25 02:27:58 -07:00
|
|
|
max_base_lots: 1,
|
|
|
|
max_quote_lots: i64::MAX,
|
|
|
|
client_order_id: 0,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
|
|
|
|
send_tx(
|
|
|
|
solana,
|
2022-06-09 09:27:31 -07:00
|
|
|
TokenDepositInstruction {
|
2022-05-25 02:27:58 -07:00
|
|
|
amount: 10,
|
|
|
|
account,
|
2022-09-30 04:21:01 -07:00
|
|
|
owner,
|
2022-05-25 02:27:58 -07:00
|
|
|
token_account: payer_mint_accounts[0],
|
2022-06-29 02:18:59 -07:00
|
|
|
token_authority: payer.clone(),
|
|
|
|
bank_index: 0,
|
2022-05-25 02:27:58 -07:00
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: actual explicit CU comparisons.
|
2022-11-21 06:01:10 -08:00
|
|
|
// On 2022-11-18 the final deposit costs 50502 CU and each new market increases it by roughly 3037 CU
|
2022-03-17 06:17:28 -07:00
|
|
|
|
|
|
|
Ok(())
|
|
|
|
}
|