Add a new "userOptInV2" prop (#222)

MetaMask · Nov 3, 2023 · b21c8bf · b21c8bf
1 parent 697941a
commit b21c8bf
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/src/SmartTransactionsController.test.ts b/src/SmartTransactionsController.test.ts
@@ -287,6 +287,7 @@ const defaultState = {
       [CHAIN_IDS.ETHEREUM]: [],
     },
     userOptIn: undefined,
+    userOptInV2: undefined,
     fees: {
       approvalTxFees: undefined,
       tradeTxFees: undefined,
@@ -496,15 +497,15 @@ describe('SmartTransactionsController', () => {
     it('sets optIn state', () => {
       smartTransactionsController.setOptInState(true);
       expect(
-        smartTransactionsController.state.smartTransactionsState.userOptIn,
+        smartTransactionsController.state.smartTransactionsState.userOptInV2,
       ).toBe(true);
       smartTransactionsController.setOptInState(false);
       expect(
-        smartTransactionsController.state.smartTransactionsState.userOptIn,
+        smartTransactionsController.state.smartTransactionsState.userOptInV2,
       ).toBe(false);
       smartTransactionsController.setOptInState(undefined);
       expect(
-        smartTransactionsController.state.smartTransactionsState.userOptIn,
+        smartTransactionsController.state.smartTransactionsState.userOptInV2,
       ).toBeUndefined();
     });
   });
@@ -611,6 +612,7 @@ describe('SmartTransactionsController', () => {
             [CHAIN_IDS.ETHEREUM]: [pendingTransaction],
           },
           userOptIn: undefined,
+          userOptInV2: undefined,
           fees: {
             approvalTxFees: undefined,
             tradeTxFees: undefined,
@@ -666,6 +668,7 @@ describe('SmartTransactionsController', () => {
             ],
           },
           userOptIn: undefined,
+          userOptInV2: undefined,
           fees: {
             approvalTxFees: undefined,
             tradeTxFees: undefined,

diff --git a/src/SmartTransactionsController.ts b/src/SmartTransactionsController.ts
@@ -55,6 +55,7 @@ export type SmartTransactionsControllerState = BaseState & {
   smartTransactionsState: {
     smartTransactions: Record<Hex, SmartTransaction[]>;
     userOptIn: boolean | undefined;
+    userOptInV2: boolean | undefined;
     liveness: boolean | undefined;
     fees: FeeEstimates;
     feesByChainId: Record<Hex, FeeEstimates>;
@@ -126,6 +127,7 @@ export default class SmartTransactionsController extends PollingControllerV1<
       smartTransactionsState: {
         smartTransactions: {},
         userOptIn: undefined,
+        userOptInV2: undefined,
         fees: {
           approvalTxFees: undefined,
           tradeTxFees: undefined,
@@ -232,7 +234,7 @@ export default class SmartTransactionsController extends PollingControllerV1<
     this.update({
       smartTransactionsState: {
         ...this.state.smartTransactionsState,
-        userOptIn: state,
+        userOptInV2: state,
       },
     });
   }