import {initializeOrg} from 'sentry-test/initializeOrg'; import {makeTestQueryClient} from 'sentry-test/queryClient'; import {renderHook, waitFor} from 'sentry-test/reactTestingLibrary'; import {QueryClientProvider} from 'sentry/utils/queryClient'; import { AlertRuleTriggerType, type MetricRule, } from 'sentry/views/alerts/rules/metric/types'; import {OrganizationContext} from 'sentry/views/organizationContext'; import useFetchThresholdsListData from './useFetchThresholdsListData'; describe('useFetchThresholdsListData', () => { const {organization} = initializeOrg({ organization: { name: 'test-org', slug: 'test-thresholds', features: ['releases-v2'], }, }); const queryClient = makeTestQueryClient(); const Wrapper = (org = organization) => { return function WrappedComponent({children}) { return ( {children} ); }; }; const mockThresholdApis = (data = {}) => { const thresholdMock = MockApiClient.addMockResponse({ url: `/organizations/${organization.slug}/release-thresholds/`, method: 'GET', body: data, }); const alertMock = MockApiClient.addMockResponse({ url: `/organizations/${organization.slug}/alert-rules/`, method: 'GET', body: data, }); return {thresholdMock, alertMock}; }; afterEach(() => { queryClient.clear(); }); it('fetches release thresholds by default', async () => { const thresholds = []; const {thresholdMock, alertMock} = mockThresholdApis(thresholds); const {result} = renderHook(() => useFetchThresholdsListData(), { wrapper: Wrapper(), }); // ensure the async fetch is working expect(result.current.isLoading).toBeTruthy(); await waitFor(() => expect(result.current.isLoading).toBeFalsy()); expect(result.current.data).toEqual(thresholds); expect(thresholdMock).toHaveBeenCalled(); expect(alertMock).not.toHaveBeenCalled(); }); it('fetches activated alerts if the feature is enabled', async () => { const alerts = []; const {thresholdMock, alertMock} = mockThresholdApis(alerts); const {organization: org} = initializeOrg({ organization: { name: 'test-org', slug: 'test-thresholds', features: ['activated-alert-rules'], }, }); const {result} = renderHook(() => useFetchThresholdsListData(), { wrapper: Wrapper(org), }); expect(result.current.isLoading).toBeTruthy(); await waitFor(() => expect(result.current.isLoading).toBeFalsy()); expect(result.current.data).toEqual(alerts); expect(thresholdMock).not.toHaveBeenCalled(); expect(alertMock).toHaveBeenCalled(); }); it('formats actiavted alerts as thresholds', async () => { const alerts: Partial[] = [ { id: '1', dateCreated: '2021-01-01', environment: 'production', projects: ['test-project'], timeWindow: 60, triggers: [ { label: AlertRuleTriggerType.CRITICAL, actions: [], alertThreshold: 100, }, ], }, ]; mockThresholdApis(alerts); const {organization: org} = initializeOrg({ organization: { name: 'test-org', slug: 'test-thresholds', features: ['activated-alert-rules'], }, }); const {result} = renderHook(() => useFetchThresholdsListData(), { wrapper: Wrapper(org), }); await waitFor(() => expect(result.current.isLoading).toBeFalsy()); const expectedResult = [ { date_added: '2021-01-01', environment: { name: 'production', displayName: 'production', }, id: '1', project: { id: 'test-project', slug: 'test-project', }, threshold_type: 'total_error_count', trigger_type: 'over', value: 100, window_in_seconds: 60, }, ]; expect(result.current.data).toEqual(expectedResult); }); });