import { ContextService } from "../../../../src/services/targeting/ContextService";
import { CampaignCreative } from "../../../../src/steps/ordering/ordering.model";

class ContextServiceFake extends ContextService {
  expected: any;

  constructor(expected: any) {
    super({} as any);
    this.expected = expected;
  }

  async addScores(request: any, campaignCreatives: CampaignCreative[]) {
    campaignCreatives.map((campaignCreative) => {
      let score = this.expected[campaignCreative.campaign.id];
      if (score) {
        score = score[campaignCreative.creativeId];
      }

      campaignCreative.metadata.context_score = score && score.score >= 0 ? score.score : null;
      campaignCreative.metadata.acc_rate = campaignCreative.metadata.context_score;
      const bucketsMap = score && score.bucketsMap ? score.bucketsMap : null;
      campaignCreative.metadataTmp.accMetadata = {
        bucketsMap,
      };
    });
    return this.expected;
  }
}

module.exports = ContextServiceFake;