Address PR comment

Author: pedronfigueiredo

packages/transaction-pay-controller/src/TransactionPayController.test.ts

@@ -365,6 +365,45 @@ describe('TransactionPayController', () => {
         ),
       ).toBe(TransactionPayStrategy.Test);
     });
+
+    it('passes payment token route args into feature flag fallback', async () => {
+      const controller = createController();
+
+      controller.updatePaymentToken({
+        transactionId: TRANSACTION_ID_MOCK,
+        tokenAddress: TOKEN_ADDRESS_MOCK,
+        chainId: CHAIN_ID_MOCK,
+      });
+
+      const { updateTransactionData } = updatePaymentTokenMock.mock.calls[0][1];
+
+      updateTransactionData(TRANSACTION_ID_MOCK, (data) => {
+        data.paymentToken = {
+          address: TOKEN_ADDRESS_MOCK,
+          balanceFiat: '1',
+          balanceHuman: '1',
+          balanceRaw: '1',
+          balanceUsd: '1',
+          chainId: CHAIN_ID_MOCK,
+          decimals: 6,
+          symbol: 'USDC',
+        };
+      });
+
+      const transactionMeta = {
+        id: TRANSACTION_ID_MOCK,
+        type: 'perpsDeposit',
+      } as TransactionMeta;
+
+      messenger.call('TransactionPayController:getStrategy', transactionMeta);
+
+      expect(getStrategyOrderMock).toHaveBeenCalledWith(
+        messenger,
+        CHAIN_ID_MOCK,
+        TOKEN_ADDRESS_MOCK,
+        'perpsDeposit',
+      );
+    });
   });
 
   describe('transaction data update', () => {

packages/transaction-pay-controller/src/TransactionPayController.ts

@@ -276,6 +276,14 @@ export class TransactionPayController extends BaseController<
       return validStrategies;
     }
 
-    return getStrategyOrder(this.messenger);
+    const paymentToken =
+      this.state.transactionData[transaction.id]?.paymentToken;
+
+    return getStrategyOrder(
+      this.messenger,
+      paymentToken?.chainId,
+      paymentToken?.address,
+      transaction.type,
+    );
   }
 }

packages/transaction-pay-controller/src/index.ts

@@ -29,5 +29,3 @@ export { TransactionPayStrategy } from './constants';
 export { TransactionPayController } from './TransactionPayController';
 export { TransactionPayPublishHook } from './helpers/TransactionPayPublishHook';
 export type { TransactionPayBridgeQuote } from './strategy/bridge/types';
-export type { TransactionPayRouteContext } from './utils/strategy-routing';
-export { getStrategiesForRoute } from './utils/strategy-routing';