1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
|
// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_BROWSER_ASH_WALLPAPER_HANDLERS_MOCK_WALLPAPER_HANDLERS_H_
#define CHROME_BROWSER_ASH_WALLPAPER_HANDLERS_MOCK_WALLPAPER_HANDLERS_H_
#include <stdint.h>
#include <string>
#include "chrome/browser/ash/wallpaper_handlers/wallpaper_handlers.h"
#include "testing/gmock/include/gmock/gmock.h"
namespace wallpaper_handlers {
// Fetcher that returns a list of backdrop image collections. Used to avoid
// network requests in unit tests.
class MockBackdropCollectionInfoFetcher : public BackdropCollectionInfoFetcher {
public:
MockBackdropCollectionInfoFetcher();
MockBackdropCollectionInfoFetcher(const MockBackdropCollectionInfoFetcher&) =
delete;
MockBackdropCollectionInfoFetcher& operator=(
const MockBackdropCollectionInfoFetcher&) = delete;
~MockBackdropCollectionInfoFetcher() override;
MOCK_METHOD(void, Start, (OnCollectionsInfoFetched callback), (override));
};
// Fetcher that returns a list of backdrop images. Used to avoid network
// requests in unit tests.
class MockBackdropImageInfoFetcher : public BackdropImageInfoFetcher {
public:
static constexpr uint64_t kTimeOfDayUnitId = 77;
explicit MockBackdropImageInfoFetcher(const std::string& collection_id);
MockBackdropImageInfoFetcher(const MockBackdropImageInfoFetcher&) = delete;
MockBackdropImageInfoFetcher& operator=(const MockBackdropImageInfoFetcher&) =
delete;
~MockBackdropImageInfoFetcher() override;
MOCK_METHOD(void, Start, (OnImagesInfoFetched callback), (override));
private:
std::string collection_id_;
};
// Fetcher that returns a backdrop image and empty resume token. Used to avoid
// network requests in unit tests.
class MockBackdropSurpriseMeImageFetcher
: public BackdropSurpriseMeImageFetcher {
public:
explicit MockBackdropSurpriseMeImageFetcher(const std::string& collection_id);
MockBackdropSurpriseMeImageFetcher(
const MockBackdropSurpriseMeImageFetcher&) = delete;
MockBackdropSurpriseMeImageFetcher& operator=(
const MockBackdropSurpriseMeImageFetcher&) = delete;
~MockBackdropSurpriseMeImageFetcher() override;
MOCK_METHOD(void, Start, (OnSurpriseMeImageFetched callback), (override));
private:
std::string collection_id_;
int id_incrementer_ = 0;
};
} // namespace wallpaper_handlers
#endif // CHROME_BROWSER_ASH_WALLPAPER_HANDLERS_MOCK_WALLPAPER_HANDLERS_H_
|