2022-03-18 02:08:53 -07:00
|
|
|
#![cfg(feature = "test-bpf")]
|
|
|
|
|
2022-03-21 12:29:28 -07:00
|
|
|
use mango_v4::state::BookSide;
|
2022-03-18 02:08:53 -07:00
|
|
|
use solana_program_test::*;
|
|
|
|
use solana_sdk::{signature::Keypair, transport::TransportError};
|
|
|
|
|
|
|
|
use program_test::*;
|
|
|
|
|
|
|
|
mod program_test;
|
|
|
|
|
|
|
|
#[tokio::test]
|
|
|
|
async fn test_perp() -> Result<(), TransportError> {
|
2022-03-21 02:45:55 -07:00
|
|
|
let context = TestContext::new().await;
|
2022-03-18 02:08:53 -07:00
|
|
|
let solana = &context.solana.clone();
|
|
|
|
|
|
|
|
let admin = &Keypair::new();
|
|
|
|
let owner = &context.users[0].key;
|
|
|
|
let payer = &context.users[1].key;
|
2022-03-20 23:49:51 -07:00
|
|
|
let mints = &context.mints[0..2];
|
2022-03-18 02:08:53 -07:00
|
|
|
let payer_mint_accounts = &context.users[1].token_accounts[0..=2];
|
|
|
|
|
|
|
|
//
|
|
|
|
// SETUP: Create a group and an account
|
|
|
|
//
|
|
|
|
|
2022-03-20 23:49:51 -07:00
|
|
|
let mango_setup::GroupWithTokens { group, tokens } = mango_setup::GroupWithTokensConfig {
|
|
|
|
admin,
|
|
|
|
payer,
|
|
|
|
mints,
|
|
|
|
}
|
|
|
|
.create(solana)
|
|
|
|
.await;
|
2022-03-18 02:08:53 -07:00
|
|
|
|
|
|
|
let account = send_tx(
|
|
|
|
solana,
|
|
|
|
CreateAccountInstruction {
|
|
|
|
account_num: 0,
|
|
|
|
group,
|
|
|
|
owner,
|
|
|
|
payer,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap()
|
|
|
|
.account;
|
|
|
|
|
|
|
|
//
|
|
|
|
// SETUP: Deposit user funds
|
|
|
|
//
|
|
|
|
{
|
|
|
|
let deposit_amount = 1000;
|
|
|
|
|
|
|
|
send_tx(
|
|
|
|
solana,
|
|
|
|
DepositInstruction {
|
|
|
|
amount: deposit_amount,
|
|
|
|
account,
|
|
|
|
token_account: payer_mint_accounts[0],
|
|
|
|
token_authority: payer,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
|
|
|
|
send_tx(
|
|
|
|
solana,
|
|
|
|
DepositInstruction {
|
|
|
|
amount: deposit_amount,
|
|
|
|
account,
|
|
|
|
token_account: payer_mint_accounts[1],
|
|
|
|
token_authority: payer,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// TEST: Create a perp market
|
|
|
|
//
|
2022-03-21 12:29:28 -07:00
|
|
|
let mango_v4::accounts::CreatePerpMarket {
|
|
|
|
perp_market,
|
|
|
|
asks,
|
|
|
|
bids,
|
|
|
|
..
|
|
|
|
} = send_tx(
|
2022-03-18 02:08:53 -07:00
|
|
|
solana,
|
|
|
|
CreatePerpMarketInstruction {
|
|
|
|
group,
|
2022-03-21 12:29:28 -07:00
|
|
|
oracle: tokens[0].oracle,
|
|
|
|
asks: context
|
|
|
|
.solana
|
|
|
|
.create_account::<BookSide>(&mango_v4::id())
|
|
|
|
.await,
|
|
|
|
bids: context
|
|
|
|
.solana
|
|
|
|
.create_account::<BookSide>(&mango_v4::id())
|
|
|
|
.await,
|
2022-03-18 02:08:53 -07:00
|
|
|
admin,
|
|
|
|
payer,
|
2022-03-18 23:28:37 -07:00
|
|
|
perp_market_index: 0,
|
2022-03-20 23:49:51 -07:00
|
|
|
base_token_index: tokens[0].index,
|
|
|
|
quote_token_index: tokens[1].index,
|
2022-03-18 02:08:53 -07:00
|
|
|
// e.g. BTC mango-v3 mainnet.1
|
|
|
|
quote_lot_size: 10,
|
|
|
|
base_lot_size: 100,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
2022-03-21 12:29:28 -07:00
|
|
|
.unwrap();
|
|
|
|
|
|
|
|
send_tx(
|
|
|
|
solana,
|
|
|
|
PlacePerpOrderInstruction {
|
|
|
|
group,
|
|
|
|
account,
|
|
|
|
perp_market,
|
|
|
|
asks,
|
|
|
|
bids,
|
|
|
|
oracle: tokens[0].oracle,
|
|
|
|
owner,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
2022-03-18 02:08:53 -07:00
|
|
|
|
|
|
|
Ok(())
|
|
|
|
}
|