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
|
// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef THIRD_PARTY_BLINK_RENDERER_PLATFORM_LOADER_TESTING_MOCK_FETCH_CONTEXT_H_
#define THIRD_PARTY_BLINK_RENDERER_PLATFORM_LOADER_TESTING_MOCK_FETCH_CONTEXT_H_
#include <memory>
#include <optional>
#include "base/memory/raw_ptr.h"
#include "base/task/single_thread_task_runner.h"
#include "base/types/optional_ref.h"
#include "services/network/public/cpp/permissions_policy/permissions_policy.h"
#include "third_party/blink/public/mojom/fetch/fetch_api_request.mojom-blink-forward.h"
#include "third_party/blink/public/mojom/timing/resource_timing.mojom-blink.h"
#include "third_party/blink/public/platform/platform.h"
#include "third_party/blink/public/platform/resource_load_info_notifier_wrapper.h"
#include "third_party/blink/public/platform/weak_wrapper_resource_load_info_notifier.h"
#include "third_party/blink/renderer/platform/exported/wrapped_resource_request.h"
#include "third_party/blink/renderer/platform/loader/fetch/fetch_client_settings_object_snapshot.h"
#include "third_party/blink/renderer/platform/loader/fetch/fetch_context.h"
#include "third_party/blink/renderer/platform/loader/fetch/fetch_parameters.h"
#include "third_party/blink/renderer/platform/scheduler/test/fake_task_runner.h"
namespace blink {
namespace mojom {
class ResourceLoadInfoNotifier;
} // namespace mojom
class KURL;
class ResourceRequest;
struct ResourceLoaderOptions;
// Mocked FetchContext for testing.
class MockFetchContext : public FetchContext {
public:
MockFetchContext() = default;
~MockFetchContext() override = default;
void set_blocked_urls(Vector<String> blocked_urls) {
blocked_urls_ = std::move(blocked_urls);
}
void set_tagged_urls(Vector<String> tagged_urls) {
tagged_urls_ = std::move(tagged_urls);
}
bool AllowImage() const override { return true; }
std::optional<ResourceRequestBlockedReason> CanRequest(
ResourceType,
const ResourceRequest&,
const KURL&,
const ResourceLoaderOptions&,
ReportingDisposition,
base::optional_ref<const ResourceRequest::RedirectInfo> redirect_info)
const override {
return std::nullopt;
}
std::optional<ResourceRequestBlockedReason>
CanRequestBasedOnSubresourceFilterOnly(
ResourceType type,
const ResourceRequest& resource_request,
const KURL& url,
const ResourceLoaderOptions& options,
ReportingDisposition reporting_disposition,
base::optional_ref<const ResourceRequest::RedirectInfo> redirect_info)
const override {
if (blocked_urls_.Contains(url.GetString())) {
return ResourceRequestBlockedReason::kSubresourceFilter;
}
return std::nullopt;
}
std::optional<ResourceRequestBlockedReason> CheckCSPForRequest(
mojom::blink::RequestContextType,
network::mojom::RequestDestination request_destination,
network::mojom::RequestMode request_mode,
const KURL& url,
const ResourceLoaderOptions& options,
ReportingDisposition reporting_disposition,
const KURL& url_before_redirects,
ResourceRequest::RedirectStatus redirect_status) const override {
return std::nullopt;
}
std::optional<ResourceRequestBlockedReason> CheckAndEnforceCSPForRequest(
mojom::blink::RequestContextType,
network::mojom::RequestDestination request_destination,
network::mojom::RequestMode request_mode,
const KURL& url,
const ResourceLoaderOptions& options,
ReportingDisposition reporting_disposition,
const KURL& url_before_redirects,
ResourceRequest::RedirectStatus redirect_status) const override {
return std::nullopt;
}
void AddResourceTiming(
mojom::blink::ResourceTimingInfoPtr resource_timing_info,
const AtomicString& initiator_type) override {}
std::unique_ptr<ResourceLoadInfoNotifierWrapper>
CreateResourceLoadInfoNotifierWrapper() override {
if (!resource_load_info_notifier_)
return nullptr;
if (!weak_wrapper_resource_load_info_notifier_) {
weak_wrapper_resource_load_info_notifier_ =
std::make_unique<WeakWrapperResourceLoadInfoNotifier>(
resource_load_info_notifier_);
}
return std::make_unique<ResourceLoadInfoNotifierWrapper>(
weak_wrapper_resource_load_info_notifier_->AsWeakPtr());
}
bool CalculateIfAdSubresource(
const ResourceRequestHead& resource_request,
base::optional_ref<const KURL> alias_url,
ResourceType type,
const FetchInitiatorInfo& initiator_info) override {
const KURL url =
alias_url.has_value() ? alias_url.value() : resource_request.Url();
return tagged_urls_.Contains(url.GetString());
}
void SetResourceLoadInfoNotifier(
mojom::ResourceLoadInfoNotifier* resource_load_info_notifier) {
resource_load_info_notifier_ = resource_load_info_notifier;
}
void SetPotentiallyUnusedPreload(const Vector<KURL>& urls) {
potentially_unused_preloads_ = urls;
}
const Vector<KURL>& GetPotentiallyUnusedPreloads() const override {
return potentially_unused_preloads_;
}
void SetPermissionsPolicy(
std::unique_ptr<network::PermissionsPolicy> policy) {
permissions_policy_ = std::move(policy);
}
const network::PermissionsPolicy* GetPermissionsPolicy() const override {
return permissions_policy_.get();
}
private:
raw_ptr<mojom::ResourceLoadInfoNotifier> resource_load_info_notifier_ =
nullptr;
std::unique_ptr<WeakWrapperResourceLoadInfoNotifier>
weak_wrapper_resource_load_info_notifier_;
Vector<String> blocked_urls_;
Vector<String> tagged_urls_;
Vector<KURL> potentially_unused_preloads_;
std::unique_ptr<network::PermissionsPolicy> permissions_policy_;
};
} // namespace blink
#endif // THIRD_PARTY_BLINK_RENDERER_PLATFORM_LOADER_TESTING_MOCK_FETCH_CONTEXT_H_
|
