privacy.sexy/tests/unit/shared/Stubs/EnumParserStub.ts

import { IEnumParser } from '@/application/Common/Enum';

export class EnumParserStub<T> implements IEnumParser<T> {
  private readonly scenarios = new Array<{
    inputName: string, inputValue: string, outputValue: T }>();

  private defaultValue: T;

  public setup(inputName: string, inputValue: string, outputValue: T) {
    this.scenarios.push({ inputName, inputValue, outputValue });
    return this;
  }

  public setupDefaultValue(outputValue: T) {
    this.defaultValue = outputValue;
    return this;
  }

  public parseEnum(value: string, propertyName: string): T {
    const scenario = this.scenarios.find(
      (s) => s.inputName === propertyName && s.inputValue === value,
    );
    if (scenario) {
      return scenario.outputValue;
    }
    if (this.defaultValue) {
      return this.defaultValue;
    }
    throw new Error('enum parser is not set up');
  }
}