test: add testing case

MetaMask · Aug 21, 2024 · ab5ba5f · ab5ba5f
1 parent 9774dab
commit ab5ba5f
Showing 1 changed file with 54 additions and 0 deletions.
diff --git a/packages/transaction-controller/src/utils/swaps.test.ts b/packages/transaction-controller/src/utils/swaps.test.ts
@@ -375,6 +375,60 @@ describe('updateSwapsTransaction', () => {
       swapAndSendRecipient,
     });
   });
+
+  it('should update swap and send approval transaction and publish TransactionController:transactionNewSwapAndSendApproval', async () => {
+    const sourceTokenSymbol = 'ETH';
+    const type = TransactionType.swapAndSendApproval;
+
+    swaps.meta = {
+      sourceTokenSymbol,
+      type,
+    };
+    transactionType = TransactionType.swapAndSendApproval;
+
+    const transactionNewSwapAndSendApprovalEventListener = jest.fn();
+    messenger.subscribe(
+      'TransactionController:transactionNewSwapAndSendApproval',
+      transactionNewSwapAndSendApprovalEventListener,
+    );
+
+    updateSwapsTransaction(transactionMeta, transactionType, swaps, request);
+    expect(
+      transactionNewSwapAndSendApprovalEventListener,
+    ).toHaveBeenCalledTimes(1);
+    expect(transactionNewSwapAndSendApprovalEventListener).toHaveBeenCalledWith(
+      {
+        transactionMeta: {
+          ...transactionMeta,
+          sourceTokenSymbol,
+          type,
+        },
+      },
+    );
+  });
+
+  it('should return the swap and send approval transaction updated with information', async () => {
+    const sourceTokenSymbol = 'ETH';
+    const type = TransactionType.swapAndSendApproval;
+
+    swaps.meta = {
+      sourceTokenSymbol,
+      type,
+    };
+    transactionType = TransactionType.swapAndSendApproval;
+
+    const updatedSwapsTransaction = updateSwapsTransaction(
+      transactionMeta,
+      transactionType,
+      swaps,
+      request,
+    );
+    expect(updatedSwapsTransaction).toStrictEqual({
+      ...transactionMeta,
+      sourceTokenSymbol,
+      type,
+    });
+  });
 });
 
 describe('updatePostTransactionBalance', () => {