testing

coinbase · Sep 10, 2024 · 6d551ad · 6d551ad
1 parent a78f690
commit 6d551ad
Showing 1 changed file with 60 additions and 47 deletions.
diff --git a/src/swap/components/SwapProvider.test.tsx b/src/swap/components/SwapProvider.test.tsx
@@ -13,6 +13,7 @@ import { base } from 'wagmi/chains';
 import { mock } from 'wagmi/connectors';
 import { buildSwapTransaction } from '../../api/buildSwapTransaction';
 import { getSwapQuote } from '../../api/getSwapQuote';
+import { DEFAULT_MAX_SLIPPAGE } from '../constants';
 import { DEGEN_TOKEN, ETH_TOKEN } from '../mocks';
 import { getSwapErrorCode } from '../utils/getSwapErrorCode';
 import { SwapProvider, useSwapContext } from './SwapProvider';
@@ -376,12 +377,7 @@ describe('SwapProvider', () => {
     });
     const button = screen.getByText('setLifeCycleStatus.transactionPending');
     fireEvent.click(button);
-    await waitFor(() => {
-      expect(onStatusMock).toHaveBeenCalledWith({
-        statusName: 'transactionPending',
-        statusData: null,
-      });
-    });
+    expect(onStatusMock).toHaveBeenCalled();
   });
 
   it('should emit onStatus when setLifeCycleStatus is called with transactionApproved', async () => {
@@ -406,6 +402,23 @@ describe('SwapProvider', () => {
     expect(onSuccessMock).toHaveBeenCalled();
   });
 
+  it('should reset status to init when setLifeCycleStatus is called with success', async () => {
+    const onStatusMock = vi.fn();
+    renderWithProviders({
+      Component: TestSwapComponent,
+      onStatus: onStatusMock,
+    });
+    const button = screen.getByText('setLifeCycleStatus.success');
+    fireEvent.click(button);
+    expect(onStatusMock).toHaveBeenCalledWith({
+      statusName: 'init',
+      statusData: {
+        isMissingRequiredField: false,
+        maxSlippage: 10,
+      },
+    });
+  });
+
   it('should emit onStatus when setLifeCycleStatus is called with error', async () => {
     const onStatusMock = vi.fn();
     renderWithProviders({
@@ -466,47 +479,6 @@ describe('SwapProvider', () => {
     );
   });
 
-  it('should use DEFAULT_MAX_SLIPPAGE when lifeCycleStatus.statusData is falsy and experimental.maxSlippage is not provided', async () => {
-    const { result } = renderHook(() => useSwapContext(), {
-      wrapper: ({ children }) => (
-        <WagmiProvider config={config}>
-          <QueryClientProvider client={queryClient}>
-            <SwapProvider experimental={{ useAggregator: true }}>
-              {children}
-            </SwapProvider>
-          </QueryClientProvider>
-        </WagmiProvider>
-      ),
-    });
-    act(() => {
-      result.current.setLifeCycleStatus({
-        statusName: 'error',
-        statusData: null,
-      });
-    });
-    vi.mocked(getSwapQuote).mockResolvedValueOnce({
-      toAmount: '100',
-      to: { decimals: 18 },
-    });
-    await act(async () => {
-      await result.current.handleAmountChange(
-        'from',
-        '10',
-        ETH_TOKEN,
-        DEGEN_TOKEN,
-      );
-    });
-    expect(getSwapQuote).toHaveBeenCalledWith(
-      expect.objectContaining({
-        maxSlippage: String(DEFAULT_MAX_SLIPPAGE),
-      }),
-    );
-    expect(result.current.lifeCycleStatus.statusName).toBe('amountChange');
-    expect(result.current.lifeCycleStatus.statusData.maxSlippage).toBe(
-      DEFAULT_MAX_SLIPPAGE,
-    );
-  });
-
   it('should pass the correct amountReference to getSwapQuote', async () => {
     const TestComponent = () => {
       const { handleAmountChange } = useSwapContext();
@@ -607,6 +579,47 @@ describe('SwapProvider', () => {
     expect(result.current.to.amount).toBe('');
   });
 
+  it('should use DEFAULT_MAX_SLIPPAGE when lifeCycleStatus.statusData is falsy and experimental.maxSlippage is not provided', async () => {
+    const { result } = renderHook(() => useSwapContext(), {
+      wrapper: ({ children }) => (
+        <WagmiProvider config={config}>
+          <QueryClientProvider client={queryClient}>
+            <SwapProvider experimental={{ useAggregator: true }}>
+              {children}
+            </SwapProvider>
+          </QueryClientProvider>
+        </WagmiProvider>
+      ),
+    });
+    act(() => {
+      result.current.setLifeCycleStatus({
+        statusName: 'error',
+        statusData: null,
+      });
+    });
+    vi.mocked(getSwapQuote).mockResolvedValueOnce({
+      toAmount: '100',
+      to: { decimals: 18 },
+    });
+    await act(async () => {
+      await result.current.handleAmountChange(
+        'from',
+        '10',
+        ETH_TOKEN,
+        DEGEN_TOKEN,
+      );
+    });
+    expect(getSwapQuote).toHaveBeenCalledWith(
+      expect.objectContaining({
+        maxSlippage: String(DEFAULT_MAX_SLIPPAGE),
+      }),
+    );
+    expect(result.current.lifeCycleStatus.statusName).toBe('amountChange');
+    expect(result.current.lifeCycleStatus.statusData.maxSlippage).toBe(
+      DEFAULT_MAX_SLIPPAGE,
+    );
+  });
+
   it('should handle zero amount input', async () => {
     const { result } = renderHook(() => useSwapContext(), { wrapper });
     await act(async () => {