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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
|
// Copyright 2025 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_GOOGLE_PHOTOS_WALLPAPER_HANDLERS_H_
#define CHROME_BROWSER_ASH_WALLPAPER_HANDLERS_MOCK_GOOGLE_PHOTOS_WALLPAPER_HANDLERS_H_
#include <optional>
#include <string>
#include "ash/webui/personalization_app/mojom/personalization_app.mojom.h"
#include "base/functional/callback_forward.h"
#include "base/values.h"
#include "chrome/browser/ash/wallpaper_handlers/google_photos_wallpaper_handlers.h"
#include "testing/gmock/include/gmock/gmock.h"
namespace wallpaper_handlers {
// Fetcher that returns an empty album list and no resume token in response to a
// request for the user's Google Photos albums. Used to avoid network requests
// in unit tests.
class MockGooglePhotosAlbumsFetcher : public GooglePhotosAlbumsFetcher {
public:
explicit MockGooglePhotosAlbumsFetcher(Profile* profile);
MockGooglePhotosAlbumsFetcher(const MockGooglePhotosAlbumsFetcher&) = delete;
MockGooglePhotosAlbumsFetcher& operator=(
const MockGooglePhotosAlbumsFetcher&) = delete;
~MockGooglePhotosAlbumsFetcher() override;
// GooglePhotosAlbumsFetcher:
MOCK_METHOD(void,
AddRequestAndStartIfNecessary,
(const std::optional<std::string>& resume_token,
base::OnceCallback<void(GooglePhotosAlbumsCbkArgs)> callback),
(override));
MOCK_METHOD(GooglePhotosAlbumsCbkArgs,
ParseResponse,
(const base::Value::Dict* response),
(override));
// Overridden to increase visibility.
std::optional<size_t> GetResultCount(
const GooglePhotosAlbumsCbkArgs& result) override;
};
// Fetcher that returns an empty album list and no resume token in response to a
// request for the user's Google Photos shared albums. Used to avoid network
// requests in unit tests.
class MockGooglePhotosSharedAlbumsFetcher
: public GooglePhotosSharedAlbumsFetcher {
public:
explicit MockGooglePhotosSharedAlbumsFetcher(Profile* profile);
MockGooglePhotosSharedAlbumsFetcher(
const MockGooglePhotosSharedAlbumsFetcher&) = delete;
MockGooglePhotosSharedAlbumsFetcher& operator=(
const MockGooglePhotosSharedAlbumsFetcher&) = delete;
~MockGooglePhotosSharedAlbumsFetcher() override;
// GooglePhotosSharedAlbumsFetcher:
MOCK_METHOD(void,
AddRequestAndStartIfNecessary,
(const std::optional<std::string>& resume_token,
base::OnceCallback<void(GooglePhotosAlbumsCbkArgs)> callback),
(override));
MOCK_METHOD(GooglePhotosAlbumsCbkArgs,
ParseResponse,
(const base::Value::Dict* response),
(override));
// Overridden to increase visibility.
std::optional<size_t> GetResultCount(
const GooglePhotosAlbumsCbkArgs& result) override;
};
// Fetcher that claims the user is allowed to access Google Photos data. Used to
// avoid network requests in unit tests.
class MockGooglePhotosEnabledFetcher : public GooglePhotosEnabledFetcher {
public:
explicit MockGooglePhotosEnabledFetcher(Profile* profile);
MockGooglePhotosEnabledFetcher(const MockGooglePhotosEnabledFetcher&) =
delete;
MockGooglePhotosEnabledFetcher& operator=(
const MockGooglePhotosEnabledFetcher&) = delete;
~MockGooglePhotosEnabledFetcher() override;
// GooglePhotosEnabledFetcher:
MOCK_METHOD(void,
AddRequestAndStartIfNecessary,
(base::OnceCallback<void(GooglePhotosEnablementState)> callback),
(override));
MOCK_METHOD(GooglePhotosEnablementState,
ParseResponse,
(const base::Value::Dict* response),
(override));
// Overridden to increase visibility.
std::optional<size_t> GetResultCount(
const GooglePhotosEnablementState& result) override;
};
// Fetcher that returns an empty photo list and no resume token in response to a
// request for photos from the user's Google Photos library. Used to avoid
// network requests in unit tests.
class MockGooglePhotosPhotosFetcher : public GooglePhotosPhotosFetcher {
public:
explicit MockGooglePhotosPhotosFetcher(Profile* profile);
MockGooglePhotosPhotosFetcher(const MockGooglePhotosPhotosFetcher&) = delete;
MockGooglePhotosPhotosFetcher& operator=(
const MockGooglePhotosPhotosFetcher&) = delete;
~MockGooglePhotosPhotosFetcher() override;
// GooglePhotosPhotosFetcher:
MOCK_METHOD(void,
AddRequestAndStartIfNecessary,
(const std::optional<std::string>& item_id,
const std::optional<std::string>& album_id,
const std::optional<std::string>& resume_token,
bool shuffle,
base::OnceCallback<void(GooglePhotosPhotosCbkArgs)> callback),
(override));
MOCK_METHOD(GooglePhotosPhotosCbkArgs,
ParseResponse,
(const base::Value::Dict* response),
(override));
// Overridden to increase visibility.
std::optional<size_t> GetResultCount(
const GooglePhotosPhotosCbkArgs& result) override;
};
} // namespace wallpaper_handlers
#endif // CHROME_BROWSER_ASH_WALLPAPER_HANDLERS_MOCK_GOOGLE_PHOTOS_WALLPAPER_HANDLERS_H_
|
