test: update integrated swap tests to select a different token

VenusProtocol · Mar 13, 2024 · 2a0d2e1 · 2a0d2e1
1 parent dd23201
commit 2a0d2e1
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 5 deletions.
diff --git a/apps/evm/src/hooks/useOperationModal/Modal/RepayForm/__tests__/indexIntegratedSwap.spec.tsx b/apps/evm/src/hooks/useOperationModal/Modal/RepayForm/__tests__/indexIntegratedSwap.spec.tsx
@@ -214,13 +214,19 @@ describe('RepayForm - Feature flag enabled: integratedSwap', () => {
       isLoading: false,
     }));
 
-    const { getByTestId, getByText } = renderComponent(
+    const { getByTestId, getByText, container } = renderComponent(
       <Repay asset={fakeAsset} pool={fakePool} onCloseModal={noop} />,
       {
         accountAddress: fakeAccountAddress,
       },
     );
 
+    selectToken({
+      container,
+      selectTokenTextFieldTestId: TEST_IDS.selectTokenTextField,
+      token: busd,
+    });
+
     const selectTokenTextField = getByTestId(
       getTokenTextFieldTestId({
         parentTestId: TEST_IDS.selectTokenTextField,

diff --git a/apps/evm/src/hooks/useOperationModal/Modal/SupplyForm/__tests__/indexIntegratedSwap.spec.tsx b/apps/evm/src/hooks/useOperationModal/Modal/SupplyForm/__tests__/indexIntegratedSwap.spec.tsx
@@ -60,9 +60,10 @@ vi.mock('hooks/useGetSwapRouterContractAddress');
 
 describe('hooks/useSupplyWithdrawModal/Supply - Feature flag enabled: integratedSwap', () => {
   beforeEach(() => {
-    (useIsFeatureEnabled as Vi.Mock).mockImplementation(
-      ({ name }: UseIsFeatureEnabled) => name === 'integratedSwap',
-    );
+    (useIsFeatureEnabled as Vi.Mock).mockImplementation(({ name }: UseIsFeatureEnabled) => {
+      console.log('powerslave222', name);
+      return name === 'integratedSwap';
+    });
 
     (useGetSwapInfo as Vi.Mock).mockImplementation(() => ({
       swap: undefined,
@@ -192,13 +193,19 @@ describe('hooks/useSupplyWithdrawModal/Supply - Feature flag enabled: integrated
       isLoading: false,
     }));
 
-    const { getByTestId, getByText } = renderComponent(
+    const { getByTestId, getByText, container } = renderComponent(
       <Supply asset={fakeAsset} pool={fakePool} onCloseModal={noop} />,
       {
         accountAddress: fakeAccountAddress,
       },
     );
 
+    selectToken({
+      container,
+      selectTokenTextFieldTestId: TEST_IDS.selectTokenTextField,
+      token: busd,
+    });
+
     const selectTokenTextField = getByTestId(
       getTokenTextFieldTestId({
         parentTestId: TEST_IDS.selectTokenTextField,