sentry/static/gsApp/components/upgradeNowModal/usePreviewData.spec.tsx

import {OrganizationFixture} from 'sentry-fixture/organization';

import {BillingConfigFixture} from 'getsentry-test/fixtures/billingConfig';
import {PlanDetailsLookupFixture} from 'getsentry-test/fixtures/planDetailsLookup';
import {SubscriptionFixture} from 'getsentry-test/fixtures/subscription';
import {makeTestQueryClient} from 'sentry-test/queryClient';
import {renderHook, waitFor} from 'sentry-test/reactTestingLibrary';

import {QueryClientProvider} from 'sentry/utils/queryClient';

import {PreviewDataFixture} from 'getsentry/__fixtures__/previewData';
import {PlanTier} from 'getsentry/types';

import type {Reservations} from './types';
import usePreviewData from './usePreviewData';

const mockReservations: Reservations = {
  reservedErrors: 50000,
  reservedReplays: 500,
  reservedAttachments: 1,
  reservedMonitorSeats: 1,
  reservedTransactions: 100000,
  reservedUptime: 1,
};

const mockPreview = PreviewDataFixture({});

describe('usePreviewData', () => {
  const organization = OrganizationFixture({
    access: ['org:billing'],
  });

  it('should fetch a preview of a new AM2 plan, based on the current reservations', async () => {
    const teamPlan = PlanDetailsLookupFixture('am2_team');
    const subscription = SubscriptionFixture({
      organization,
      plan: 'am2_team',
    });
    const mockAM2BillingConfig = BillingConfigFixture(PlanTier.AM2);

    MockApiClient.addMockResponse({
      url: `/customers/${organization.slug}/billing-config/`,
      method: 'GET',
      body: mockAM2BillingConfig,
    });

    const previewRequest = MockApiClient.addMockResponse({
      url: `/customers/${organization.slug}/subscription/preview/`,
      method: 'GET',
      body: mockPreview,
    });

    const {result} = renderHook(usePreviewData, {
      initialProps: {
        organization,
        subscription,
      },
      wrapper: ({children}) => (
        <QueryClientProvider client={makeTestQueryClient()}>
          {children}
        </QueryClientProvider>
      ),
    });

    await waitFor(() =>
      expect(result.current).toStrictEqual({
        loading: false,
        error: false,
        plan: teamPlan,
        reservations: mockReservations,
        previewData: mockPreview,
      })
    );
    expect(previewRequest).toHaveBeenCalledTimes(1);
  });
});