accounts-frontend/packages/app/services/authFlow/helpers.ts

/**
 * A helpers for testing states in isolation from AuthFlow
 */

import sinon, { SinonExpectation, SinonMock, SinonStub } from 'sinon';
import AbstractState from './AbstractState';
import { AuthContext } from './AuthFlow';

export interface MockedAuthContext extends AuthContext {
  getState: SinonStub;
  getRequest: SinonStub;
}

export function bootstrap(): { context: MockedAuthContext; mock: SinonMock } {
  const context: MockedAuthContext = {
    getState: sinon.stub(),
    run() {
      return Promise.resolve();
    },
    setState() {},
    getRequest: sinon.stub(),
    navigate() {},
    prevState: new (class State extends AbstractState {})(),
  };

  const mock = sinon.mock(context);
  mock.expects('run').never();
  mock.expects('navigate').never();
  mock.expects('setState').never();

  return { context, mock };
}

export function expectState(
  mock: SinonMock,
  state: typeof AbstractState,
): SinonExpectation {
  return mock
    .expects('setState')
    .once()
    .withExactArgs(sinon.match.instanceOf(state));
}

export function expectNavigate(
  mock: SinonMock,
  route: string,
  options: Record<string, any> | void,
): SinonExpectation {
  if (options) {
    return mock
      .expects('navigate')
      .once()
      .withExactArgs(route, sinon.match(options));
  }

  return mock
    .expects('navigate')
    .once()
    .withExactArgs(route);
}

export function expectRun(
  mock: SinonMock,
  ...args: Array<any>
): SinonExpectation {
  return mock
    .expects('run')
    .once()
    .withExactArgs(...args);
}