// Copyright 2020 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_MOCK_AFFILIATION_FETCHER_H_
#define COMPONENTS_AFFILIATIONS_CORE_BROWSER_MOCK_AFFILIATION_FETCHER_H_

#include "base/functional/callback.h"
#include "components/affiliations/core/browser/affiliation_fetcher_interface.h"
#include "testing/gmock/include/gmock/gmock.h"

namespace affiliations {

class MockAffiliationFetcher : public AffiliationFetcherInterface {
 public:
  MockAffiliationFetcher();
  ~MockAffiliationFetcher() override;

  MOCK_METHOD(void,
              StartRequest,
              (const std::vector<FacetURI>&,
               RequestInfo,
               base::OnceCallback<void(FetchResult)>),
              (override));
  MOCK_METHOD(std::vector<FacetURI>&,
              GetRequestedFacetURIs,
              (),
              (const, override));
};

}  // namespace affiliations

#endif  // COMPONENTS_AFFILIATIONS_CORE_BROWSER_MOCK_AFFILIATION_FETCHER_H_