|
|
@@ -0,0 +1,756 @@
|
|
|
+mod setup;
|
|
|
+
|
|
|
+use assert_matches::assert_matches;
|
|
|
+use setup::{mint, TOKEN_PROGRAM_ID};
|
|
|
+use solana_program_test::{tokio, BanksClientError, ProgramTest};
|
|
|
+use solana_sdk::{
|
|
|
+ instruction::InstructionError,
|
|
|
+ pubkey::Pubkey,
|
|
|
+ signature::{Keypair, Signer},
|
|
|
+ system_instruction,
|
|
|
+ transaction::{Transaction, TransactionError},
|
|
|
+};
|
|
|
+use spl_token_interface::state::{account::Account, mint::Mint, multisig::Multisig};
|
|
|
+use std::mem::size_of;
|
|
|
+
|
|
|
+#[test_case::test_case(TOKEN_PROGRAM_ID ; "p-token")]
|
|
|
+#[tokio::test]
|
|
|
+async fn withdraw_excess_lamports_from_mint(token_program: Pubkey) {
|
|
|
+ let context = ProgramTest::new("pinocchio_token_program", TOKEN_PROGRAM_ID, None)
|
|
|
+ .start_with_context()
|
|
|
+ .await;
|
|
|
+
|
|
|
+ let excess_lamports = 4_000_000_000_000;
|
|
|
+
|
|
|
+ // Given a mint authority, freeze authority and an account keypair.
|
|
|
+
|
|
|
+ let mint_authority = Keypair::new();
|
|
|
+ let freeze_authority = Pubkey::new_unique();
|
|
|
+ let account = Keypair::new();
|
|
|
+ let account_pubkey = account.pubkey();
|
|
|
+
|
|
|
+ let account_size = size_of::<Mint>();
|
|
|
+ let rent = context.banks_client.get_rent().await.unwrap();
|
|
|
+
|
|
|
+ let mut initialize_ix = spl_token::instruction::initialize_mint(
|
|
|
+ &spl_token::ID,
|
|
|
+ &account.pubkey(),
|
|
|
+ &mint_authority.pubkey(),
|
|
|
+ Some(&freeze_authority),
|
|
|
+ 0,
|
|
|
+ )
|
|
|
+ .unwrap();
|
|
|
+ // Switches the program id to the token program.
|
|
|
+ initialize_ix.program_id = token_program;
|
|
|
+
|
|
|
+ // And we initialize a mint account with excess lamports.
|
|
|
+
|
|
|
+ let instructions = vec![
|
|
|
+ system_instruction::create_account(
|
|
|
+ &context.payer.pubkey(),
|
|
|
+ &account.pubkey(),
|
|
|
+ rent.minimum_balance(account_size) + excess_lamports,
|
|
|
+ account_size as u64,
|
|
|
+ &token_program,
|
|
|
+ ),
|
|
|
+ initialize_ix,
|
|
|
+ ];
|
|
|
+
|
|
|
+ let tx = Transaction::new_signed_with_payer(
|
|
|
+ &instructions,
|
|
|
+ Some(&context.payer.pubkey()),
|
|
|
+ &[&context.payer, &account],
|
|
|
+ context.last_blockhash,
|
|
|
+ );
|
|
|
+ context.banks_client.process_transaction(tx).await.unwrap();
|
|
|
+
|
|
|
+ let account = context
|
|
|
+ .banks_client
|
|
|
+ .get_account(account.pubkey())
|
|
|
+ .await
|
|
|
+ .unwrap();
|
|
|
+
|
|
|
+ assert!(account.is_some());
|
|
|
+
|
|
|
+ let account = account.unwrap();
|
|
|
+ assert_eq!(
|
|
|
+ account.lamports,
|
|
|
+ rent.minimum_balance(account_size) + excess_lamports
|
|
|
+ );
|
|
|
+
|
|
|
+ // When we withdraw the excess lamports.
|
|
|
+
|
|
|
+ let destination = Pubkey::new_unique();
|
|
|
+
|
|
|
+ let mut withdraw_ix = spl_token_2022::instruction::withdraw_excess_lamports(
|
|
|
+ &spl_token_2022::ID,
|
|
|
+ &account_pubkey,
|
|
|
+ &destination,
|
|
|
+ &mint_authority.pubkey(),
|
|
|
+ &[],
|
|
|
+ )
|
|
|
+ .unwrap();
|
|
|
+ // Switches the program id to the token program.
|
|
|
+ withdraw_ix.program_id = token_program;
|
|
|
+
|
|
|
+ let tx = Transaction::new_signed_with_payer(
|
|
|
+ &[withdraw_ix],
|
|
|
+ Some(&context.payer.pubkey()),
|
|
|
+ &[&context.payer, &mint_authority],
|
|
|
+ context.last_blockhash,
|
|
|
+ );
|
|
|
+ context.banks_client.process_transaction(tx).await.unwrap();
|
|
|
+
|
|
|
+ // Then the destination account has the excess lamports.
|
|
|
+
|
|
|
+ let destination = context.banks_client.get_account(destination).await.unwrap();
|
|
|
+
|
|
|
+ assert!(destination.is_some());
|
|
|
+
|
|
|
+ let destination = destination.unwrap();
|
|
|
+ assert_eq!(destination.lamports, excess_lamports);
|
|
|
+}
|
|
|
+
|
|
|
+#[test_case::test_case(TOKEN_PROGRAM_ID ; "p-token")]
|
|
|
+#[tokio::test]
|
|
|
+async fn withdraw_excess_lamports_from_account(token_program: Pubkey) {
|
|
|
+ let mut context = ProgramTest::new("pinocchio_token_program", TOKEN_PROGRAM_ID, None)
|
|
|
+ .start_with_context()
|
|
|
+ .await;
|
|
|
+
|
|
|
+ let excess_lamports = 4_000_000_000_000;
|
|
|
+
|
|
|
+ // Given a mint account.
|
|
|
+
|
|
|
+ let mint_authority = Pubkey::new_unique();
|
|
|
+ let freeze_authority = Pubkey::new_unique();
|
|
|
+
|
|
|
+ let mint = mint::initialize(
|
|
|
+ &mut context,
|
|
|
+ mint_authority,
|
|
|
+ Some(freeze_authority),
|
|
|
+ &token_program,
|
|
|
+ )
|
|
|
+ .await
|
|
|
+ .unwrap();
|
|
|
+
|
|
|
+ // Given a mint authority, freeze authority and an account keypair.
|
|
|
+
|
|
|
+ let owner = Keypair::new();
|
|
|
+ let account = Keypair::new();
|
|
|
+ let account_pubkey = account.pubkey();
|
|
|
+
|
|
|
+ let account_size = size_of::<Account>();
|
|
|
+ let rent = context.banks_client.get_rent().await.unwrap();
|
|
|
+
|
|
|
+ let mut initialize_ix = spl_token::instruction::initialize_account(
|
|
|
+ &spl_token::ID,
|
|
|
+ &account.pubkey(),
|
|
|
+ &mint,
|
|
|
+ &owner.pubkey(),
|
|
|
+ )
|
|
|
+ .unwrap();
|
|
|
+ // Switches the program id to the token program.
|
|
|
+ initialize_ix.program_id = token_program;
|
|
|
+
|
|
|
+ // When a new mint account is created and initialized.
|
|
|
+
|
|
|
+ let instructions = vec![
|
|
|
+ system_instruction::create_account(
|
|
|
+ &context.payer.pubkey(),
|
|
|
+ &account.pubkey(),
|
|
|
+ rent.minimum_balance(account_size) + excess_lamports,
|
|
|
+ account_size as u64,
|
|
|
+ &token_program,
|
|
|
+ ),
|
|
|
+ initialize_ix,
|
|
|
+ ];
|
|
|
+
|
|
|
+ let tx = Transaction::new_signed_with_payer(
|
|
|
+ &instructions,
|
|
|
+ Some(&context.payer.pubkey()),
|
|
|
+ &[&context.payer, &account],
|
|
|
+ context.last_blockhash,
|
|
|
+ );
|
|
|
+ context.banks_client.process_transaction(tx).await.unwrap();
|
|
|
+
|
|
|
+ let account = context
|
|
|
+ .banks_client
|
|
|
+ .get_account(account.pubkey())
|
|
|
+ .await
|
|
|
+ .unwrap();
|
|
|
+
|
|
|
+ assert!(account.is_some());
|
|
|
+
|
|
|
+ let account = account.unwrap();
|
|
|
+ assert_eq!(
|
|
|
+ account.lamports,
|
|
|
+ rent.minimum_balance(account_size) + excess_lamports
|
|
|
+ );
|
|
|
+
|
|
|
+ // When we withdraw the excess lamports.
|
|
|
+
|
|
|
+ let destination = Pubkey::new_unique();
|
|
|
+
|
|
|
+ let mut withdraw_ix = spl_token_2022::instruction::withdraw_excess_lamports(
|
|
|
+ &spl_token_2022::ID,
|
|
|
+ &account_pubkey,
|
|
|
+ &destination,
|
|
|
+ &owner.pubkey(),
|
|
|
+ &[],
|
|
|
+ )
|
|
|
+ .unwrap();
|
|
|
+ // Switches the program id to the token program.
|
|
|
+ withdraw_ix.program_id = token_program;
|
|
|
+
|
|
|
+ let tx = Transaction::new_signed_with_payer(
|
|
|
+ &[withdraw_ix],
|
|
|
+ Some(&context.payer.pubkey()),
|
|
|
+ &[&context.payer, &owner],
|
|
|
+ context.last_blockhash,
|
|
|
+ );
|
|
|
+ context.banks_client.process_transaction(tx).await.unwrap();
|
|
|
+
|
|
|
+ // Then the destination account has the excess lamports.
|
|
|
+
|
|
|
+ let destination = context.banks_client.get_account(destination).await.unwrap();
|
|
|
+
|
|
|
+ assert!(destination.is_some());
|
|
|
+
|
|
|
+ let destination = destination.unwrap();
|
|
|
+ assert_eq!(destination.lamports, excess_lamports);
|
|
|
+}
|
|
|
+
|
|
|
+#[test_case::test_case(TOKEN_PROGRAM_ID ; "p-token")]
|
|
|
+#[tokio::test]
|
|
|
+async fn withdraw_excess_lamports_from_multisig(token_program: Pubkey) {
|
|
|
+ let context = ProgramTest::new("pinocchio_token_program", TOKEN_PROGRAM_ID, None)
|
|
|
+ .start_with_context()
|
|
|
+ .await;
|
|
|
+
|
|
|
+ let excess_lamports = 4_000_000_000_000;
|
|
|
+
|
|
|
+ // Given an account
|
|
|
+
|
|
|
+ let multisig = Keypair::new();
|
|
|
+ let signer1 = Keypair::new();
|
|
|
+ let signer1_pubkey = signer1.pubkey();
|
|
|
+ let signer2 = Keypair::new();
|
|
|
+ let signer2_pubkey = signer2.pubkey();
|
|
|
+ let signer3 = Keypair::new();
|
|
|
+ let signer3_pubkey = signer3.pubkey();
|
|
|
+ let signers = vec![&signer1_pubkey, &signer2_pubkey, &signer3_pubkey];
|
|
|
+
|
|
|
+ let rent = context.banks_client.get_rent().await.unwrap();
|
|
|
+ let account_size = size_of::<Multisig>();
|
|
|
+
|
|
|
+ let mut initialize_ix = spl_token::instruction::initialize_multisig(
|
|
|
+ &spl_token::ID,
|
|
|
+ &multisig.pubkey(),
|
|
|
+ &signers,
|
|
|
+ 3,
|
|
|
+ )
|
|
|
+ .unwrap();
|
|
|
+ // Switches the program id to the token program.
|
|
|
+ initialize_ix.program_id = token_program;
|
|
|
+
|
|
|
+ // And we initialize the multisig account.
|
|
|
+
|
|
|
+ let instructions = vec![
|
|
|
+ system_instruction::create_account(
|
|
|
+ &context.payer.pubkey(),
|
|
|
+ &multisig.pubkey(),
|
|
|
+ rent.minimum_balance(account_size) + excess_lamports,
|
|
|
+ account_size as u64,
|
|
|
+ &token_program,
|
|
|
+ ),
|
|
|
+ initialize_ix,
|
|
|
+ ];
|
|
|
+
|
|
|
+ let tx = Transaction::new_signed_with_payer(
|
|
|
+ &instructions,
|
|
|
+ Some(&context.payer.pubkey()),
|
|
|
+ &[&context.payer, &multisig],
|
|
|
+ context.last_blockhash,
|
|
|
+ );
|
|
|
+ context.banks_client.process_transaction(tx).await.unwrap();
|
|
|
+
|
|
|
+ let account = context
|
|
|
+ .banks_client
|
|
|
+ .get_account(multisig.pubkey())
|
|
|
+ .await
|
|
|
+ .unwrap();
|
|
|
+
|
|
|
+ assert!(account.is_some());
|
|
|
+
|
|
|
+ let account = account.unwrap();
|
|
|
+ assert_eq!(
|
|
|
+ account.lamports,
|
|
|
+ rent.minimum_balance(account_size) + excess_lamports
|
|
|
+ );
|
|
|
+
|
|
|
+ // When we withdraw the excess lamports.
|
|
|
+
|
|
|
+ let destination = Pubkey::new_unique();
|
|
|
+
|
|
|
+ let mut withdraw_ix = spl_token_2022::instruction::withdraw_excess_lamports(
|
|
|
+ &spl_token_2022::ID,
|
|
|
+ &multisig.pubkey(),
|
|
|
+ &destination,
|
|
|
+ &multisig.pubkey(),
|
|
|
+ &signers,
|
|
|
+ )
|
|
|
+ .unwrap();
|
|
|
+ // Switches the program id to the token program.
|
|
|
+ withdraw_ix.program_id = token_program;
|
|
|
+
|
|
|
+ let tx = Transaction::new_signed_with_payer(
|
|
|
+ &[withdraw_ix],
|
|
|
+ Some(&context.payer.pubkey()),
|
|
|
+ &[&context.payer, &signer1, &signer2, &signer3],
|
|
|
+ context.last_blockhash,
|
|
|
+ );
|
|
|
+ context.banks_client.process_transaction(tx).await.unwrap();
|
|
|
+
|
|
|
+ // Then the destination account has the excess lamports.
|
|
|
+
|
|
|
+ let destination = context.banks_client.get_account(destination).await.unwrap();
|
|
|
+
|
|
|
+ assert!(destination.is_some());
|
|
|
+
|
|
|
+ let destination = destination.unwrap();
|
|
|
+ assert_eq!(destination.lamports, excess_lamports);
|
|
|
+}
|
|
|
+
|
|
|
+#[test_case::test_case(TOKEN_PROGRAM_ID ; "p-token")]
|
|
|
+#[tokio::test]
|
|
|
+async fn fail_withdraw_excess_lamports_from_mint_wrong_authority(token_program: Pubkey) {
|
|
|
+ let context = ProgramTest::new("pinocchio_token_program", TOKEN_PROGRAM_ID, None)
|
|
|
+ .start_with_context()
|
|
|
+ .await;
|
|
|
+
|
|
|
+ let excess_lamports = 4_000_000_000_000;
|
|
|
+
|
|
|
+ // Given a mint authority, freeze authority and an account keypair.
|
|
|
+
|
|
|
+ let mint_authority = Keypair::new();
|
|
|
+ let freeze_authority = Pubkey::new_unique();
|
|
|
+ let account = Keypair::new();
|
|
|
+ let account_pubkey = account.pubkey();
|
|
|
+
|
|
|
+ let account_size = size_of::<Mint>();
|
|
|
+ let rent = context.banks_client.get_rent().await.unwrap();
|
|
|
+
|
|
|
+ let mut initialize_ix = spl_token::instruction::initialize_mint(
|
|
|
+ &spl_token::ID,
|
|
|
+ &account.pubkey(),
|
|
|
+ &mint_authority.pubkey(),
|
|
|
+ Some(&freeze_authority),
|
|
|
+ 0,
|
|
|
+ )
|
|
|
+ .unwrap();
|
|
|
+ // Switches the program id to the token program.
|
|
|
+ initialize_ix.program_id = token_program;
|
|
|
+
|
|
|
+ // And we initialize a mint account with excess lamports.
|
|
|
+
|
|
|
+ let instructions = vec![
|
|
|
+ system_instruction::create_account(
|
|
|
+ &context.payer.pubkey(),
|
|
|
+ &account.pubkey(),
|
|
|
+ rent.minimum_balance(account_size) + excess_lamports,
|
|
|
+ account_size as u64,
|
|
|
+ &token_program,
|
|
|
+ ),
|
|
|
+ initialize_ix,
|
|
|
+ ];
|
|
|
+
|
|
|
+ let tx = Transaction::new_signed_with_payer(
|
|
|
+ &instructions,
|
|
|
+ Some(&context.payer.pubkey()),
|
|
|
+ &[&context.payer, &account],
|
|
|
+ context.last_blockhash,
|
|
|
+ );
|
|
|
+ context.banks_client.process_transaction(tx).await.unwrap();
|
|
|
+
|
|
|
+ let account = context
|
|
|
+ .banks_client
|
|
|
+ .get_account(account.pubkey())
|
|
|
+ .await
|
|
|
+ .unwrap();
|
|
|
+
|
|
|
+ assert!(account.is_some());
|
|
|
+
|
|
|
+ let account = account.unwrap();
|
|
|
+ assert_eq!(
|
|
|
+ account.lamports,
|
|
|
+ rent.minimum_balance(account_size) + excess_lamports
|
|
|
+ );
|
|
|
+
|
|
|
+ // When we try to withdraw the excess lamports with the wrong authority.
|
|
|
+
|
|
|
+ let destination = Pubkey::new_unique();
|
|
|
+ let wrong_authority = Keypair::new();
|
|
|
+
|
|
|
+ let mut withdraw_ix = spl_token_2022::instruction::withdraw_excess_lamports(
|
|
|
+ &spl_token_2022::ID,
|
|
|
+ &account_pubkey,
|
|
|
+ &destination,
|
|
|
+ &wrong_authority.pubkey(),
|
|
|
+ &[],
|
|
|
+ )
|
|
|
+ .unwrap();
|
|
|
+ // Switches the program id to the token program.
|
|
|
+ withdraw_ix.program_id = token_program;
|
|
|
+
|
|
|
+ let tx = Transaction::new_signed_with_payer(
|
|
|
+ &[withdraw_ix],
|
|
|
+ Some(&context.payer.pubkey()),
|
|
|
+ &[&context.payer, &wrong_authority],
|
|
|
+ context.last_blockhash,
|
|
|
+ );
|
|
|
+ let error = context
|
|
|
+ .banks_client
|
|
|
+ .process_transaction(tx)
|
|
|
+ .await
|
|
|
+ .unwrap_err();
|
|
|
+
|
|
|
+ // The we expect an error.
|
|
|
+
|
|
|
+ assert_matches!(
|
|
|
+ error,
|
|
|
+ BanksClientError::TransactionError(TransactionError::InstructionError(
|
|
|
+ _,
|
|
|
+ InstructionError::Custom(4) // TokenError::OwnerMismatch
|
|
|
+ ))
|
|
|
+ );
|
|
|
+}
|
|
|
+
|
|
|
+#[test_case::test_case(TOKEN_PROGRAM_ID ; "p-token")]
|
|
|
+#[tokio::test]
|
|
|
+async fn fail_withdraw_excess_lamports_from_account_wrong_authority(token_program: Pubkey) {
|
|
|
+ let mut context = ProgramTest::new("pinocchio_token_program", TOKEN_PROGRAM_ID, None)
|
|
|
+ .start_with_context()
|
|
|
+ .await;
|
|
|
+
|
|
|
+ let excess_lamports = 4_000_000_000_000;
|
|
|
+
|
|
|
+ // Given a mint account.
|
|
|
+
|
|
|
+ let mint_authority = Pubkey::new_unique();
|
|
|
+ let freeze_authority = Pubkey::new_unique();
|
|
|
+
|
|
|
+ let mint = mint::initialize(
|
|
|
+ &mut context,
|
|
|
+ mint_authority,
|
|
|
+ Some(freeze_authority),
|
|
|
+ &token_program,
|
|
|
+ )
|
|
|
+ .await
|
|
|
+ .unwrap();
|
|
|
+
|
|
|
+ // Given a mint authority, freeze authority and an account keypair.
|
|
|
+
|
|
|
+ let owner = Keypair::new();
|
|
|
+ let account = Keypair::new();
|
|
|
+ let account_pubkey = account.pubkey();
|
|
|
+
|
|
|
+ let account_size = size_of::<Account>();
|
|
|
+ let rent = context.banks_client.get_rent().await.unwrap();
|
|
|
+
|
|
|
+ let mut initialize_ix = spl_token::instruction::initialize_account(
|
|
|
+ &spl_token::ID,
|
|
|
+ &account.pubkey(),
|
|
|
+ &mint,
|
|
|
+ &owner.pubkey(),
|
|
|
+ )
|
|
|
+ .unwrap();
|
|
|
+ // Switches the program id to the token program.
|
|
|
+ initialize_ix.program_id = token_program;
|
|
|
+
|
|
|
+ // When a new mint account is created and initialized.
|
|
|
+
|
|
|
+ let instructions = vec![
|
|
|
+ system_instruction::create_account(
|
|
|
+ &context.payer.pubkey(),
|
|
|
+ &account.pubkey(),
|
|
|
+ rent.minimum_balance(account_size) + excess_lamports,
|
|
|
+ account_size as u64,
|
|
|
+ &token_program,
|
|
|
+ ),
|
|
|
+ initialize_ix,
|
|
|
+ ];
|
|
|
+
|
|
|
+ let tx = Transaction::new_signed_with_payer(
|
|
|
+ &instructions,
|
|
|
+ Some(&context.payer.pubkey()),
|
|
|
+ &[&context.payer, &account],
|
|
|
+ context.last_blockhash,
|
|
|
+ );
|
|
|
+ context.banks_client.process_transaction(tx).await.unwrap();
|
|
|
+
|
|
|
+ let account = context
|
|
|
+ .banks_client
|
|
|
+ .get_account(account.pubkey())
|
|
|
+ .await
|
|
|
+ .unwrap();
|
|
|
+
|
|
|
+ assert!(account.is_some());
|
|
|
+
|
|
|
+ let account = account.unwrap();
|
|
|
+ assert_eq!(
|
|
|
+ account.lamports,
|
|
|
+ rent.minimum_balance(account_size) + excess_lamports
|
|
|
+ );
|
|
|
+
|
|
|
+ // When we try to withdraw the excess lamports with the wrong owner.
|
|
|
+
|
|
|
+ let destination = Pubkey::new_unique();
|
|
|
+ let wrong_owner = Keypair::new();
|
|
|
+
|
|
|
+ let mut withdraw_ix = spl_token_2022::instruction::withdraw_excess_lamports(
|
|
|
+ &spl_token_2022::ID,
|
|
|
+ &account_pubkey,
|
|
|
+ &destination,
|
|
|
+ &wrong_owner.pubkey(),
|
|
|
+ &[],
|
|
|
+ )
|
|
|
+ .unwrap();
|
|
|
+ // Switches the program id to the token program.
|
|
|
+ withdraw_ix.program_id = token_program;
|
|
|
+
|
|
|
+ let tx = Transaction::new_signed_with_payer(
|
|
|
+ &[withdraw_ix],
|
|
|
+ Some(&context.payer.pubkey()),
|
|
|
+ &[&context.payer, &wrong_owner],
|
|
|
+ context.last_blockhash,
|
|
|
+ );
|
|
|
+ let error = context
|
|
|
+ .banks_client
|
|
|
+ .process_transaction(tx)
|
|
|
+ .await
|
|
|
+ .unwrap_err();
|
|
|
+
|
|
|
+ // The we expect an error.
|
|
|
+
|
|
|
+ assert_matches!(
|
|
|
+ error,
|
|
|
+ BanksClientError::TransactionError(TransactionError::InstructionError(
|
|
|
+ _,
|
|
|
+ InstructionError::Custom(4) // TokenError::OwnerMismatch
|
|
|
+ ))
|
|
|
+ );
|
|
|
+}
|
|
|
+
|
|
|
+#[test_case::test_case(TOKEN_PROGRAM_ID ; "p-token")]
|
|
|
+#[tokio::test]
|
|
|
+async fn fail_withdraw_excess_lamports_from_multisig_wrong_authority(token_program: Pubkey) {
|
|
|
+ let context = ProgramTest::new("pinocchio_token_program", TOKEN_PROGRAM_ID, None)
|
|
|
+ .start_with_context()
|
|
|
+ .await;
|
|
|
+
|
|
|
+ let excess_lamports = 4_000_000_000_000;
|
|
|
+
|
|
|
+ // Given an account
|
|
|
+
|
|
|
+ let multisig = Keypair::new();
|
|
|
+ let signer1 = Keypair::new();
|
|
|
+ let signer1_pubkey = signer1.pubkey();
|
|
|
+ let signer2 = Keypair::new();
|
|
|
+ let signer2_pubkey = signer2.pubkey();
|
|
|
+ let signer3 = Keypair::new();
|
|
|
+ let signer3_pubkey = signer3.pubkey();
|
|
|
+ let signers = vec![&signer1_pubkey, &signer2_pubkey, &signer3_pubkey];
|
|
|
+
|
|
|
+ let rent = context.banks_client.get_rent().await.unwrap();
|
|
|
+ let account_size = size_of::<Multisig>();
|
|
|
+
|
|
|
+ let mut initialize_ix = spl_token::instruction::initialize_multisig(
|
|
|
+ &spl_token::ID,
|
|
|
+ &multisig.pubkey(),
|
|
|
+ &signers,
|
|
|
+ 3,
|
|
|
+ )
|
|
|
+ .unwrap();
|
|
|
+ // Switches the program id to the token program.
|
|
|
+ initialize_ix.program_id = token_program;
|
|
|
+
|
|
|
+ // And we initialize the multisig account.
|
|
|
+
|
|
|
+ let instructions = vec![
|
|
|
+ system_instruction::create_account(
|
|
|
+ &context.payer.pubkey(),
|
|
|
+ &multisig.pubkey(),
|
|
|
+ rent.minimum_balance(account_size) + excess_lamports,
|
|
|
+ account_size as u64,
|
|
|
+ &token_program,
|
|
|
+ ),
|
|
|
+ initialize_ix,
|
|
|
+ ];
|
|
|
+
|
|
|
+ let tx = Transaction::new_signed_with_payer(
|
|
|
+ &instructions,
|
|
|
+ Some(&context.payer.pubkey()),
|
|
|
+ &[&context.payer, &multisig],
|
|
|
+ context.last_blockhash,
|
|
|
+ );
|
|
|
+ context.banks_client.process_transaction(tx).await.unwrap();
|
|
|
+
|
|
|
+ let account = context
|
|
|
+ .banks_client
|
|
|
+ .get_account(multisig.pubkey())
|
|
|
+ .await
|
|
|
+ .unwrap();
|
|
|
+
|
|
|
+ assert!(account.is_some());
|
|
|
+
|
|
|
+ let account = account.unwrap();
|
|
|
+ assert_eq!(
|
|
|
+ account.lamports,
|
|
|
+ rent.minimum_balance(account_size) + excess_lamports
|
|
|
+ );
|
|
|
+
|
|
|
+ // When we try to withdraw the excess lamports with the wrong authority.
|
|
|
+
|
|
|
+ let destination = Pubkey::new_unique();
|
|
|
+ let wrong_authority = Keypair::new();
|
|
|
+
|
|
|
+ let mut withdraw_ix = spl_token_2022::instruction::withdraw_excess_lamports(
|
|
|
+ &spl_token_2022::ID,
|
|
|
+ &multisig.pubkey(),
|
|
|
+ &destination,
|
|
|
+ &wrong_authority.pubkey(),
|
|
|
+ &signers,
|
|
|
+ )
|
|
|
+ .unwrap();
|
|
|
+ // Switches the program id to the token program.
|
|
|
+ withdraw_ix.program_id = token_program;
|
|
|
+
|
|
|
+ let tx = Transaction::new_signed_with_payer(
|
|
|
+ &[withdraw_ix],
|
|
|
+ Some(&context.payer.pubkey()),
|
|
|
+ &[&context.payer, &signer1, &signer2, &signer3],
|
|
|
+ context.last_blockhash,
|
|
|
+ );
|
|
|
+ let error = context
|
|
|
+ .banks_client
|
|
|
+ .process_transaction(tx)
|
|
|
+ .await
|
|
|
+ .unwrap_err();
|
|
|
+
|
|
|
+ // The we expect an error.
|
|
|
+
|
|
|
+ assert_matches!(
|
|
|
+ error,
|
|
|
+ BanksClientError::TransactionError(TransactionError::InstructionError(
|
|
|
+ _,
|
|
|
+ InstructionError::Custom(4) // TokenError::OwnerMismatch
|
|
|
+ ))
|
|
|
+ );
|
|
|
+}
|
|
|
+
|
|
|
+#[test_case::test_case(TOKEN_PROGRAM_ID ; "p-token")]
|
|
|
+#[tokio::test]
|
|
|
+async fn fail_withdraw_excess_lamports_from_multisig_missing_signer(token_program: Pubkey) {
|
|
|
+ let context = ProgramTest::new("pinocchio_token_program", TOKEN_PROGRAM_ID, None)
|
|
|
+ .start_with_context()
|
|
|
+ .await;
|
|
|
+
|
|
|
+ let excess_lamports = 4_000_000_000_000;
|
|
|
+
|
|
|
+ // Given an account
|
|
|
+
|
|
|
+ let multisig = Keypair::new();
|
|
|
+ let signer1 = Keypair::new();
|
|
|
+ let signer1_pubkey = signer1.pubkey();
|
|
|
+ let signer2 = Keypair::new();
|
|
|
+ let signer2_pubkey = signer2.pubkey();
|
|
|
+ let signer3 = Keypair::new();
|
|
|
+ let signer3_pubkey = signer3.pubkey();
|
|
|
+ let signers = vec![&signer1_pubkey, &signer2_pubkey, &signer3_pubkey];
|
|
|
+
|
|
|
+ let rent = context.banks_client.get_rent().await.unwrap();
|
|
|
+ let account_size = size_of::<Multisig>();
|
|
|
+
|
|
|
+ let mut initialize_ix = spl_token::instruction::initialize_multisig(
|
|
|
+ &spl_token::ID,
|
|
|
+ &multisig.pubkey(),
|
|
|
+ &signers,
|
|
|
+ 3,
|
|
|
+ )
|
|
|
+ .unwrap();
|
|
|
+ // Switches the program id to the token program.
|
|
|
+ initialize_ix.program_id = token_program;
|
|
|
+
|
|
|
+ // And we initialize the multisig account.
|
|
|
+
|
|
|
+ let instructions = vec![
|
|
|
+ system_instruction::create_account(
|
|
|
+ &context.payer.pubkey(),
|
|
|
+ &multisig.pubkey(),
|
|
|
+ rent.minimum_balance(account_size) + excess_lamports,
|
|
|
+ account_size as u64,
|
|
|
+ &token_program,
|
|
|
+ ),
|
|
|
+ initialize_ix,
|
|
|
+ ];
|
|
|
+
|
|
|
+ let tx = Transaction::new_signed_with_payer(
|
|
|
+ &instructions,
|
|
|
+ Some(&context.payer.pubkey()),
|
|
|
+ &[&context.payer, &multisig],
|
|
|
+ context.last_blockhash,
|
|
|
+ );
|
|
|
+ context.banks_client.process_transaction(tx).await.unwrap();
|
|
|
+
|
|
|
+ let account = context
|
|
|
+ .banks_client
|
|
|
+ .get_account(multisig.pubkey())
|
|
|
+ .await
|
|
|
+ .unwrap();
|
|
|
+
|
|
|
+ assert!(account.is_some());
|
|
|
+
|
|
|
+ let account = account.unwrap();
|
|
|
+ assert_eq!(
|
|
|
+ account.lamports,
|
|
|
+ rent.minimum_balance(account_size) + excess_lamports
|
|
|
+ );
|
|
|
+
|
|
|
+ // When we try to withdraw the excess lamports with the wrong authority.
|
|
|
+
|
|
|
+ let destination = Pubkey::new_unique();
|
|
|
+
|
|
|
+ let mut withdraw_ix = spl_token_2022::instruction::withdraw_excess_lamports(
|
|
|
+ &spl_token_2022::ID,
|
|
|
+ &multisig.pubkey(),
|
|
|
+ &destination,
|
|
|
+ &multisig.pubkey(),
|
|
|
+ &[&signer1_pubkey, &signer2_pubkey],
|
|
|
+ )
|
|
|
+ .unwrap();
|
|
|
+ // Switches the program id to the token program.
|
|
|
+ withdraw_ix.program_id = token_program;
|
|
|
+
|
|
|
+ let tx = Transaction::new_signed_with_payer(
|
|
|
+ &[withdraw_ix],
|
|
|
+ Some(&context.payer.pubkey()),
|
|
|
+ &[&context.payer, &signer1, &signer2],
|
|
|
+ context.last_blockhash,
|
|
|
+ );
|
|
|
+ let error = context
|
|
|
+ .banks_client
|
|
|
+ .process_transaction(tx)
|
|
|
+ .await
|
|
|
+ .unwrap_err();
|
|
|
+
|
|
|
+ // The we expect an error.
|
|
|
+
|
|
|
+ assert_matches!(
|
|
|
+ error,
|
|
|
+ BanksClientError::TransactionError(TransactionError::InstructionError(
|
|
|
+ _,
|
|
|
+ InstructionError::MissingRequiredSignature
|
|
|
+ ))
|
|
|
+ );
|
|
|
+}
|
Fernando Otero