2022-03-11 03:59:42 -08:00
|
|
|
#![cfg(feature = "test-bpf")]
|
|
|
|
|
|
|
|
use solana_program::pubkey::Pubkey;
|
|
|
|
use solana_program_test::*;
|
|
|
|
use solana_sdk::{signature::Keypair, transport::TransportError};
|
|
|
|
|
2022-03-11 08:49:40 -08:00
|
|
|
use mango_v4::state::*;
|
2022-03-11 03:59:42 -08:00
|
|
|
use program_test::*;
|
|
|
|
|
|
|
|
mod program_test;
|
|
|
|
|
|
|
|
#[tokio::test]
|
|
|
|
async fn test_serum() -> Result<(), TransportError> {
|
|
|
|
let context = TestContext::new(Option::None, Option::None, Option::None, Option::None).await;
|
|
|
|
let solana = &context.solana.clone();
|
|
|
|
|
|
|
|
let admin = &Keypair::new();
|
|
|
|
let owner = &context.users[0].key;
|
|
|
|
let payer = &context.users[1].key;
|
|
|
|
let mint0 = &context.mints[0];
|
|
|
|
let mint1 = &context.mints[1];
|
2022-03-14 09:21:16 -07:00
|
|
|
let payer_mint_accounts = &context.users[1].token_accounts[0..=2];
|
2022-03-11 06:28:27 -08:00
|
|
|
|
|
|
|
//
|
|
|
|
// SETUP: Create serum market
|
|
|
|
//
|
|
|
|
let serum_market_cookie = context.serum.list_spot_market(mint0, mint1).await;
|
2022-03-11 03:59:42 -08:00
|
|
|
|
|
|
|
//
|
2022-03-11 05:26:42 -08:00
|
|
|
// SETUP: Create a group and an account
|
2022-03-11 03:59:42 -08:00
|
|
|
//
|
|
|
|
|
|
|
|
let group = send_tx(solana, CreateGroupInstruction { admin, payer })
|
|
|
|
.await
|
|
|
|
.unwrap()
|
|
|
|
.group;
|
|
|
|
|
2022-03-11 05:26:42 -08:00
|
|
|
let account = send_tx(
|
|
|
|
solana,
|
|
|
|
CreateAccountInstruction {
|
|
|
|
account_num: 0,
|
|
|
|
group,
|
|
|
|
owner,
|
|
|
|
payer,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap()
|
|
|
|
.account;
|
|
|
|
|
2022-03-11 03:59:42 -08:00
|
|
|
//
|
|
|
|
// SETUP: Register mints (and make oracles for them)
|
|
|
|
//
|
|
|
|
|
2022-03-14 05:19:50 -07:00
|
|
|
let register_mint = |index: TokenIndex, mint: MintCookie, address_lookup_table: Pubkey| async move {
|
2022-03-11 03:59:42 -08:00
|
|
|
let create_stub_oracle_accounts = send_tx(
|
|
|
|
solana,
|
|
|
|
CreateStubOracle {
|
|
|
|
mint: mint.pubkey,
|
|
|
|
payer,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
let oracle = create_stub_oracle_accounts.oracle;
|
|
|
|
send_tx(
|
|
|
|
solana,
|
|
|
|
SetStubOracle {
|
|
|
|
mint: mint.pubkey,
|
|
|
|
payer,
|
|
|
|
price: "1.0",
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
let register_token_accounts = send_tx(
|
|
|
|
solana,
|
|
|
|
RegisterTokenInstruction {
|
2022-03-14 05:19:50 -07:00
|
|
|
token_index: index,
|
2022-03-11 03:59:42 -08:00
|
|
|
decimals: mint.decimals,
|
|
|
|
maint_asset_weight: 0.9,
|
|
|
|
init_asset_weight: 0.8,
|
|
|
|
maint_liab_weight: 1.1,
|
|
|
|
init_liab_weight: 1.2,
|
|
|
|
group,
|
|
|
|
admin,
|
|
|
|
mint: mint.pubkey,
|
|
|
|
address_lookup_table,
|
|
|
|
payer,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
let bank = register_token_accounts.bank;
|
|
|
|
|
|
|
|
(oracle, bank)
|
|
|
|
};
|
|
|
|
|
|
|
|
let address_lookup_table = solana.create_address_lookup_table(admin, payer).await;
|
2022-03-14 05:19:50 -07:00
|
|
|
let base_token_index = 0;
|
2022-03-15 06:44:47 -07:00
|
|
|
let (_oracle0, bank0) =
|
2022-03-14 05:19:50 -07:00
|
|
|
register_mint(base_token_index, mint0.clone(), address_lookup_table).await;
|
|
|
|
let quote_token_index = 1;
|
2022-03-15 06:44:47 -07:00
|
|
|
let (_oracle1, bank1) =
|
2022-03-14 05:19:50 -07:00
|
|
|
register_mint(quote_token_index, mint1.clone(), address_lookup_table).await;
|
2022-03-11 03:59:42 -08:00
|
|
|
|
2022-03-14 09:21:16 -07:00
|
|
|
//
|
|
|
|
// SETUP: Deposit user funds
|
|
|
|
//
|
|
|
|
{
|
2022-03-15 06:44:47 -07:00
|
|
|
let deposit_amount = 1000;
|
2022-03-14 09:21:16 -07:00
|
|
|
|
|
|
|
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();
|
|
|
|
}
|
|
|
|
|
2022-03-11 03:59:42 -08:00
|
|
|
//
|
|
|
|
// TEST: Register a serum market
|
|
|
|
//
|
2022-03-11 05:26:42 -08:00
|
|
|
let serum_market = send_tx(
|
2022-03-11 03:59:42 -08:00
|
|
|
solana,
|
|
|
|
RegisterSerumMarketInstruction {
|
|
|
|
group,
|
|
|
|
admin,
|
2022-03-11 06:28:27 -08:00
|
|
|
serum_program: context.serum.program_id,
|
|
|
|
serum_market_external: serum_market_cookie.market,
|
2022-03-14 05:47:34 -07:00
|
|
|
market_index: 0,
|
2022-03-14 05:19:50 -07:00
|
|
|
base_token_index,
|
|
|
|
quote_token_index,
|
2022-03-11 03:59:42 -08:00
|
|
|
payer,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
2022-03-11 05:26:42 -08:00
|
|
|
.unwrap()
|
|
|
|
.serum_market;
|
|
|
|
|
|
|
|
//
|
|
|
|
// TEST: Create an open orders account
|
|
|
|
//
|
2022-03-11 08:49:40 -08:00
|
|
|
let open_orders = send_tx(
|
2022-03-11 05:26:42 -08:00
|
|
|
solana,
|
|
|
|
CreateSerumOpenOrdersInstruction {
|
|
|
|
account,
|
|
|
|
serum_market,
|
|
|
|
owner,
|
|
|
|
payer,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
2022-03-11 08:49:40 -08:00
|
|
|
.unwrap()
|
|
|
|
.open_orders;
|
|
|
|
|
|
|
|
let account_data: MangoAccount = solana.get_account(account).await;
|
|
|
|
assert_eq!(
|
|
|
|
account_data
|
2022-03-15 07:08:53 -07:00
|
|
|
.serum_account_map
|
2022-03-11 08:49:40 -08:00
|
|
|
.iter_active()
|
|
|
|
.map(|v| (v.open_orders, v.market_index))
|
|
|
|
.collect::<Vec<_>>(),
|
|
|
|
[(open_orders, 0)]
|
|
|
|
);
|
2022-03-11 03:59:42 -08:00
|
|
|
|
2022-03-14 09:21:16 -07:00
|
|
|
//
|
|
|
|
// TEST: Place an order
|
|
|
|
//
|
|
|
|
send_tx(
|
|
|
|
solana,
|
|
|
|
PlaceSerumOrderInstruction {
|
2022-03-15 06:44:47 -07:00
|
|
|
side: 0, // TODO: Bid
|
|
|
|
limit_price: 10, // in quote_lot (10) per base lot (100)
|
|
|
|
max_base_qty: 1, // in base lot (100)
|
|
|
|
max_native_quote_qty_including_fees: 100,
|
2022-03-14 09:21:16 -07:00
|
|
|
self_trade_behavior: 0,
|
2022-03-15 06:44:47 -07:00
|
|
|
order_type: 0, // TODO: Limit
|
2022-03-14 09:21:16 -07:00
|
|
|
client_order_id: 0,
|
|
|
|
limit: 10,
|
|
|
|
account,
|
|
|
|
owner,
|
|
|
|
serum_market,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
|
2022-03-15 06:44:47 -07:00
|
|
|
let native0 = account_position(solana, account, bank0).await;
|
|
|
|
let native1 = account_position(solana, account, bank1).await;
|
|
|
|
assert_eq!(native0, 1000);
|
|
|
|
assert_eq!(native1, 900);
|
|
|
|
|
2022-03-11 03:59:42 -08:00
|
|
|
Ok(())
|
|
|
|
}
|