Add Failure Flow Massa Approve Test

massalabs · Nov 20, 2023 · 5d61dfb · 5d61dfb
1 parent 1681b38
commit 5d61dfb
Show file tree

Hide file tree

Showing 2 changed files with 52 additions and 4 deletions.
diff --git a/jest/__tests__/handleApproveMassa.test.ts b/jest/__tests__/handleApproveMassa.test.ts
@@ -37,7 +37,7 @@ describe('handleApproveMASSA', () => {
     const decimals = 18;
 
     const mockSetLoading = jest.fn().mockImplementation();
-    const handleErrorMessage = jest.fn();
+    // const handleErrorMessage = jest.fn();
 
     const result = await handleApproveMASSA(
       client as any,
@@ -56,7 +56,7 @@ describe('handleApproveMASSA', () => {
 
     expect(mockSetLoading).toHaveBeenNthCalledWith(2, { approve: 'success' });
 
-    expect(handleErrorMessage).not.toHaveBeenCalled();
+    // expect(handleErrorMessage).not.toHaveBeenCalled();
     expect(result).toBeTruthy();
   });
 
@@ -79,7 +79,7 @@ describe('handleApproveMASSA', () => {
     const amount = '13';
     const decimals = 18;
     const mockSetLoading = jest.fn().mockImplementation();
-    const handleErrorMessage = jest.fn();
+    // const handleErrorMessage = jest.fn();
     const result = await handleApproveMASSA(
       client as any,
       mockSetLoading,
@@ -97,7 +97,54 @@ describe('handleApproveMASSA', () => {
 
     expect(mockSetLoading).toHaveBeenNthCalledWith(2, { approve: 'success' });
 
-    expect(handleErrorMessage).not.toHaveBeenCalled();
+    // expect(handleErrorMessage).not.toHaveBeenCalled();
     expect(result).toBeTruthy();
   });
+
+  test('should throw', async () => {
+    const mockCallSmartContract = jest
+      .fn()
+      .mockImplementationOnce(() => Promise.reject(new Error('error')));
+
+    const mockGetOperationStatus = jest
+      .fn()
+      .mockResolvedValueOnce(EOperationStatus.FINAL_SUCCESS);
+
+    const client = {
+      smartContracts: () => ({
+        callSmartContract: mockCallSmartContract,
+        getOperationStatus: mockGetOperationStatus,
+      }),
+    };
+
+    const amount = '1313';
+    const decimals = 18;
+    const mockSetLoading = jest.fn().mockImplementation();
+    // const handleErrorMessage = jest.fn().mockImplementation();
+    const result = await handleApproveMASSA(
+      client as any,
+      mockSetLoading,
+      {} as any,
+      {} as any,
+      token,
+      amount,
+      decimals,
+    );
+
+    expect(mockSetLoading).toHaveBeenNthCalledWith(1, { approve: 'loading' });
+
+    expect(mockCallSmartContract).toHaveBeenCalled();
+    expect(mockGetOperationStatus).not.toHaveBeenCalled();
+
+    expect(mockSetLoading).toHaveBeenNthCalledWith(2, {
+      box: 'error',
+      approve: 'error',
+      burn: 'error',
+      redeem: 'error',
+    });
+
+    // expect(handleErrorMessage).toHaveBeenCalled();
+
+    expect(result).toBeFalsy();
+  });
 });
diff --git a/src/custom/bridge/handlers/handleErrorMessage.tsx b/src/custom/bridge/handlers/handleErrorMessage.tsx
@@ -30,6 +30,7 @@ export function handleErrorMessage(
   const cause = (error as ICustomError)?.cause;
   const isTimeout = cause?.error === 'timeout';
 
+  // bridge side this fn only show timeout error
   if (isTimeout) {
     setLoading({
       box: 'warning',