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
|
// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "content/public/test/mock_client_hints_controller_delegate.h"
#include "content/public/common/origin_util.h"
#include "services/network/public/cpp/is_potentially_trustworthy.h"
#include "third_party/blink/public/common/user_agent/user_agent_metadata.h"
#include "ui/gfx/geometry/size_f.h"
#include "url/gurl.h"
namespace content {
namespace {
bool PersistClientHintsHelper(const GURL& url,
const blink::EnabledClientHints& client_hints,
ClientHintsContainer* container) {
DCHECK(container);
if (!network::IsUrlPotentiallyTrustworthy(url)) {
return false;
}
const url::Origin origin = url::Origin::Create(url);
(*container)[origin] = client_hints;
return true;
}
void GetAllowedClientHintsFromSourceHelper(
const url::Origin& origin,
const ClientHintsContainer& container,
blink::EnabledClientHints* client_hints) {
const auto& it = container.find(origin);
DCHECK(client_hints);
if (it != container.end()) {
*client_hints = it->second;
}
}
} // namespace
MockClientHintsControllerDelegate::MockClientHintsControllerDelegate(
const blink::UserAgentMetadata& metadata)
: metadata_(metadata) {}
MockClientHintsControllerDelegate::~MockClientHintsControllerDelegate() =
default;
network::NetworkQualityTracker*
MockClientHintsControllerDelegate::GetNetworkQualityTracker() {
return &network_quality_tracker_;
}
bool MockClientHintsControllerDelegate::IsJavaScriptAllowed(
const GURL& url,
content::RenderFrameHost* parent_rfh) {
return true;
}
blink::UserAgentMetadata
MockClientHintsControllerDelegate::GetUserAgentMetadata() {
return metadata_;
}
void MockClientHintsControllerDelegate::PersistClientHints(
const url::Origin& primary_origin,
content::RenderFrameHost* parent_rfh,
const std::vector<::network::mojom::WebClientHintsType>& client_hints) {
blink::EnabledClientHints enabled_client_hints;
for (const auto& type : client_hints) {
enabled_client_hints.SetIsEnabled(type, true);
}
PersistClientHintsHelper(primary_origin.GetURL(), enabled_client_hints,
&client_hints_map_);
}
// Get which client hints opt-ins were persisted on current origin.
void MockClientHintsControllerDelegate::GetAllowedClientHintsFromSource(
const url::Origin& origin,
blink::EnabledClientHints* client_hints) {
GetAllowedClientHintsFromSourceHelper(origin, client_hints_map_,
client_hints);
for (auto hint : additional_hints_)
client_hints->SetIsEnabled(hint, true);
}
void MockClientHintsControllerDelegate::ResetForTesting() {
client_hints_map_.clear();
additional_hints_.clear();
}
void MockClientHintsControllerDelegate::SetAdditionalClientHints(
const std::vector<network::mojom::WebClientHintsType>& hints) {
additional_hints_ = hints;
}
void MockClientHintsControllerDelegate::ClearAdditionalClientHints() {
additional_hints_.clear();
}
void MockClientHintsControllerDelegate::SetMostRecentMainFrameViewportSize(
const gfx::Size& viewport_size) {
viewport_size_ = viewport_size;
}
gfx::Size
MockClientHintsControllerDelegate::GetMostRecentMainFrameViewportSize() {
return viewport_size_;
}
} // end namespace content
|
