sentry/static/app/views/replays/detail/trace/useReplayTraces.spec.tsx

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

import {makeTestQueryClient} from 'sentry-test/queryClient';
import {renderHook, waitFor} from 'sentry-test/reactTestingLibrary';

import * as useOrganization from 'sentry/utils/useOrganization';

import {useReplayTraces} from './useReplayTraces';

const organization = OrganizationFixture();
const queryClient = makeTestQueryClient();
const replayRecord = ReplayRecordFixture();

describe('useTraceMeta', () => {
  beforeEach(function () {
    queryClient.clear();
    jest.clearAllMocks();
    jest.spyOn(useOrganization, 'default').mockReturnValue(organization);
  });

  it('Returns replay traces', async () => {
    const pageLinks =
      '<https://sentry.io/fake/previous>; rel="previous"; results="false"; cursor="0:0:1", ' +
      '<https://sentry.io/fake/next>; rel="next"; results="false"; cursor="0:20:0"';

    // Mock the API calls
    MockApiClient.addMockResponse({
      method: 'GET',
      headers: {Link: pageLinks},
      url: '/organizations/org-slug/events/',
      body: {
        data: [
          {
            trace: 'trace1',
            'min(timestamp)': 1,
          },
          {
            trace: 'trace2',
            'min(timestamp)': 2,
          },
        ],
      },
    });

    const {result} = renderHook(() => useReplayTraces({replayRecord}));

    expect(result.current.indexComplete).toEqual(false);

    await waitFor(() => expect(result.current.indexComplete).toBe(true));

    expect(result.current.indexComplete).toEqual(true);
    expect(result.current.replayTraces).toEqual([
      {traceSlug: 'trace1', timestamp: 1},
      {traceSlug: 'trace2', timestamp: 2},
    ]);
  });

  it('Collects errors', async () => {
    const pageLinks =
      '<https://sentry.io/fake/previous>; rel="previous"; results="false"; cursor="0:0:1", ' +
      '<https://sentry.io/fake/next>; rel="next"; results="false"; cursor="0:20:0"';

    // Mock the API calls
    const mockedResponse = MockApiClient.addMockResponse({
      method: 'GET',
      headers: {Link: pageLinks},
      url: '/organizations/org-slug/events/',
      statusCode: 400,
    });

    const {result} = renderHook(() => useReplayTraces({replayRecord}));

    expect(result.current.indexComplete).toEqual(false);

    await waitFor(() => expect(result.current.indexComplete).toBe(true));

    expect(result.current.indexComplete).toEqual(true);
    expect(result.current.replayTraces).toEqual(undefined);
    expect(result.current.indexError as Error).toEqual(
      expect.objectContaining({status: 400})
    );
    expect(mockedResponse).toHaveBeenCalledTimes(1);
  });
});