Step 4: Replace spl_token::id() with a parameter

This commit is contained in:
Jon Cinque 2022-08-02 01:39:49 +02:00
parent 3955c3f994
commit 4b4214e1aa
1 changed files with 5 additions and 4 deletions

View File

@ -141,6 +141,7 @@ impl Processor {
let swap_bytes = swap.to_bytes(); let swap_bytes = swap.to_bytes();
let authority_signature_seeds = [&swap_bytes[..32], &[bump_seed]]; let authority_signature_seeds = [&swap_bytes[..32], &[bump_seed]];
let signers = &[&authority_signature_seeds[..]]; let signers = &[&authority_signature_seeds[..]];
#[allow(deprecated)]
let ix = spl_token_2022::instruction::transfer( let ix = spl_token_2022::instruction::transfer(
token_program.key, token_program.key,
source.key, source.key,
@ -1139,7 +1140,7 @@ mod tests {
let mut new_account_infos = vec![]; let mut new_account_infos = vec![];
// mimic check for token program in accounts // mimic check for token program in accounts
if !account_infos.iter().any(|x| *x.key == spl_token::id()) { if !account_infos.iter().any(|x| *x.key == spl_token::id() || *x.key == spl_token_2022::id()) {
return Err(ProgramError::InvalidAccountData); return Err(ProgramError::InvalidAccountData);
} }
@ -1225,7 +1226,7 @@ mod tests {
Pubkey::find_program_address(&[&swap_key.to_bytes()[..]], &SWAP_PROGRAM_ID); Pubkey::find_program_address(&[&swap_key.to_bytes()[..]], &SWAP_PROGRAM_ID);
let (pool_mint_key, mut pool_mint_account) = let (pool_mint_key, mut pool_mint_account) =
create_mint(&spl_token::id(), &authority_key, None); create_mint(&token_program_id, &authority_key, None);
let (pool_token_key, pool_token_account) = mint_token( let (pool_token_key, pool_token_account) = mint_token(
token_program_id, token_program_id,
&pool_mint_key, &pool_mint_key,
@ -1243,7 +1244,7 @@ mod tests {
0, 0,
); );
let (token_a_mint_key, mut token_a_mint_account) = let (token_a_mint_key, mut token_a_mint_account) =
create_mint(&spl_token::id(), user_key, None); create_mint(&token_program_id, user_key, None);
let (token_a_key, token_a_account) = mint_token( let (token_a_key, token_a_account) = mint_token(
token_program_id, token_program_id,
&token_a_mint_key, &token_a_mint_key,
@ -1253,7 +1254,7 @@ mod tests {
token_a_amount, token_a_amount,
); );
let (token_b_mint_key, mut token_b_mint_account) = let (token_b_mint_key, mut token_b_mint_account) =
create_mint(&spl_token::id(), user_key, None); create_mint(&token_program_id, user_key, None);
let (token_b_key, token_b_account) = mint_token( let (token_b_key, token_b_account) = mint_token(
token_program_id, token_program_id,
&token_b_mint_key, &token_b_mint_key,