Files
smart-updown-token/program/programs/smart-updown/tests/test_smart_updown.rs
T

561 lines
19 KiB
Rust

use {
anchor_lang::{
solana_program::{
instruction::Instruction,
program_option::COption,
program_pack::Pack,
pubkey::Pubkey,
rent::Rent,
system_instruction,
},
AccountDeserialize, InstructionData, ToAccountMetas,
},
smart_updown::{constants::VAULT_SEED, state::{Bet, Global}},
solana_account::{Account as SolanaAccount},
solana_clock::Clock,
solana_keypair::Keypair,
solana_program_test::{BanksClient, ProgramTest, ProgramTestContext},
solana_signer::Signer,
solana_transaction::Transaction,
spl_associated_token_account,
spl_token,
spl_token::state::{Account as TokenData, AccountState, Mint},
};
const CLOCK_ADDR: &str = "SysvarC1ock11111111111111111111111111111111";
const SYSTEM_ADDR: &str = "11111111111111111111111111111111";
const MIN_AMOUNT: u64 = 1_000_000_000; // minimum bet, token units
const MULTIPLIER_BPS: u64 = 13_000; // 1.3x
const EXPIRY_SECONDS: i64 = 15;
const START_TIME: i64 = 1_700_000_000;
// SPL token (фантики) environment
const TOKEN_DECIMALS: u8 = 6;
const MINT_SIZE: u64 = 82; // classic SPL token mint account size
const TOKEN_ACCOUNT_SIZE: u64 = 165; // classic SPL token account size
const VAULT_TOKEN_SUPPLY: u64 = 100 * MIN_AMOUNT; // vault seed, covers many payouts
const BETTOR_TOKEN_SUPPLY: u64 = 10 * MIN_AMOUNT; // per-bettor token supply
/// Fixed фантики mint: mint + vault token accounts are pre-loaded into
/// genesis because the vault is a PDA and cannot sign its own create_account.
const MINT: Pubkey = Pubkey::new_from_array([3u8; 32]);
fn pid() -> Pubkey {
smart_updown::id()
}
fn global_pda() -> Pubkey {
Pubkey::find_program_address(&[b"global"], &pid()).0
}
fn vault_pda() -> Pubkey {
Pubkey::find_program_address(&[b"global", VAULT_SEED], &pid()).0
}
fn bet_pda(id: u64) -> Pubkey {
Pubkey::find_program_address(&[b"bet", &id.to_le_bytes()], &pid()).0
}
/// Boot the program-test context with the фантики mint and the program's
/// token vault (PDA([global, vault])) pre-loaded into genesis. The vault is a
/// PDA owned by itself, so it can never sign a top-level create_account; the
/// account data below matches what the token program would have created.
async fn setup() -> (ProgramTestContext, Keypair) {
let admin = Keypair::new();
let mint_data = {
let m = Mint {
mint_authority: COption::Some(admin.pubkey()),
supply: 0,
decimals: TOKEN_DECIMALS,
is_initialized: true,
freeze_authority: COption::None,
};
let mut buf = [0u8; MINT_SIZE as usize];
Pack::pack(m, &mut buf).expect("pack mint");
buf
};
let vault_data = {
let a = TokenData {
mint: MINT,
owner: vault_pda(),
amount: VAULT_TOKEN_SUPPLY,
delegate: COption::None,
state: AccountState::Initialized,
is_native: COption::None,
delegated_amount: 0,
close_authority: COption::None,
};
let mut buf = [0u8; TOKEN_ACCOUNT_SIZE as usize];
Pack::pack(a, &mut buf).expect("pack vault token account");
buf
};
let mut pt = ProgramTest::default();
pt.add_program("smart_updown", pid(), None);
let rent = Rent::default();
pt.add_account(
MINT,
SolanaAccount {
lamports: rent.minimum_balance(MINT_SIZE as usize),
data: mint_data.to_vec(),
owner: spl_token::ID,
executable: false,
rent_epoch: 0,
},
);
pt.add_account(
vault_pda(),
SolanaAccount {
lamports: rent.minimum_balance(TOKEN_ACCOUNT_SIZE as usize),
data: vault_data.to_vec(),
owner: spl_token::ID,
executable: false,
rent_epoch: 0,
},
);
// Fund the deploy keypair separately in the test; the validator in CI funds it.
let context = pt.start_with_context().await;
context.banks_client.process_transaction(Transaction::new_signed_with_payer(
&[system_instruction::transfer(
&context.payer.pubkey(),
&admin.pubkey(),
100_000_000_000,
)],
Some(&context.payer.pubkey()),
&[&context.payer],
context.last_blockhash,
)).await.unwrap();
(context, admin)
}
async fn set_clock(ctx: &mut ProgramTestContext, unix: i64) {
let mut clk: Clock = ctx.banks_client.get_sysvar().await.unwrap();
clk.unix_timestamp = unix;
ctx.set_sysvar(&clk);
}
async fn read_account_data(
bank: &BanksClient,
addr: &Pubkey,
) -> Vec<u8> {
bank.get_account(*addr).await.unwrap().expect("account exists").data
}
fn read_global_typed(data: &[u8]) -> Global {
Global::try_deserialize(&mut &data[..]).unwrap()
}
fn read_bet_typed(data: &[u8]) -> Bet {
Bet::try_deserialize(&mut &data[..]).unwrap()
}
fn make_initialize_ix(admin: Pubkey) -> Instruction {
Instruction::new_with_bytes(
pid(),
&smart_updown::instruction::Initialize {
min_amount: MIN_AMOUNT,
max_amount: 1_000_000_000_000,
multiplier_bps: MULTIPLIER_BPS,
expiry_seconds: EXPIRY_SECONDS,
}
.data(),
smart_updown::accounts::Initialize {
admin,
global: global_pda(),
system_program: SYSTEM_ADDR.parse().unwrap(),
}
.to_account_metas(None),
)
}
fn make_create_ix(
id: u64,
side: u64,
amount: u64,
entry_price: u64,
bettor: Pubkey,
bettor_ata: &Pubkey,
) -> Instruction {
Instruction::new_with_bytes(
pid(),
&smart_updown::instruction::CreateBet {
id,
side,
amount,
entry_price,
}
.data(),
smart_updown::accounts::CreateBet {
bettor,
global: global_pda(),
bet: bet_pda(id),
clock: CLOCK_ADDR.parse().unwrap(),
system_program: SYSTEM_ADDR.parse().unwrap(),
bettor_token: *bettor_ata,
vault_token: vault_pda(),
token_program: spl_token::ID,
}
.to_account_metas(None),
)
}
fn make_close_ix(
id: u64,
exit_price: u64,
relayer: Pubkey,
bettor: Pubkey,
bettor_ata: &Pubkey,
) -> Instruction {
Instruction::new_with_bytes(
pid(),
&smart_updown::instruction::CloseBet { id, exit_price }.data(),
smart_updown::accounts::CloseBet {
relayer,
bet: bet_pda(id),
global: global_pda(),
bettor,
clock: CLOCK_ADDR.parse().unwrap(),
vault_token: vault_pda(),
bettor_token: *bettor_ata,
vault_authority: vault_pda(),
token_program: spl_token::ID,
}
.to_account_metas(None),
)
}
async fn run_ix(
ctx: &mut ProgramTestContext,
signer: &Keypair,
ix: Instruction,
) -> Result<(), String> {
let tx = Transaction::new_signed_with_payer(
&[ix],
Some(&signer.pubkey()),
&[signer],
ctx.last_blockhash,
);
ctx.banks_client
.process_transaction(tx)
.await
.map(|_| ())
.map_err(|e| format!("{:?}", e))
}
async fn initialize(ctx: &mut ProgramTestContext, admin: &Keypair) {
run_ix(ctx, admin, make_initialize_ix(admin.pubkey()))
.await
.expect("initialize");
}
async fn airdrop_to(ctx: &mut ProgramTestContext, to: &Pubkey, lamports: u64) {
let tx = Transaction::new_signed_with_payer(
&[system_instruction::transfer(
&ctx.payer.pubkey(),
to,
lamports,
)],
Some(&ctx.payer.pubkey()),
&[&ctx.payer],
ctx.last_blockhash,
);
ctx.banks_client.process_transaction(tx).await.unwrap();
}
/// Create the token mint, the program's token vault (PDA([global, vault]),
/// owned by the global PDA), and seed the vault with tokens.
// The фантики env (mint + seeded vault) is pre-loaded into genesis by setup();
// a PDA vault can never sign its own top-level create_account, so this only
// returns the fixed mint the genesis accounts were built around.
async fn create_token_env(_ctx: &mut ProgramTestContext, _admin: &Keypair) -> Pubkey {
MINT
}
/// Mint tokens to a destination account (admin is the mint authority).
async fn mint_to(
ctx: &mut ProgramTestContext,
admin: &Keypair,
mint: &Pubkey,
dest: &Pubkey,
amount: u64,
) {
let ix = spl_token::instruction::mint_to(&spl_token::ID, mint, dest, &admin.pubkey(), &[], amount)
.expect("mint_to");
run_ix(ctx, admin, ix).await.expect("mint_to");
}
/// New bettor: lamports account + associated token account + token supply.
async fn create_bettor(
ctx: &mut ProgramTestContext,
admin: &Keypair,
mint: &Pubkey,
) -> (Keypair, Pubkey) {
let bettor = Keypair::new();
airdrop_to(ctx, &bettor.pubkey(), 100_000_000_000).await;
let ata = spl_associated_token_account::get_associated_token_address(&bettor.pubkey(), mint);
let ix = spl_associated_token_account::instruction::create_associated_token_account(
&admin.pubkey(),
&bettor.pubkey(),
mint,
&spl_token::ID,
);
run_ix(ctx, admin, ix).await.expect("create ata");
mint_to(ctx, admin, mint, &ata, BETTOR_TOKEN_SUPPLY).await;
(bettor, ata)
}
/// Read the `amount` field of a classic SPL token account (offset 64).
async fn read_token_balance(ctx: &mut ProgramTestContext, addr: &Pubkey) -> u64 {
let data = read_account_data(&ctx.banks_client, addr).await;
u64::from_le_bytes(data[64..72].try_into().expect("amount offset"))
}
/// close_bet pays out by moving tokens out of the vault; the vault PDA signs
/// that transfer CPI inside the program, so the transaction needs only the
/// relayer's signature.
async fn run_close_tx(
ctx: &mut ProgramTestContext,
signer: &Keypair,
ix: &Instruction,
) -> Result<(), String> {
// The vault PDA signs the payout CPI inside the program (via signer_seeds);
// the transaction itself needs only the relayer's signature.
let tx = Transaction::new_signed_with_payer(
std::slice::from_ref(ix),
Some(&signer.pubkey()),
&[signer],
ctx.last_blockhash,
);
ctx.banks_client
.process_transaction(tx)
.await
.map(|_| ())
.map_err(|e| format!("{:?}", e))
}
async fn create_up_bet(
ctx: &mut ProgramTestContext,
side: u64,
mint: &Pubkey,
admin: &Keypair,
) -> (Keypair, Pubkey, u64, u64) {
let (bettor, ata) = create_bettor(ctx, admin, mint).await;
set_clock(ctx, START_TIME).await;
let amount = MIN_AMOUNT;
let entry_price = 100_000_000u64;
let create_ix = make_create_ix(0, side, amount, entry_price, bettor.pubkey(), &ata);
run_ix(ctx, &bettor, create_ix).await.expect("create_bet");
(bettor, ata, amount, entry_price)
}
#[tokio::test]
async fn initialize_creates_global() {
let (mut ctx, admin) = setup().await;
initialize(&mut ctx, &admin).await;
let acc = read_account_data(&ctx.banks_client, &global_pda()).await;
let g = read_global_typed(&acc);
assert_eq!(g.admin, admin.pubkey());
assert_eq!(g.relayer, admin.pubkey());
assert_eq!(g.counter, 0);
assert_eq!(g.multiplier_bps, MULTIPLIER_BPS);
assert_eq!(g.expiry_seconds, EXPIRY_SECONDS);
assert_eq!(g.paused, 0);
}
#[tokio::test]
async fn create_bet_escrows_tokens() {
let (mut ctx, admin) = setup().await;
initialize(&mut ctx, &admin).await;
let mint = create_token_env(&mut ctx, &admin).await;
let (bettor, ata, amount, entry_price) = create_up_bet(&mut ctx, 0, &mint, &admin).await;
assert_eq!(
read_token_balance(&mut ctx, &vault_pda()).await,
VAULT_TOKEN_SUPPLY + amount
);
assert_eq!(
read_token_balance(&mut ctx, &ata).await,
BETTOR_TOKEN_SUPPLY - amount
);
let b = read_bet_typed(&read_account_data(&ctx.banks_client, &bet_pda(0)).await);
assert_eq!(b.bettor, bettor.pubkey());
assert_eq!(b.side, 0);
assert_eq!(b.amount, amount);
assert_eq!(b.entry_price, entry_price);
assert_eq!(b.bet_time, START_TIME);
assert_eq!(b.expire_time, START_TIME + EXPIRY_SECONDS);
assert_eq!(b.status, 0);
assert_eq!(read_global_typed(&read_account_data(&ctx.banks_client, &global_pda()).await).counter, 1);
}
#[tokio::test]
async fn close_bet_user_wins_up() {
let (mut ctx, admin) = setup().await;
initialize(&mut ctx, &admin).await;
let mint = create_token_env(&mut ctx, &admin).await;
let (bettor, ata, amount, entry_price) = create_up_bet(&mut ctx, 0, &mint, &admin).await;
set_clock(&mut ctx, START_TIME + EXPIRY_SECONDS + 1).await;
let exit_price = entry_price + 1_000;
let before = read_token_balance(&mut ctx, &ata).await;
let payout = amount * MULTIPLIER_BPS / 10_000;
run_close_tx(
&mut ctx,
&admin,
&make_close_ix(0, exit_price, admin.pubkey(), bettor.pubkey(), &ata),
)
.await
.expect("close_bet");
assert_eq!(read_token_balance(&mut ctx, &ata).await, before + payout);
assert_eq!(
read_token_balance(&mut ctx, &vault_pda()).await,
VAULT_TOKEN_SUPPLY + amount - payout
);
let b = read_bet_typed(&read_account_data(&ctx.banks_client, &bet_pda(0)).await);
assert_eq!(b.status, 1);
assert_eq!(b.exit_price, exit_price);
}
#[tokio::test]
async fn close_bet_house_wins_down() {
let (mut ctx, admin) = setup().await;
initialize(&mut ctx, &admin).await;
let mint = create_token_env(&mut ctx, &admin).await;
let (bettor, ata, _amount, entry_price) = create_up_bet(&mut ctx, 1, &mint, &admin).await;
set_clock(&mut ctx, START_TIME + EXPIRY_SECONDS + 1).await;
let exit_price = entry_price + 1_000; // side=DOWN but price rose -> lose
let before = read_token_balance(&mut ctx, &ata).await;
run_close_tx(
&mut ctx,
&admin,
&make_close_ix(0, exit_price, admin.pubkey(), bettor.pubkey(), &ata),
)
.await
.expect("close_bet");
assert_eq!(read_token_balance(&mut ctx, &ata).await, before);
assert_eq!(read_bet_typed(&read_account_data(&ctx.banks_client, &bet_pda(0)).await).status, 2);
}
#[tokio::test]
async fn close_bet_equal_price_stays_open() {
let (mut ctx, admin) = setup().await;
initialize(&mut ctx, &admin).await;
let mint = create_token_env(&mut ctx, &admin).await;
let (bettor, ata, amount, entry_price) = create_up_bet(&mut ctx, 0, &mint, &admin).await;
set_clock(&mut ctx, START_TIME + EXPIRY_SECONDS + 1).await;
let before = read_token_balance(&mut ctx, &ata).await;
let payout = amount * MULTIPLIER_BPS / 10_000;
let equal_ix = make_close_ix(0, entry_price, admin.pubkey(), bettor.pubkey(), &ata);
run_close_tx(&mut ctx, &admin, &equal_ix).await.expect("equal close");
assert_eq!(read_bet_typed(&read_account_data(&ctx.banks_client, &bet_pda(0)).await).status, 0);
assert_eq!(read_token_balance(&mut ctx, &ata).await, before);
let exit_price = entry_price + 1_000;
let second_ix = make_close_ix(0, exit_price, admin.pubkey(), bettor.pubkey(), &ata);
run_close_tx(&mut ctx, &admin, &second_ix).await.expect("second close");
assert_eq!(
read_token_balance(&mut ctx, &ata).await,
before + payout
);
assert_eq!(read_bet_typed(&read_account_data(&ctx.banks_client, &bet_pda(0)).await).status, 1);
}
#[tokio::test]
async fn close_bet_already_finalized() {
let (mut ctx, admin) = setup().await;
initialize(&mut ctx, &admin).await;
let mint = create_token_env(&mut ctx, &admin).await;
let (bettor, ata, _amount, entry_price) = create_up_bet(&mut ctx, 0, &mint, &admin).await;
set_clock(&mut ctx, START_TIME + EXPIRY_SECONDS + 1).await;
let win_price = entry_price + 500;
run_close_tx(
&mut ctx,
&admin,
&make_close_ix(0, win_price, admin.pubkey(), bettor.pubkey(), &ata),
)
.await
.expect("first close");
let res = run_close_tx(
&mut ctx,
&admin,
&make_close_ix(0, win_price + 1, admin.pubkey(), bettor.pubkey(), &ata),
)
.await;
assert!(res.is_err(), "expected close after finalization to be rejected");
let b = read_bet_typed(&read_account_data(&ctx.banks_client, &bet_pda(0)).await);
assert_eq!(b.status, 1);
assert_eq!(b.exit_price, win_price);
}
#[tokio::test]
async fn close_bet_before_expiry_rejected() {
let (mut ctx, admin) = setup().await;
initialize(&mut ctx, &admin).await;
let mint = create_token_env(&mut ctx, &admin).await;
let (bettor, ata, _amount, entry_price) = create_up_bet(&mut ctx, 0, &mint, &admin).await;
let close_ix = make_close_ix(0, entry_price + 500, admin.pubkey(), bettor.pubkey(), &ata);
let res = run_close_tx(&mut ctx, &admin, &close_ix).await;
assert!(res.is_err(), "expected premature close to be rejected");
assert_eq!(read_bet_typed(&read_account_data(&ctx.banks_client, &bet_pda(0)).await).status, 0);
}
#[tokio::test]
async fn close_bet_wrong_relayer_rejected() {
let (mut ctx, admin) = setup().await;
initialize(&mut ctx, &admin).await;
let mint = create_token_env(&mut ctx, &admin).await;
let (bettor, ata, _amount, entry_price) = create_up_bet(&mut ctx, 0, &mint, &admin).await;
let attacker = Keypair::new();
airdrop_to(&mut ctx, &attacker.pubkey(), 100_000_000_000).await;
set_clock(&mut ctx, START_TIME + EXPIRY_SECONDS + 1).await;
let close_ix = make_close_ix(0, entry_price + 500, attacker.pubkey(), bettor.pubkey(), &ata);
let res = run_close_tx(&mut ctx, &attacker, &close_ix).await;
assert!(res.is_err(), "expected wrong relayer to be rejected");
assert_eq!(read_bet_typed(&read_account_data(&ctx.banks_client, &bet_pda(0)).await).status, 0);
}
#[tokio::test]
async fn close_bet_replay_rejected() {
let (mut ctx, admin) = setup().await;
initialize(&mut ctx, &admin).await;
let mint = create_token_env(&mut ctx, &admin).await;
let (bettor, ata, _amount, entry_price) = create_up_bet(&mut ctx, 0, &mint, &admin).await;
set_clock(&mut ctx, START_TIME + EXPIRY_SECONDS + 1).await;
let close_ix = make_close_ix(0, entry_price + 500, admin.pubkey(), bettor.pubkey(), &ata);
run_close_tx(&mut ctx, &admin, &close_ix).await.expect("first close");
let before = read_token_balance(&mut ctx, &ata).await;
let res = run_close_tx(
&mut ctx,
&admin,
&make_close_ix(0, entry_price + 501, admin.pubkey(), bettor.pubkey(), &ata),
)
.await;
assert!(res.is_err(), "expected replay close to be rejected");
assert_eq!(read_token_balance(&mut ctx, &ata).await, before);
}
#[tokio::test]
async fn create_bet_rejects_out_of_range() {
let (mut ctx, admin) = setup().await;
initialize(&mut ctx, &admin).await;
let mint = create_token_env(&mut ctx, &admin).await;
let (bettor, ata) = create_bettor(&mut ctx, &admin, &mint).await;
set_clock(&mut ctx, START_TIME).await;
let vault_before = read_token_balance(&mut ctx, &vault_pda()).await;
let ix = make_create_ix(0, 0, 1, 100_000_000u64, bettor.pubkey(), &ata);
let res = run_ix(&mut ctx, &bettor, ix).await;
assert!(res.is_err(), "expected below-min bet to be rejected");
assert_eq!(read_token_balance(&mut ctx, &vault_pda()).await, vault_before);
}