|
| 1 | +extern crate std; |
| 2 | +use { |
| 3 | + alloc::vec, |
| 4 | + quasar_svm::{Account, Instruction, Pubkey, QuasarSvm}, |
| 5 | + spl_token_interface::state::{Account as TokenAccount, AccountState, Mint}, |
| 6 | + std::println, |
| 7 | +}; |
| 8 | + |
| 9 | +fn setup() -> QuasarSvm { |
| 10 | + let elf = std::fs::read("target/deploy/quasar_pda_mint_authority.so").unwrap(); |
| 11 | + QuasarSvm::new() |
| 12 | + .with_program(&crate::ID, &elf) |
| 13 | + .with_token_program() |
| 14 | +} |
| 15 | + |
| 16 | +fn signer(address: Pubkey) -> Account { |
| 17 | + quasar_svm::token::create_keyed_system_account(&address, 1_000_000_000) |
| 18 | +} |
| 19 | + |
| 20 | +fn empty(address: Pubkey) -> Account { |
| 21 | + Account { |
| 22 | + address, |
| 23 | + lamports: 0, |
| 24 | + data: vec![], |
| 25 | + owner: quasar_svm::system_program::ID, |
| 26 | + executable: false, |
| 27 | + } |
| 28 | +} |
| 29 | + |
| 30 | +fn mint_account(address: Pubkey, authority: Pubkey) -> Account { |
| 31 | + quasar_svm::token::create_keyed_mint_account( |
| 32 | + &address, |
| 33 | + &Mint { |
| 34 | + mint_authority: Some(authority).into(), |
| 35 | + supply: 0, |
| 36 | + decimals: 9, |
| 37 | + is_initialized: true, |
| 38 | + freeze_authority: None.into(), |
| 39 | + }, |
| 40 | + ) |
| 41 | +} |
| 42 | + |
| 43 | +fn token_account(address: Pubkey, mint: Pubkey, owner: Pubkey, amount: u64) -> Account { |
| 44 | + quasar_svm::token::create_keyed_token_account( |
| 45 | + &address, |
| 46 | + &TokenAccount { |
| 47 | + mint, |
| 48 | + owner, |
| 49 | + amount, |
| 50 | + state: AccountState::Initialized, |
| 51 | + ..TokenAccount::default() |
| 52 | + }, |
| 53 | + ) |
| 54 | +} |
| 55 | + |
| 56 | +/// Build create_mint instruction data. |
| 57 | +/// Wire format: [discriminator: u8 = 0] [decimals: u8] |
| 58 | +fn build_create_mint_data(decimals: u8) -> Vec<u8> { |
| 59 | + vec![0u8, decimals] |
| 60 | +} |
| 61 | + |
| 62 | +/// Build mint_tokens instruction data. |
| 63 | +/// Wire format: [discriminator: u8 = 1] [amount: u64 LE] |
| 64 | +fn build_mint_tokens_data(amount: u64) -> Vec<u8> { |
| 65 | + let mut data = vec![1u8]; |
| 66 | + data.extend_from_slice(&amount.to_le_bytes()); |
| 67 | + data |
| 68 | +} |
| 69 | + |
| 70 | +#[test] |
| 71 | +fn test_create_mint() { |
| 72 | + let mut svm = setup(); |
| 73 | + |
| 74 | + let payer = Pubkey::new_unique(); |
| 75 | + let (mint_pda, _) = Pubkey::find_program_address(&[b"mint"], &crate::ID); |
| 76 | + let token_program = quasar_svm::SPL_TOKEN_PROGRAM_ID; |
| 77 | + let system_program = quasar_svm::system_program::ID; |
| 78 | + let rent = quasar_svm::solana_sdk_ids::sysvar::rent::ID; |
| 79 | + |
| 80 | + let data = build_create_mint_data(9); |
| 81 | + |
| 82 | + let instruction = Instruction { |
| 83 | + program_id: crate::ID, |
| 84 | + accounts: vec![ |
| 85 | + solana_instruction::AccountMeta::new(payer.into(), true), |
| 86 | + solana_instruction::AccountMeta::new(mint_pda.into(), false), |
| 87 | + solana_instruction::AccountMeta::new_readonly(rent.into(), false), |
| 88 | + solana_instruction::AccountMeta::new_readonly(token_program.into(), false), |
| 89 | + solana_instruction::AccountMeta::new_readonly(system_program.into(), false), |
| 90 | + ], |
| 91 | + data, |
| 92 | + }; |
| 93 | + |
| 94 | + let result = svm.process_instruction(&instruction, &[signer(payer), empty(mint_pda)]); |
| 95 | + assert!(result.is_ok(), "create_mint failed: {:?}", result.raw_result); |
| 96 | + println!(" CREATE MINT CU: {}", result.compute_units_consumed); |
| 97 | +} |
| 98 | + |
| 99 | +#[test] |
| 100 | +fn test_mint_with_pda_authority() { |
| 101 | + let mut svm = setup(); |
| 102 | + |
| 103 | + let payer = Pubkey::new_unique(); |
| 104 | + let (mint_pda, _) = Pubkey::find_program_address(&[b"mint"], &crate::ID); |
| 105 | + let token_addr = Pubkey::new_unique(); |
| 106 | + let token_program = quasar_svm::SPL_TOKEN_PROGRAM_ID; |
| 107 | + |
| 108 | + let amount = 1_000_000_000u64; |
| 109 | + let data = build_mint_tokens_data(amount); |
| 110 | + |
| 111 | + let instruction = Instruction { |
| 112 | + program_id: crate::ID, |
| 113 | + accounts: vec![ |
| 114 | + solana_instruction::AccountMeta::new(payer.into(), true), |
| 115 | + solana_instruction::AccountMeta::new(mint_pda.into(), false), |
| 116 | + solana_instruction::AccountMeta::new(token_addr.into(), false), |
| 117 | + solana_instruction::AccountMeta::new_readonly(token_program.into(), false), |
| 118 | + ], |
| 119 | + data, |
| 120 | + }; |
| 121 | + |
| 122 | + let result = svm.process_instruction( |
| 123 | + &instruction, |
| 124 | + &[ |
| 125 | + signer(payer), |
| 126 | + // The mint authority is the mint_pda itself |
| 127 | + mint_account(mint_pda, mint_pda), |
| 128 | + token_account(token_addr, mint_pda, payer, 0), |
| 129 | + ], |
| 130 | + ); |
| 131 | + |
| 132 | + assert!(result.is_ok(), "mint_tokens failed: {:?}", result.raw_result); |
| 133 | + println!(" MINT WITH PDA CU: {}", result.compute_units_consumed); |
| 134 | +} |
0 commit comments