// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef COMPONENTS_AFFILIATIONS_CORE_BROWSER_FAKE_AFFILIATION_API_H_
#define COMPONENTS_AFFILIATIONS_CORE_BROWSER_FAKE_AFFILIATION_API_H_

#include <vector>

#include "base/memory/raw_ptr.h"
#include "components/affiliations/core/browser/affiliation_utils.h"
#include "components/affiliations/core/browser/fake_affiliation_fetcher.h"

namespace affiliations {

// Wraps the fake factory. Manufactures API responses to AffiliationFetcher
// requests based on a set of equivalence classes predefined by the tests.
class FakeAffiliationAPI {
 public:
  FakeAffiliationAPI();
  ~FakeAffiliationAPI();

  // Adds |affiliated_facets| to the set of equivalence classes that will form
  // the basis for calculating the fake API responses.
  void AddTestEquivalenceClass(const AffiliatedFacets& affiliated_facets);

  // Adds |group| to the array of |groups_| that will form
  // the basis for calculating the fake API responses.
  void AddTestGrouping(const GroupedFacets& group);

  // Returns whether or not there is at least one pending fetch.
  bool HasPendingRequest();

  // Returns AffiliationFetcher for the next pending fetch;
  // or nullptr if there are no pending fetches.
  FakeAffiliationFetcher* GetNextAffiliationFetcher();

  // Calculates the response to, and completes the next pending fetch, if any,
  // with success.
  void ServeNextRequest();

  // Completes the next pending fetch, if any, with failure.
  void FailNextRequest();

  // Sets the fetcher factory through which the affiliation fetchers are
  // accessed.
  void SetFetcherFactory(FakeAffiliationFetcherFactory* fake_fetcher_factory) {
    fake_fetcher_factory_ = fake_fetcher_factory;
  }

 private:
  raw_ptr<FakeAffiliationFetcherFactory> fake_fetcher_factory_ = nullptr;
  std::vector<AffiliatedFacets> preset_equivalence_relation_;
  std::vector<GroupedFacets> groups_;
};

}  // namespace affiliations

#endif  // COMPONENTS_AFFILIATIONS_CORE_BROWSER_FAKE_AFFILIATION_API_H_