import AudienceHashMap, {
  AudienceScores,
} from "../../../../src/services/scoringSystem/audienceHashMap.service";

export default class AudienceHashMapFake extends AudienceHashMap {
  static fromAudienceScores(audienceScores: AudienceScores) {
    const audience = new AudienceHashMapFake();
    audience.audienceScores = audienceScores;
    return audience;
  }

  setS3Prefix(s3Prefix: string) {
    this.s3Prefix = s3Prefix;
  }

  setMaxRetries(maxRetries: number) {
    this.maxRetries = maxRetries;
  }

  async runRefresh() {
    return this.refresh();
  }

  getAudienceScores() {
    return this.audienceScores;
  }
}