sentry/static/app/components/events/autofix/autofixBanner.spec.tsx

import {
  render,
  renderGlobalModal,
  screen,
  userEvent,
} from 'sentry-test/reactTestingLibrary';

import {AutofixBanner} from './autofixBanner';

function mockIsSentryEmployee(isEmployee: boolean) {
  jest
    .spyOn(require('sentry/utils/useIsSentryEmployee'), 'useIsSentryEmployee')
    .mockImplementation(() => isEmployee);
}

describe('AutofixBanner', () => {
  afterEach(() => {
    jest.resetAllMocks();
  });

  const defaultProps = {
    groupId: '1',
    hasSuccessfulSetup: true,
    triggerAutofix: jest.fn(),
  };

  it('shows PII check for sentry employee users', () => {
    mockIsSentryEmployee(true);

    render(<AutofixBanner {...defaultProps} projectId="1" />);
    expect(
      screen.getByText(
        'By clicking the button above, you confirm that there is no PII in this event.'
      )
    ).toBeInTheDocument();
  });

  it('does not show PII check for non sentry employee users', () => {
    mockIsSentryEmployee(false);

    render(<AutofixBanner {...defaultProps} projectId="1" />);
    expect(
      screen.queryByText(
        'By clicking the button above, you confirm that there is no PII in this event.'
      )
    ).not.toBeInTheDocument();
  });

  it('can run without instructions', async () => {
    const mockTriggerAutofix = jest.fn();

    render(
      <AutofixBanner
        {...defaultProps}
        triggerAutofix={mockTriggerAutofix}
        projectId="1"
      />
    );
    renderGlobalModal();

    await userEvent.click(screen.getByRole('button', {name: 'Get root causes'}));
    expect(mockTriggerAutofix).toHaveBeenCalledWith('');
  });

  it('can provide instructions', async () => {
    const mockTriggerAutofix = jest.fn();

    render(
      <AutofixBanner
        {...defaultProps}
        triggerAutofix={mockTriggerAutofix}
        projectId="1"
      />
    );
    renderGlobalModal();

    await userEvent.click(screen.getByRole('button', {name: 'Provide context first'}));
    await userEvent.type(screen.getByRole('textbox'), 'instruction!');
    await userEvent.click(screen.getByRole('button', {name: "Let's go!"}));

    expect(mockTriggerAutofix).toHaveBeenCalledWith('instruction!');
  });
});