insomnia/packages/insomnia-app/app/test-utils.js

// @flow

import * as modals from './ui/components/modals';
import type { ErrorModalOptions } from './ui/components/modals/error-modal';

export const getAndClearShowPromptMockArgs = () => {
  const mockFn = (modals.showPrompt: JestMockFn);
  const { title, submitName, defaultValue, onComplete } = mockFn.mock.calls[0][0];
  mockFn.mockClear();

  return { title, submitName, defaultValue, onComplete };
};

export const getAndClearShowAlertMockArgs = () => {
  const mockFn = (modals.showAlert: JestMockFn);
  const { title, okLabel, addCancel, message, onConfirm } = mockFn.mock.calls[0][0];
  mockFn.mockClear();

  return { title, okLabel, addCancel, message, onConfirm };
};

export const getAndClearShowErrorMockArgs = (): ErrorModalOptions => {
  const mockFn = (modals.showError: JestMockFn);
  const options: ErrorModalOptions = mockFn.mock.calls[0][0];
  mockFn.mockClear();

  return options;
};

export const getAndClearShowModalMockArgs = () => {
  const mockFn = (modals.showModal: JestMockFn);
  const args = mockFn.mock.calls[0][1];
  mockFn.mockClear();

  return args;
};