d0zingcat/ore
1
0
Fork 0
mirror of https://github.com/d0zingcat/ore.git synced 2026-05-14 15:10:13 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
d718360cafbd9d931b16bdbd795efa7488eb53f6
ore/tests/test_reset.rs
Hardhat Chad 3a109e7a11 groundwork for pick your difficulty
2024-04-16 03:26:40 +00:00

441 lines
17 KiB
XML
Raw Blame History

// use std::str::FromStr;
// use ore::{
// instruction::OreInstruction,
// state::{Bus, Treasury},
// utils::{AccountDeserialize, Discriminator},
// BUS, BUS_ADDRESSES, BUS_COUNT, BUS_EPOCH_REWARDS, INITIAL_DIFFICULTY, INITIAL_REWARD_RATE,
// MAX_EPOCH_REWARDS, MINT_ADDRESS, START_AT, TOKEN_DECIMALS, TREASURY, TREASURY_ADDRESS,
// };
// use rand::seq::SliceRandom;
// use solana_program::{
// clock::Clock,
// epoch_schedule::DEFAULT_SLOTS_PER_EPOCH,
// hash::Hash,
// instruction::{AccountMeta, Instruction},
// native_token::LAMPORTS_PER_SOL,
// program_option::COption,
// program_pack::Pack,
// pubkey::Pubkey,
// system_program, sysvar,
// };
// use solana_program_test::{processor, BanksClient, ProgramTest};
// use solana_sdk::{
// account::Account,
// signature::{Keypair, Signer},
// transaction::Transaction,
// };
// use spl_token::state::{AccountState, Mint};
// #[tokio::test]
// async fn test_reset() {
// // Setup
// let (mut banks, payer, _, blockhash) = setup_program_test_env(ClockState::Normal).await;
// // Pdas
// let bus_pdas = vec![
// Pubkey::find_program_address(&[BUS, &[0]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[1]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[2]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[3]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[4]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[5]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[6]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[7]], &ore::id()),
// ];
// // let mint_pda = Pubkey::find_program_address(&[MINT], &ore::id());
// let treasury_tokens_address = spl_associated_token_account::get_associated_token_address(
// &TREASURY_ADDRESS,
// &MINT_ADDRESS,
// );
// // Submit tx
// let ix = ore::instruction::reset(payer.pubkey());
// let tx = Transaction::new_signed_with_payer(&[ix], Some(&payer.pubkey()), &[&payer], blockhash);
// let res = banks.process_transaction(tx).await;
// assert!(res.is_ok());
// // Test bus state
// for i in 0..BUS_COUNT {
// let bus_account = banks.get_account(bus_pdas[i].0).await.unwrap().unwrap();
// assert_eq!(bus_account.owner, ore::id());
// let bus = Bus::try_from_bytes(&bus_account.data).unwrap();
// assert_eq!(bus.id as u8, i as u8);
// assert_eq!(bus.rewards, BUS_EPOCH_REWARDS);
// }
// // Test treasury state
// let treasury_account = banks.get_account(TREASURY_ADDRESS).await.unwrap().unwrap();
// assert_eq!(treasury_account.owner, ore::id());
// let treasury = Treasury::try_from_bytes(&treasury_account.data).unwrap();
// assert_eq!(
// treasury.admin,
// Pubkey::from_str("AeNqnoLwFanMd3ig9WoMxQZVwQHtCtqKMMBsT1sTrvz6").unwrap()
// );
// assert_eq!(treasury.difficulty, INITIAL_DIFFICULTY.into());
// assert_eq!(treasury.last_reset_at, START_AT + 1);
// assert_eq!(treasury.reward_rate, INITIAL_REWARD_RATE.saturating_div(2));
// assert_eq!(treasury.total_claimed_rewards as u8, 0);
// // Test mint state
// let mint_account = banks.get_account(MINT_ADDRESS).await.unwrap().unwrap();
// assert_eq!(mint_account.owner, spl_token::id());
// let mint = Mint::unpack(&mint_account.data).unwrap();
// assert_eq!(mint.mint_authority, COption::Some(TREASURY_ADDRESS));
// assert_eq!(mint.supply, MAX_EPOCH_REWARDS);
// assert_eq!(mint.decimals, ore::TOKEN_DECIMALS);
// assert_eq!(mint.is_initialized, true);
// assert_eq!(mint.freeze_authority, COption::None);
// // Test treasury token state
// let treasury_tokens_account = banks
// .get_account(treasury_tokens_address)
// .await
// .unwrap()
// .unwrap();
// assert_eq!(treasury_tokens_account.owner, spl_token::id());
// let treasury_tokens = spl_token::state::Account::unpack(&treasury_tokens_account.data).unwrap();
// assert_eq!(treasury_tokens.mint, MINT_ADDRESS);
// assert_eq!(treasury_tokens.owner, TREASURY_ADDRESS);
// assert_eq!(treasury_tokens.amount, MAX_EPOCH_REWARDS);
// assert_eq!(treasury_tokens.delegate, COption::None);
// assert_eq!(treasury_tokens.state, AccountState::Initialized);
// assert_eq!(treasury_tokens.is_native, COption::None);
// assert_eq!(treasury_tokens.delegated_amount, 0);
// assert_eq!(treasury_tokens.close_authority, COption::None);
// }
// #[tokio::test]
// async fn test_reset_bad_key() {
// // Setup
// let (mut banks, payer, _, blockhash) = setup_program_test_env(ClockState::Normal).await;
// // Bad addresses
// let bad_pda = Pubkey::find_program_address(&[b"t"], &ore::id());
// for i in 1..13 {
// let mut ix = ore::instruction::reset(payer.pubkey());
// ix.accounts[i].pubkey = bad_pda.0;
// let tx =
// Transaction::new_signed_with_payer(&[ix], Some(&payer.pubkey()), &[&payer], blockhash);
// let res = banks.process_transaction(tx).await;
// assert!(res.is_err());
// }
// }
// #[tokio::test]
// async fn test_reset_busses_out_of_order_fail() {
// // Setup
// let (mut banks, payer, _, blockhash) = setup_program_test_env(ClockState::Normal).await;
// // Pdas
// let signer = payer.pubkey();
// let bus_pdas = vec![
// Pubkey::find_program_address(&[BUS, &[5]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[0]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[6]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[2]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[3]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[7]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[1]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[4]], &ore::id()),
// ];
// let treasury_tokens = spl_associated_token_account::get_associated_token_address(
// &TREASURY_ADDRESS,
// &MINT_ADDRESS,
// );
// // Submit tx
// let ix = Instruction {
// program_id: ore::id(),
// accounts: vec![
// AccountMeta::new(signer, true),
// AccountMeta::new(bus_pdas[0].0, false),
// AccountMeta::new(bus_pdas[1].0, false),
// AccountMeta::new(bus_pdas[2].0, false),
// AccountMeta::new(bus_pdas[3].0, false),
// AccountMeta::new(bus_pdas[4].0, false),
// AccountMeta::new(bus_pdas[5].0, false),
// AccountMeta::new(bus_pdas[6].0, false),
// AccountMeta::new(bus_pdas[7].0, false),
// AccountMeta::new(MINT_ADDRESS, false),
// AccountMeta::new(TREASURY_ADDRESS, false),
// AccountMeta::new(treasury_tokens, false),
// AccountMeta::new_readonly(spl_token::id(), false),
// ],
// data: OreInstruction::Reset.to_vec(),
// };
// let tx = Transaction::new_signed_with_payer(&[ix], Some(&payer.pubkey()), &[&payer], blockhash);
// let res = banks.process_transaction(tx).await;
// assert!(res.is_err());
// }
// #[tokio::test]
// async fn test_reset_race() {
// // Setup
// let (mut banks, payer, payer_alt, blockhash) = setup_program_test_env(ClockState::Normal).await;
// // Reset one passes
// let ix = ore::instruction::reset(payer.pubkey());
// let tx = Transaction::new_signed_with_payer(&[ix], Some(&payer.pubkey()), &[&payer], blockhash);
// let res = banks.process_transaction(tx).await;
// assert!(res.is_ok());
// // Reset two fails
// let ix = ore::instruction::reset(payer_alt.pubkey());
// let tx = Transaction::new_signed_with_payer(
// &[ix],
// Some(&payer_alt.pubkey()),
// &[&payer_alt],
// blockhash,
// );
// let res = banks.process_transaction(tx).await;
// assert!(res.is_err());
// }
// #[tokio::test]
// async fn test_reset_too_early() {
// // Setup
// let (mut banks, payer, _, blockhash) = setup_program_test_env(ClockState::TooEarly).await;
// // Reset one passes
// let ix = ore::instruction::reset(payer.pubkey());
// let tx = Transaction::new_signed_with_payer(&[ix], Some(&payer.pubkey()), &[&payer], blockhash);
// let res = banks.process_transaction(tx).await;
// assert!(res.is_err());
// }
// #[tokio::test]
// async fn test_reset_not_enough_keys() {
// // Setup
// let (mut banks, payer, _, blockhash) = setup_program_test_env(ClockState::Normal).await;
// // Reset with missing account
// let mut ix = ore::instruction::reset(payer.pubkey());
// ix.accounts.remove(1);
// let tx = Transaction::new_signed_with_payer(&[ix], Some(&payer.pubkey()), &[&payer], blockhash);
// let res = banks.process_transaction(tx).await;
// assert!(res.is_err());
// }
// #[tokio::test]
// async fn test_reset_busses_duplicate_fail() {
// // Setup
// let (mut banks, payer, _, blockhash) = setup_program_test_env(ClockState::Normal).await;
// // Pdas
// let signer = payer.pubkey();
// let bus_pda = Pubkey::find_program_address(&[BUS, &[0]], &ore::id());
// let treasury_tokens = spl_associated_token_account::get_associated_token_address(
// &TREASURY_ADDRESS,
// &MINT_ADDRESS,
// );
// // Submit tx
// let ix = Instruction {
// program_id: ore::id(),
// accounts: vec![
// AccountMeta::new(signer, true),
// AccountMeta::new(bus_pda.0, false),
// AccountMeta::new(bus_pda.0, false),
// AccountMeta::new(bus_pda.0, false),
// AccountMeta::new(bus_pda.0, false),
// AccountMeta::new(bus_pda.0, false),
// AccountMeta::new(bus_pda.0, false),
// AccountMeta::new(bus_pda.0, false),
// AccountMeta::new(bus_pda.0, false),
// AccountMeta::new(MINT_ADDRESS, false),
// AccountMeta::new(TREASURY_ADDRESS, false),
// AccountMeta::new(treasury_tokens, false),
// AccountMeta::new_readonly(spl_token::id(), false),
// ],
// data: OreInstruction::Reset.to_vec(),
// };
// let tx = Transaction::new_signed_with_payer(&[ix], Some(&payer.pubkey()), &[&payer], blockhash);
// let res = banks.process_transaction(tx).await;
// assert!(res.is_err());
// }
// #[tokio::test]
// async fn test_reset_shuffle_error() {
// // Setup
// const FUZZ: u64 = 100;
// let (mut banks, payer, _, blockhash) = setup_program_test_env(ClockState::Normal).await;
// // Pdas
// let signer = payer.pubkey();
// let bus_pdas = vec![
// Pubkey::find_program_address(&[BUS, &[5]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[0]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[6]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[2]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[3]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[7]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[1]], &ore::id()),
// Pubkey::find_program_address(&[BUS, &[4]], &ore::id()),
// ];
// let treasury_tokens = spl_associated_token_account::get_associated_token_address(
// &TREASURY_ADDRESS,
// &MINT_ADDRESS,
// );
// // Fuzz test shuffled accounts.
// // Note some shuffles may still be valid if signer and non-bus accounts are all in correct positions.
// let mut rng = rand::thread_rng();
// for _ in 0..FUZZ {
// let mut accounts = vec![
// AccountMeta::new(signer, true),
// AccountMeta::new(bus_pdas[0].0, false),
// AccountMeta::new(bus_pdas[1].0, false),
// AccountMeta::new(bus_pdas[2].0, false),
// AccountMeta::new(bus_pdas[3].0, false),
// AccountMeta::new(bus_pdas[4].0, false),
// AccountMeta::new(bus_pdas[5].0, false),
// AccountMeta::new(bus_pdas[6].0, false),
// AccountMeta::new(bus_pdas[7].0, false),
// AccountMeta::new(MINT_ADDRESS, false),
// AccountMeta::new(TREASURY_ADDRESS, false),
// AccountMeta::new(treasury_tokens, false),
// AccountMeta::new_readonly(spl_token::id(), false),
// ];
// accounts.shuffle(&mut rng);
// let ix = Instruction {
// program_id: ore::id(),
// accounts,
// data: OreInstruction::Reset.to_vec(),
// };
// let tx =
// Transaction::new_signed_with_payer(&[ix], Some(&payer.pubkey()), &[&payer], blockhash);
// let res = banks.process_transaction(tx).await;
// assert!(res.is_err());
// }
// }
// enum ClockState {
// Normal,
// TooEarly,
// }
// async fn setup_program_test_env(clock_state: ClockState) -> (BanksClient, Keypair, Keypair, Hash) {
// let mut program_test = ProgramTest::new("ore", ore::ID, processor!(ore::process_instruction));
// program_test.prefer_bpf(true);
// // Busses
// for i in 0..BUS_COUNT {
// program_test.add_account_with_base64_data(
// BUS_ADDRESSES[i],
// 1057920,
// ore::id(),
// bs64::encode(
// &[
// &(Bus::discriminator() as u64).to_le_bytes(),
// Bus {
// id: i as u64,
// rewards: 0,
// }
// .to_bytes(),
// ]
// .concat(),
// )
// .as_str(),
// );
// }
// // Treasury
// let admin_address = Pubkey::from_str("AeNqnoLwFanMd3ig9WoMxQZVwQHtCtqKMMBsT1sTrvz6").unwrap();
// let treasury_pda = Pubkey::find_program_address(&[TREASURY], &ore::id());
// program_test.add_account_with_base64_data(
// treasury_pda.0,
// 1614720,
// ore::id(),
// bs64::encode(
// &[
// &(Treasury::discriminator() as u64).to_le_bytes(),
// Treasury {
// bump: treasury_pda.1 as u64,
// admin: admin_address,
// difficulty: INITIAL_DIFFICULTY.into(),
// last_reset_at: 0,
// reward_rate: INITIAL_REWARD_RATE,
// total_claimed_rewards: 0,
// }
// .to_bytes(),
// ]
// .concat(),
// )
// .as_str(),
// );
// // Mint
// let mut mint_src: [u8; Mint::LEN] = [0; Mint::LEN];
// Mint {
// mint_authority: COption::Some(TREASURY_ADDRESS),
// supply: 0,
// decimals: TOKEN_DECIMALS,
// is_initialized: true,
// freeze_authority: COption::None,
// }
// .pack_into_slice(&mut mint_src);
// program_test.add_account_with_base64_data(
// MINT_ADDRESS,
// 1461600,
// spl_token::id(),
// bs64::encode(&mint_src).as_str(),
// );
// // Treasury tokens
// let tokens_address = spl_associated_token_account::get_associated_token_address(
// &TREASURY_ADDRESS,
// &MINT_ADDRESS,
// );
// let mut tokens_src: [u8; spl_token::state::Account::LEN] = [0; spl_token::state::Account::LEN];
// spl_token::state::Account {
// mint: MINT_ADDRESS,
// owner: TREASURY_ADDRESS,
// amount: 0,
// delegate: COption::None,
// state: AccountState::Initialized,
// is_native: COption::None,
// delegated_amount: 0,
// close_authority: COption::None,
// }
// .pack_into_slice(&mut tokens_src);
// program_test.add_account_with_base64_data(
// tokens_address,
// 2039280,
// spl_token::id(),
// bs64::encode(&tokens_src).as_str(),
// );
// // Set sysvar
// let ts = match clock_state {
// ClockState::Normal => START_AT + 1,
// ClockState::TooEarly => START_AT - 1,
// };
// program_test.add_sysvar_account(
// sysvar::clock::id(),
// &Clock {
// slot: 0,
// epoch_start_timestamp: 0,
// epoch: 0,
// leader_schedule_epoch: DEFAULT_SLOTS_PER_EPOCH,
// unix_timestamp: ts,
// },
// );
// // Setup alt payer
// let payer_alt = Keypair::new();
// program_test.add_account(
// payer_alt.pubkey(),
// Account {
// lamports: LAMPORTS_PER_SOL,
// data: vec![],
// owner: system_program::id(),
// executable: false,
// rent_epoch: 0,
// },
// );
// let (banks, payer, blockhash) = program_test.start().await;
// (banks, payer, payer_alt, blockhash)
// }
Reference in New Issue View Git Blame Copy Permalink