Hide FreeTrialSection if there is no workspace

Author: adamgrzybowski

src/pages/home/FreeTrialSection/useFreeTrial.ts

@@ -1,7 +1,9 @@
 import {useEffect, useState} from 'react';
+import useCurrentUserPersonalDetails from '@hooks/useCurrentUserPersonalDetails';
 import useHasTeam2025Pricing from '@hooks/useHasTeam2025Pricing';
 import useOnyx from '@hooks/useOnyx';
 import useSubscriptionPlan from '@hooks/useSubscriptionPlan';
+import {getOwnedPaidPolicies} from '@libs/PolicyUtils';
 import type {DiscountInfo} from '@libs/SubscriptionUtils';
 import {calculateRemainingFreeTrialDays, doesUserHavePaymentCardAdded, getEarlyDiscountInfo, isUserOnFreeTrial, shouldShowDiscountBanner} from '@libs/SubscriptionUtils';
 import CONST from '@src/CONST';
@@ -32,12 +34,14 @@ function useFreeTrial(): FreeTrialState {
     const [firstDayFreeTrial] = useOnyx(ONYXKEYS.NVP_FIRST_DAY_FREE_TRIAL);
     const [lastDayFreeTrial] = useOnyx(ONYXKEYS.NVP_LAST_DAY_FREE_TRIAL);
     const [userBillingFundID] = useOnyx(ONYXKEYS.NVP_BILLING_FUND_ID);
+    const {accountID} = useCurrentUserPersonalDetails();
     const hasTeam2025Pricing = useHasTeam2025Pricing();
     const subscriptionPlan = useSubscriptionPlan();
     const [allPolicies] = useOnyx(ONYXKEYS.COLLECTION.POLICY);
 
     const onFreeTrial = isUserOnFreeTrial(firstDayFreeTrial, lastDayFreeTrial);
     const hasPaymentCard = doesUserHavePaymentCardAdded(userBillingFundID);
+    const hasOwnedPaidPolicies = getOwnedPaidPolicies(allPolicies, accountID).length > 0;
     const showDiscount = shouldShowDiscountBanner(hasTeam2025Pricing, subscriptionPlan, firstDayFreeTrial, lastDayFreeTrial, userBillingFundID, allPolicies);
     const daysLeft = calculateRemainingFreeTrialDays(lastDayFreeTrial);
 
@@ -58,7 +62,7 @@ function useFreeTrial(): FreeTrialState {
         };
     }, [firstDayFreeTrial, showDiscount]);
 
-    if (!onFreeTrial || hasPaymentCard) {
+    if (!onFreeTrial || hasPaymentCard || !hasOwnedPaidPolicies) {
         return {shouldShowFreeTrialSection: false, discountType: null, daysLeft: 0, discountInfo: null};
     }
 

tests/unit/hooks/useFreeTrial.test.ts

@@ -3,6 +3,7 @@ import {renderHook} from '@testing-library/react-native';
 import Onyx from 'react-native-onyx';
 import useHasTeam2025Pricing from '@hooks/useHasTeam2025Pricing';
 import useSubscriptionPlan from '@hooks/useSubscriptionPlan';
+import {getOwnedPaidPolicies} from '@libs/PolicyUtils';
 import {calculateRemainingFreeTrialDays, doesUserHavePaymentCardAdded, getEarlyDiscountInfo, isUserOnFreeTrial, shouldShowDiscountBanner} from '@libs/SubscriptionUtils';
 import useFreeTrial from '@pages/home/FreeTrialSection/useFreeTrial';
 import ONYXKEYS from '@src/ONYXKEYS';
@@ -18,6 +19,10 @@ jest.mock('@hooks/useSubscriptionPlan', () => ({
     default: jest.fn(() => 'corporate'),
 }));
 
+jest.mock('@libs/PolicyUtils', () => ({
+    getOwnedPaidPolicies: jest.fn(() => [{id: 'policyID'}]),
+}));
+
 jest.mock('@libs/SubscriptionUtils', () => ({
     shouldShowDiscountBanner: jest.fn(() => false),
     getEarlyDiscountInfo: jest.fn(() => null),
@@ -28,6 +33,7 @@ jest.mock('@libs/SubscriptionUtils', () => ({
 
 const mockedUseHasTeam2025Pricing = useHasTeam2025Pricing as jest.Mock;
 const mockedUseSubscriptionPlan = useSubscriptionPlan as jest.Mock;
+const mockedGetOwnedPaidPolicies = getOwnedPaidPolicies as jest.Mock;
 const mockedShouldShowDiscountBanner = shouldShowDiscountBanner as jest.Mock;
 const mockedGetEarlyDiscountInfo = getEarlyDiscountInfo as jest.Mock;
 const mockedIsUserOnFreeTrial = isUserOnFreeTrial as jest.Mock;
@@ -43,6 +49,7 @@ describe('useFreeTrial', () => {
         await Onyx.clear();
         await waitForBatchedUpdates();
         jest.clearAllMocks();
+        mockedGetOwnedPaidPolicies.mockReturnValue([{id: 'policyID'}]);
     });
 
     afterEach(async () => {
@@ -77,6 +84,16 @@ describe('useFreeTrial', () => {
 
             expect(result.current.shouldShowFreeTrialSection).toBe(true);
         });
+
+        it("should not show section when user doesn't own any paid workspaces", () => {
+            mockedIsUserOnFreeTrial.mockReturnValue(true);
+            mockedDoesUserHavePaymentCardAdded.mockReturnValue(false);
+            mockedGetOwnedPaidPolicies.mockReturnValue([]);
+
+            const {result} = renderHook(() => useFreeTrial());
+
+            expect(result.current.shouldShowFreeTrialSection).toBe(false);
+        });
     });
 
     describe('discount state - 50% off (first 24 hours)', () => {