more unit tests

valora-inc · Oct 11, 2024 · 928b893 · 928b893
1 parent 253d8c4
commit 928b893
Show file tree

Hide file tree

Showing 2 changed files with 104 additions and 2 deletions.
diff --git a/src/earn/EarnPoolInfoScreen.test.tsx b/src/earn/EarnPoolInfoScreen.test.tsx
@@ -7,7 +7,7 @@ import EarnPoolInfoScreen from 'src/earn/EarnPoolInfoScreen'
 import { CICOFlow } from 'src/fiatExchanges/utils'
 import { navigate } from 'src/navigator/NavigationService'
 import { Screens } from 'src/navigator/Screens'
-import { EarnPosition } from 'src/positions/types'
+import { ClaimType, EarnPosition } from 'src/positions/types'
 import { getFeatureGate, getMultichainFeatures } from 'src/statsig'
 import { StatsigFeatureGates } from 'src/statsig/types'
 import { NetworkId } from 'src/transactions/types'
@@ -757,4 +757,106 @@ describe('EarnPoolInfoScreen', () => {
     })
     // TODO (ACT-1389): Check that navigate called with confirmation screen for Claim
   })
+  it('shows correct copy when ClaimType is Earnings', () => {
+    jest.mocked(getFeatureGate).mockImplementation(
+      (gate) =>
+        //       gate === StatsigFeatureGates.ALLOW_EARN_PARTIAL_WITHDRAWAL ||
+        gate === StatsigFeatureGates.SHOW_POSITIONS
+    )
+    const { getByTestId, getByText } = render(
+      <Provider store={getStore()}>
+        <MockedNavigator
+          component={EarnPoolInfoScreen}
+          params={{
+            pool: {
+              ...mockEarnPositions[0],
+              dataProps: {
+                ...mockEarnPositions[0].dataProps,
+                claimType: ClaimType.Earnings,
+              },
+              balance: '100',
+              availableShortcutIds: ['withdraw', 'deposit', 'claim-rewards'],
+            },
+          }}
+        />
+      </Provider>
+    )
+    fireEvent.press(getByTestId('WithdrawButton'))
+    expect(getByTestId('Earn/WithdrawBottomSheet')).toBeVisible()
+    expect(
+      getByText('earnFlow.poolInfoScreen.withdrawBottomSheet.withdrawDescription')
+    ).toBeTruthy()
+    expect(getByText('earnFlow.poolInfoScreen.withdrawBottomSheet.claimEarnings')).toBeTruthy()
+    expect(
+      getByText('earnFlow.poolInfoScreen.withdrawBottomSheet.claimEarningsDescription')
+    ).toBeTruthy()
+    expect(
+      getByText('earnFlow.poolInfoScreen.withdrawBottomSheet.exitWithEarningsDescription')
+    ).toBeTruthy()
+  })
+  it('shows correct copy when ClaimType is Earnings', () => {
+    jest.mocked(getFeatureGate).mockImplementation(
+      (gate) =>
+        //       gate === StatsigFeatureGates.ALLOW_EARN_PARTIAL_WITHDRAWAL ||
+        gate === StatsigFeatureGates.SHOW_POSITIONS
+    )
+    const { getByTestId, getByText } = render(
+      <Provider store={getStore()}>
+        <MockedNavigator
+          component={EarnPoolInfoScreen}
+          params={{
+            pool: {
+              ...mockEarnPositions[0],
+              dataProps: {
+                ...mockEarnPositions[0].dataProps,
+                claimType: ClaimType.Rewards,
+              },
+              balance: '100',
+              availableShortcutIds: ['withdraw', 'deposit', 'claim-rewards'],
+            },
+          }}
+        />
+      </Provider>
+    )
+    fireEvent.press(getByTestId('WithdrawButton'))
+    expect(getByTestId('Earn/WithdrawBottomSheet')).toBeVisible()
+    expect(getByText('earnFlow.poolInfoScreen.withdrawBottomSheet.claimRewards')).toBeTruthy()
+    expect(
+      getByText('earnFlow.poolInfoScreen.withdrawBottomSheet.claimRewardsDescription')
+    ).toBeTruthy()
+    expect(
+      getByText('earnFlow.poolInfoScreen.withdrawBottomSheet.exitWithRewardsDescription')
+    ).toBeTruthy()
+  })
+  it('shows correct copy when withdrawalIncludesClaim is true', () => {
+    jest.mocked(getFeatureGate).mockImplementation(
+      (gate) =>
+        //       gate === StatsigFeatureGates.ALLOW_EARN_PARTIAL_WITHDRAWAL ||
+        gate === StatsigFeatureGates.SHOW_POSITIONS
+    )
+    const { getByTestId, getByText } = render(
+      <Provider store={getStore()}>
+        <MockedNavigator
+          component={EarnPoolInfoScreen}
+          params={{
+            pool: {
+              ...mockEarnPositions[0],
+              dataProps: {
+                ...mockEarnPositions[0].dataProps,
+                withdrawalIncludesClaim: true,
+              },
+              balance: '100',
+              availableShortcutIds: ['withdraw', 'deposit', 'claim-rewards'],
+            },
+          }}
+        />
+      </Provider>
+    )
+    fireEvent.press(getByTestId('WithdrawButton'))
+    expect(getByTestId('Earn/WithdrawBottomSheet')).toBeVisible()
+    expect(getByText('earnFlow.poolInfoScreen.withdrawBottomSheet.withdrawAndClaim')).toBeTruthy()
+    expect(
+      getByText('earnFlow.poolInfoScreen.withdrawBottomSheet.withdrawAndClaimDescription')
+    ).toBeTruthy()
+  })
 })
diff --git a/src/earn/WithdrawBottomSheet.tsx b/src/earn/WithdrawBottomSheet.tsx
@@ -66,7 +66,7 @@ function ClaimAction({
     details:
       claimType === ClaimType.Rewards
         ? t('earnFlow.poolInfoScreen.withdrawBottomSheet.claimRewardsDescription')
-        : t('earnFlow.poolInfoScreen.withdrawBottomSheet.claimRewardsDescription'),
+        : t('earnFlow.poolInfoScreen.withdrawBottomSheet.claimEarningsDescription'),
     iconComponent: Claim,
     onPress: () => {
       AppAnalytics.track(EarnEvents.earn_select_withdraw_type, { type: 'claim' })