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
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
|
// Copyright 2023 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_WEB_APPLICATIONS_TEST_FAKE_WEB_CONTENTS_MANAGER_H_
#define CHROME_BROWSER_WEB_APPLICATIONS_TEST_FAKE_WEB_CONTENTS_MANAGER_H_
#include <map>
#include <memory>
#include <optional>
#include <string>
#include <string_view>
#include "base/functional/callback_helpers.h"
#include "base/memory/weak_ptr.h"
#include "chrome/browser/web_applications/web_app_install_info.h"
#include "chrome/browser/web_applications/web_contents/web_app_data_retriever.h"
#include "chrome/browser/web_applications/web_contents/web_app_icon_downloader.h"
#include "chrome/browser/web_applications/web_contents/web_contents_manager.h"
#include "components/webapps/browser/installable/installable_logging.h"
#include "components/webapps/browser/web_contents/web_app_url_loader.h"
#include "components/webapps/common/web_page_metadata.mojom-forward.h"
#include "third_party/blink/public/mojom/manifest/manifest.mojom.h"
#include "url/gurl.h"
class GURL;
class SkBitmap;
namespace content {
class WebContents;
}
namespace web_app {
// This class facilities creating WebAppUrlLoaders and WebAppDataRetrievers that
// reflect a fake network state. This class can be re-used when creating a
// general dependency wrapper for the web contents system, see
// http://b/262606416.
class FakeWebContentsManager : public WebContentsManager {
public:
// Some helper methods.
static webapps::mojom::WebPageMetadataPtr CreateMetadataWithTitle(
std::u16string title);
static webapps::mojom::WebPageMetadataPtr CreateMetadataWithIconAndTitle(
std::u16string title,
GURL document_icon_url,
int32_t icon_size);
// State used to represent a page at a url, which is retrieved through
// `LoadUrl`, `GetWebAppInstallInfo`, and
// `CheckInstallabilityAndRetrieveManifest`.
struct FakePageState {
FakePageState();
~FakePageState();
FakePageState(FakePageState&&);
FakePageState& operator=(FakePageState&&);
webapps::AppId PopulateWithBasicManifest(GURL install_url,
GURL manifest_url,
GURL start_url);
// `WebAppUrlLoader::LoadUrl`:
// If this is populated, then a redirection is always assumed. If the
// redirection is allowed by the `LoadUrlComparison`, then `url_load_result`
// will be given as the result. Otherwise,`kRedirectedUrlLoaded` is
// returned.
std::optional<GURL> redirection_url = std::nullopt;
webapps::WebAppUrlLoaderResult url_load_result =
webapps::WebAppUrlLoaderResult::kFailedErrorPageLoaded;
// `WebAppDataRetriever::GetWebAppInstallInfo`:
bool return_null_info = false;
std::optional<std::u16string> title;
webapps::mojom::WebPageMetadataPtr opt_metadata;
// `WebAppDataRetriever::CheckInstallabilityAndRetrieveManifest`:
// Called when this call is executed.
base::OnceClosure on_manifest_fetch;
bool has_service_worker = false;
// An empty url is considered an absent url.
GURL manifest_url;
bool valid_manifest_for_web_app = false;
// Manifests always go through default processing (e.g. start_url defaulting
// to the document_url, and manifest_id defaulting to start_url). This fake
// system will ensure that `CheckInstallabilityAndRetrieveManifest` this
// behavior is replicated before the manifest is returned for
// `CheckInstallabilityAndRetrieveManifest`.
blink::mojom::ManifestPtr manifest_before_default_processing;
webapps::InstallableStatusCode error_code =
webapps::InstallableStatusCode::NO_ERROR_DETECTED;
GURL favicon_url;
std::vector<SkBitmap> favicon;
};
// State used to represent an icon at a url, which is retrieved through
// `GetIcons`.
struct FakeIconState {
FakeIconState();
~FakeIconState();
int http_status_code = 200;
std::vector<SkBitmap> bitmaps;
// This can be used to test the early-exit normally caused by the
// WebContents closing or navigating away.
bool trigger_primary_page_changed_if_fetched = false;
};
FakeWebContentsManager();
~FakeWebContentsManager() override;
void SetUrlLoaded(content::WebContents* web_contents, const GURL& url);
std::unique_ptr<webapps::WebAppUrlLoader> CreateUrlLoader() override;
std::unique_ptr<WebAppDataRetriever> CreateDataRetriever() override;
std::unique_ptr<WebAppIconDownloader> CreateIconDownloader() override;
// Set the behavior for calls to `GetIcons` from wrappers returned by this
// fake class.
void SetIconState(const GURL& icon_url, const FakeIconState& icon_state);
FakeIconState& GetOrCreateIconState(const GURL& icon_url);
void DeleteIconState(const GURL& icon_url);
// Set the behavior for calls to `LoadUrl`, `GetWebAppInstallInfo`, and
// `CheckInstallabilityAndRetrieveManifest` from wrappers returned by this
// fake class.
webapps::AppId CreateBasicInstallPageState(
const GURL& install_url,
const GURL& manifest_url,
const GURL& start_url,
std::u16string_view name = u"Basic app name");
void SetPageState(const GURL& gurl, FakePageState page_state);
FakePageState& GetOrCreatePageState(const GURL& gurl);
void DeletePageState(const GURL& gurl);
bool HasPageState(const GURL& gurl);
using LoadUrlTracker = base::RepeatingCallback<void(
content::NavigationController::LoadURLParams& load_url_params,
content::WebContents* web_contents,
webapps::WebAppUrlLoader::UrlComparison url_comparison)>;
// Specify a `base::RepeatingCallback` that is invoked with the arguments
// passed to `WebAppUrlLoader::LoadUrl` whenever it is called on one of the
// `WebAppUrlLoader`s created by this class.
void TrackLoadUrlCalls(LoadUrlTracker load_url_tracker);
base::WeakPtr<FakeWebContentsManager> GetWeakPtr();
private:
class FakeUrlLoader;
class FakeWebAppIconDownloader;
class FakeWebAppDataRetriever;
LoadUrlTracker load_url_tracker_ = base::DoNothing();
std::map<GURL, FakeIconState> icon_state_;
std::map<GURL, FakePageState> page_state_;
// This is a hack that can be removed once we centralize WebContents usage
// through one wrapper class. This keeps track of the last loaded url for the
// given web contents pointer, allowing the data retriver to know what the url
// loader did.
std::map<content::WebContents*, GURL> loaded_urls_;
base::WeakPtrFactory<FakeWebContentsManager> weak_factory_{this};
};
} // namespace web_app
#endif // CHROME_BROWSER_WEB_APPLICATIONS_TEST_FAKE_WEB_CONTENTS_MANAGER_H_
|
