fix: poll checkout intent API within billing details

src/components/StatefulButton/StatefulSubscribeButton.tsx

@@ -8,8 +8,8 @@ import { useQueryClient } from '@tanstack/react-query';
 import { useContext, useEffect, useState } from 'react';
 import { useNavigate } from 'react-router-dom';
 
-import { useCheckoutIntent } from '@/components/app/data';
-import { queryBffContext } from '@/components/app/data/queries/queries';
+import { useCheckoutIntent, usePolledCheckoutIntent } from '@/components/app/data';
+import { determineExistingSuccessfulCheckoutIntent } from '@/components/app/data/services/context';
 import { CheckoutPageRoute, CheckoutSubstepKey, DataStoreKey } from '@/constants/checkout';
 import EVENT_NAMES from '@/constants/events';
 import { useCheckoutFormStore } from '@/hooks/useCheckoutFormStore';
@@ -57,7 +57,7 @@ const StatefulSubscribeButton = () => {
   const [errorMessageKey, setErrorMessageKey] = useState('fallback');
   const { data: checkoutIntent } = useCheckoutIntent();
   const intl = useIntl();
-
+  const { data: polledCheckoutIntent } = usePolledCheckoutIntent();
   const queryClient = useQueryClient();
   const navigate = useNavigate();
   const { authenticatedUser }: AppContextValue = useContext(AppContext);
@@ -92,7 +92,6 @@ const StatefulSubscribeButton = () => {
     }
     // Set the button to the appropriate state based on the response.
     // Stripe responses map 1:1 to button states except for 'default' which is the initial state.
-    setStatefulButtonState(response.type || 'default');
     if (response.type === 'error') {
       setErrorMessageKey(buttonMessages.error[response.error?.code] ? response.error?.code : 'fallback');
       logError(
@@ -101,18 +100,18 @@ const StatefulSubscribeButton = () => {
     }
   };
 
+  // Refetch the checkout intent if the polled intent state changes.
+  useEffect(() => {
+    if (statefulButtonState === 'pending' && determineExistingSuccessfulCheckoutIntent(polledCheckoutIntent?.state)) {
+      setStatefulButtonState('success');
+    }
+  }, [polledCheckoutIntent?.state, navigate, statefulButtonState]);
+
   useEffect(() => {
     if (statefulButtonState === 'success') {
-      // If the payment succeeded, update the checkout session status.
+      // If the payment succeeded from the stripe API, update the checkout session status.
       if (status.type === 'complete' && status.paymentStatus === 'paid') {
         setCheckoutSessionStatus(status);
-        queryClient.invalidateQueries({
-          queryKey: queryBffContext(
-            authenticatedUser.id,
-          ).queryKey,
-        })
-          .then(data => data)
-          .catch(error => logError(error));
         sendEnterpriseCheckoutTrackingEvent({
           checkoutIntentId: checkoutIntent?.id ?? null,
           eventName: EVENT_NAMES.SUBSCRIPTION_CHECKOUT.PAYMENT_PROCESSED_SUCCESSFULLY,

src/components/StatefulButton/tests/StatefulSubscribeButton.test.tsx

@@ -42,6 +42,7 @@ jest.mock('react-router-dom', () => ({
 
 jest.mock('@/components/app/data', () => ({
   useCheckoutIntent: jest.fn(() => ({ data: { id: 'test-intent' } })),
+  usePolledCheckoutIntent: jest.fn(() => ({ data: { state: 'paid' } })),
 }));
 
 jest.mock('@/hooks/useCheckoutFormStore', () => ({
@@ -333,7 +334,6 @@ describe('StatefulSubscribeButton', () => {
       // Wait for the useEffect to trigger after state changes to 'success'
       await waitFor(() => {
         expect(mockSetCheckoutSessionStatus).toHaveBeenCalledWith({ type: 'complete', paymentStatus: 'paid' });
-        expect(mockInvalidateQueries).toHaveBeenCalled();
         expect(jest.mocked(sendEnterpriseCheckoutTrackingEvent)).toHaveBeenCalledWith({
           checkoutIntentId: 'test-intent',
           eventName: EVENT_NAMES.SUBSCRIPTION_CHECKOUT.PAYMENT_PROCESSED_SUCCESSFULLY,

src/components/app/data/services/context.ts

@@ -80,5 +80,6 @@ const fetchCheckoutSuccess = async (): Promise<CheckoutContextResponse> => {
 export {
   fetchCheckoutContext,
   fetchCheckoutSuccess,
+  determineExistingSuccessfulCheckoutIntent,
   paymentProcessedCheckoutIntentStates,
 };

src/components/app/routes/loaders/checkoutStepperLoader.ts

@@ -2,7 +2,8 @@ import { getAuthenticatedUser } from '@edx/frontend-platform/auth';
 import { QueryClient } from '@tanstack/react-query';
 import { redirect } from 'react-router-dom';
 
-import { queryBffContext } from '@/components/app/data/queries/queries';
+import { queryBffContext, queryBffSuccess, queryCheckoutIntent } from '@/components/app/data/queries/queries';
+import { determineExistingSuccessfulCheckoutIntent } from '@/components/app/data/services/context';
 import { getCheckoutSessionClientSecret, validateFormState } from '@/components/app/routes/loaders/utils';
 import { CheckoutPageRoute } from '@/constants/checkout';
 import { checkoutFormStore } from '@/hooks/useCheckoutFormStore';
@@ -144,12 +145,32 @@ async function billingDetailsSuccessLoader(queryClient: QueryClient): Promise<Re
   const contextMetadata: CheckoutContextResponse = await queryClient.ensureQueryData(
     queryBffContext(authenticatedUser?.userId || null),
   );
-
   const { checkoutIntent } = contextMetadata;
 
+  if (!checkoutIntent) {
+    return redirect(CheckoutPageRoute.PlanDetails);
+  }
+
+  await queryClient.ensureQueryData(
+    queryBffSuccess(authenticatedUser?.userId || null),
+  );
+  if (checkoutIntent.id) {
+    await queryClient.ensureQueryData(
+      queryCheckoutIntent(checkoutIntent.id),
+    );
+  }
+
   const checkoutIntentType = checkoutFormStore.getState().checkoutSessionStatus?.type;
 
-  if (checkoutIntentType !== 'complete' && !checkoutIntent?.existingSuccessfulCheckoutIntent) {
+  // Explicitly check that the intent is in a successful state
+  // If the intent is successful but the type is not 'complete',
+  // or if there is no existingSuccessfulCheckoutIntent flag,
+  // redirect to Plan Details to restart the process.
+  if (
+    !determineExistingSuccessfulCheckoutIntent(checkoutIntent.state)
+    && checkoutIntentType !== 'complete'
+    && !checkoutIntent.existingSuccessfulCheckoutIntent
+  ) {
     return redirect(CheckoutPageRoute.PlanDetails);
   }
 

src/components/billing-details-pages/tests/BillingDetailsPage.test.tsx

@@ -45,6 +45,12 @@ describe('BillingDetailsPage', () => {
       },
     });
     (useCheckoutSessionClientSecret as jest.Mock).mockReturnValue('secret-123');
+    (usePolledCheckoutIntent as jest.Mock).mockReturnValue({
+      data: {
+        id: 1,
+        state: 'pending',
+      },
+    });
     (useCheckout as jest.Mock).mockReturnValue({
       canConfirm: true,
       confirm: jest.fn().mockResolvedValue({ type: 'success' }),