2022-03-11 03:59:42 -08:00
|
|
|
#![cfg(feature = "test-bpf")]
|
|
|
|
|
|
|
|
use solana_program_test::*;
|
2022-06-09 09:27:31 -07:00
|
|
|
use solana_sdk::{signature::Keypair, signer::Signer, transport::TransportError};
|
2022-03-11 03:59:42 -08:00
|
|
|
|
2022-08-14 13:06:52 -07:00
|
|
|
use mango_v4::instructions::{Serum3OrderType, Serum3SelfTradeBehavior, Serum3Side};
|
2022-03-11 03:59:42 -08:00
|
|
|
use program_test::*;
|
|
|
|
|
|
|
|
mod program_test;
|
|
|
|
|
|
|
|
#[tokio::test]
|
2022-05-18 08:16:14 -07:00
|
|
|
async fn test_serum() -> Result<(), TransportError> {
|
2022-08-23 05:10:38 -07:00
|
|
|
let mut test_builder = TestContextBuilder::new();
|
|
|
|
test_builder.test().set_compute_max_units(95_000); // Serum3PlaceOrder needs 92.8k
|
|
|
|
let context = test_builder.start_default().await;
|
2022-03-11 03:59:42 -08: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-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
|
|
|
|
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
|
|
|
//
|
|
|
|
|
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,
|
|
|
|
mints,
|
|
|
|
}
|
|
|
|
.create(solana)
|
|
|
|
.await;
|
|
|
|
let base_token = &tokens[0];
|
|
|
|
let quote_token = &tokens[1];
|
2022-03-11 03:59:42 -08:00
|
|
|
|
2022-03-11 05:26:42 -08:00
|
|
|
let account = send_tx(
|
|
|
|
solana,
|
2022-07-06 05:51:15 -07:00
|
|
|
AccountCreateInstruction {
|
2022-03-11 05:26:42 -08: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-11 05:26:42 -08:00
|
|
|
group,
|
|
|
|
owner,
|
|
|
|
payer,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap()
|
|
|
|
.account;
|
|
|
|
|
2022-03-11 03:59:42 -08:00
|
|
|
//
|
2022-03-20 23:49:51 -07:00
|
|
|
// SETUP: Create serum market
|
2022-03-11 03:59:42 -08:00
|
|
|
//
|
2022-03-20 23:49:51 -07:00
|
|
|
let serum_market_cookie = context
|
|
|
|
.serum
|
|
|
|
.list_spot_market(&base_token.mint, "e_token.mint)
|
|
|
|
.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,
|
2022-06-09 09:27:31 -07:00
|
|
|
TokenDepositInstruction {
|
2022-03-14 09:21:16 -07:00
|
|
|
amount: deposit_amount,
|
|
|
|
account,
|
|
|
|
token_account: payer_mint_accounts[0],
|
2022-06-29 02:18:59 -07:00
|
|
|
token_authority: payer.clone(),
|
|
|
|
bank_index: 0,
|
2022-03-14 09:21:16 -07:00
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
|
|
|
|
send_tx(
|
|
|
|
solana,
|
2022-06-09 09:27:31 -07:00
|
|
|
TokenDepositInstruction {
|
2022-03-14 09:21:16 -07:00
|
|
|
amount: deposit_amount,
|
|
|
|
account,
|
|
|
|
token_account: payer_mint_accounts[1],
|
2022-06-29 02:18:59 -07:00
|
|
|
token_authority: payer.clone(),
|
|
|
|
bank_index: 0,
|
2022-03-14 09:21:16 -07:00
|
|
|
},
|
|
|
|
)
|
|
|
|
.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,
|
2022-03-18 05:42:20 -07:00
|
|
|
Serum3RegisterMarketInstruction {
|
2022-03-11 03:59:42 -08:00
|
|
|
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-30 03:24:07 -07:00
|
|
|
base_bank: base_token.bank,
|
|
|
|
quote_bank: quote_token.bank,
|
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,
|
2022-03-18 05:42:20 -07:00
|
|
|
Serum3CreateOpenOrdersInstruction {
|
2022-03-11 05:26:42 -08:00
|
|
|
account,
|
|
|
|
serum_market,
|
|
|
|
owner,
|
|
|
|
payer,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
2022-03-11 08:49:40 -08:00
|
|
|
.unwrap()
|
|
|
|
.open_orders;
|
|
|
|
|
2022-07-25 07:07:53 -07:00
|
|
|
let account_data = get_mango_account(solana, account).await;
|
2022-03-11 08:49:40 -08:00
|
|
|
assert_eq!(
|
|
|
|
account_data
|
2022-08-18 04:45:31 -07:00
|
|
|
.active_serum3_orders()
|
2022-03-11 08:49:40 -08:00
|
|
|
.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,
|
2022-03-18 05:42:20 -07:00
|
|
|
Serum3PlaceOrderInstruction {
|
2022-04-01 23:59:07 -07:00
|
|
|
side: Serum3Side::Bid,
|
2022-03-15 06:44:47 -07:00
|
|
|
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-04-01 23:59:07 -07:00
|
|
|
self_trade_behavior: Serum3SelfTradeBehavior::DecrementTake,
|
|
|
|
order_type: Serum3OrderType::Limit,
|
2022-03-14 09:21:16 -07:00
|
|
|
client_order_id: 0,
|
|
|
|
limit: 10,
|
|
|
|
account,
|
|
|
|
owner,
|
|
|
|
serum_market,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
|
2022-03-20 23:49:51 -07:00
|
|
|
let native0 = account_position(solana, account, base_token.bank).await;
|
|
|
|
let native1 = account_position(solana, account, quote_token.bank).await;
|
2022-03-15 06:44:47 -07:00
|
|
|
assert_eq!(native0, 1000);
|
|
|
|
assert_eq!(native1, 900);
|
|
|
|
|
2022-03-19 04:11:56 -07:00
|
|
|
// get the order id
|
|
|
|
let open_orders_bytes = solana.get_account_data(open_orders).await.unwrap();
|
|
|
|
let open_orders_data: &serum_dex::state::OpenOrders = bytemuck::from_bytes(
|
|
|
|
&open_orders_bytes[5..5 + std::mem::size_of::<serum_dex::state::OpenOrders>()],
|
|
|
|
);
|
|
|
|
let order_id = open_orders_data.orders[0];
|
|
|
|
assert!(order_id != 0);
|
|
|
|
|
|
|
|
//
|
|
|
|
// TEST: Cancel the order
|
|
|
|
//
|
|
|
|
send_tx(
|
|
|
|
solana,
|
|
|
|
Serum3CancelOrderInstruction {
|
2022-04-01 23:59:07 -07:00
|
|
|
side: Serum3Side::Bid,
|
2022-03-19 04:11:56 -07:00
|
|
|
order_id,
|
|
|
|
account,
|
|
|
|
owner,
|
|
|
|
serum_market,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
|
|
|
|
//
|
|
|
|
// TEST: Settle, moving the freed up funds back
|
|
|
|
//
|
2022-03-18 07:58:39 -07:00
|
|
|
send_tx(
|
|
|
|
solana,
|
|
|
|
Serum3SettleFundsInstruction {
|
|
|
|
account,
|
|
|
|
owner,
|
|
|
|
serum_market,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
|
2022-03-20 23:49:51 -07:00
|
|
|
let native0 = account_position(solana, account, base_token.bank).await;
|
|
|
|
let native1 = account_position(solana, account, quote_token.bank).await;
|
2022-03-19 04:11:56 -07:00
|
|
|
assert_eq!(native0, 1000);
|
|
|
|
assert_eq!(native1, 1000);
|
|
|
|
|
2022-06-09 09:27:31 -07:00
|
|
|
// close oo account
|
|
|
|
// TODO: custom program error: 0x2a TooManyOpenOrders https://github.com/project-serum/serum-dex/blob/master/dex/src/error.rs#L88
|
|
|
|
// send_tx(
|
|
|
|
// solana,
|
|
|
|
// Serum3CloseOpenOrdersInstruction {
|
|
|
|
// account,
|
|
|
|
// serum_market,
|
|
|
|
// owner,
|
|
|
|
// sol_destination: payer.pubkey(),
|
|
|
|
// },
|
|
|
|
// )
|
|
|
|
// .await
|
|
|
|
// .unwrap();
|
|
|
|
|
|
|
|
// deregister serum3 market
|
|
|
|
send_tx(
|
|
|
|
solana,
|
|
|
|
Serum3DeregisterMarketInstruction {
|
|
|
|
group,
|
|
|
|
admin,
|
|
|
|
serum_market_external: serum_market_cookie.market,
|
|
|
|
sol_destination: payer.pubkey(),
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
|
2022-03-11 03:59:42 -08:00
|
|
|
Ok(())
|
|
|
|
}
|