chore: Add extra test for program fee

Author: Barnabas Ratki

contracts/svm/testing/src/express_relay/swap.rs

@@ -35,6 +35,7 @@ pub struct SwapParamOverride {
     pub searcher_ta_mint_user:          Option<Pubkey>,
     pub express_relay_fee_receiver_ata: Option<Pubkey>,
     pub relayer_fee_receiver_ata:       Option<Pubkey>,
+    pub token_program_fee:              Option<Pubkey>,
 }
 
 pub trait AnyVersionSwapArgs: Sized + Send + Sync {
@@ -96,6 +97,7 @@ pub fn create_swap_instruction(swap_params: SwapParams<impl AnyVersionSwapArgs>)
                 mint_fee: mint_fee_override,
                 express_relay_fee_receiver_ata,
                 relayer_fee_receiver_ata,
+                token_program_fee: token_program_fee_override,
             },
     } = swap_params;
     let express_relay_metadata = get_express_relay_metadata_key();
@@ -161,7 +163,7 @@ pub fn create_swap_instruction(swap_params: SwapParams<impl AnyVersionSwapArgs>)
         mint_fee,
         token_program_searcher,
         token_program_user,
-        token_program_fee,
+        token_program_fee: token_program_fee_override.unwrap_or(token_program_fee),
         express_relay_metadata,
         relayer_signer,
     }

contracts/svm/testing/tests/swap.rs

@@ -846,6 +846,58 @@ fn test_swap_wrong_mint_fee() {
     );
 }
 
+#[test]
+fn test_swap_wrong_token_program_fee() {
+    let SwapSetupResult {
+        mut svm,
+        user,
+        searcher,
+        token_searcher,
+        token_user,
+        router_ta_mint_searcher,
+        relayer_signer,
+        ..
+    } = setup_swap(Default::default());
+
+    let express_relay_metadata = get_express_relay_metadata(&mut svm);
+
+    let swap_args = SwapArgs {
+        deadline:         svm.get_sysvar::<Clock>().unix_timestamp,
+        amount_searcher:  token_searcher.get_amount_with_decimals(1.),
+        amount_user:      token_user.get_amount_with_decimals(1.),
+        referral_fee_bps: 1500,
+        fee_token:        FeeToken::User,
+    };
+
+    let instructions = build_swap_instructions(SwapParams {
+        searcher: searcher.pubkey(),
+        user: user.pubkey(),
+        router_fee_receiver_ta: router_ta_mint_searcher,
+        fee_receiver_relayer: express_relay_metadata.fee_receiver_relayer,
+        token_searcher: token_searcher.clone(),
+        token_user: token_user.clone(),
+        swap_args,
+        overrides: SwapParamOverride {
+            token_program_fee: Some(token_user.token_program), // wrong token program fee
+            ..Default::default()
+        },
+        relayer_signer: relayer_signer.pubkey(),
+    });
+
+    let result = submit_transaction(
+        &mut svm,
+        &instructions,
+        &searcher,
+        &[&searcher, &user, &relayer_signer],
+    )
+    .unwrap_err();
+    assert_custom_error(
+        result.err,
+        4,
+        InstructionError::Custom(AnchorErrorCode::ConstraintTokenMint.into()),
+    );
+}
+
 
 #[test]
 fn test_swap_fail_wrong_relayer_signer() {