import {
  _resetAuthenticatorForTest,
  _setAuthenticatorForTest,
} from "../../server.ts";

export type AuthStub = { restore: () => void };

export const stubAuthenticate = (didOrError: string | Error): AuthStub => {
  _setAuthenticatorForTest(() => {
    if (didOrError instanceof Error) {
      return Promise.reject(didOrError);
    }
    return Promise.resolve(didOrError);
  });
  return {
    restore: () => _resetAuthenticatorForTest(),
  };
};