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
content / test / fenced_frame_test_utils.cc [blame]
// Copyright 2022 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/test/fenced_frame_test_utils.h"
#include "base/location.h"
#include "base/memory/ref_counted.h"
#include "base/run_loop.h"
#include "base/test/test_timeouts.h"
#include "base/time/time.h"
#include "content/browser/fenced_frame/fenced_frame.h"
#include "content/browser/fenced_frame/fenced_frame_reporter.h"
#include "content/browser/renderer_host/frame_tree_node.h"
#include "content/browser/renderer_host/render_frame_host_impl.h"
#include "content/public/test/browser_test_utils.h"
#include "content/public/test/test_frame_navigation_observer.h"
#include "third_party/blink/public/common/features.h"
namespace content {
using SharedStorageReportingMap = base::flat_map<std::string, ::GURL>;
FrameTreeNode* GetFencedFrameRootNode(FrameTreeNode* node) {
FrameTreeNodeId inner_node_id =
node->current_frame_host()->inner_tree_main_frame_tree_node_id();
return FrameTreeNode::GloballyFindByID(inner_node_id);
}
void SimulateSharedStorageURNMappingComplete(
FencedFrameURLMapping& fenced_frame_url_mapping,
const GURL& urn_uuid,
const GURL& mapped_url,
const net::SchemefulSite& shared_storage_site,
double budget_to_charge,
scoped_refptr<FencedFrameReporter> fenced_frame_reporter) {
SharedStorageBudgetMetadata budget_metadata = {
.site = shared_storage_site, .budget_to_charge = budget_to_charge};
fenced_frame_url_mapping.OnSharedStorageURNMappingResultDetermined(
urn_uuid,
FencedFrameURLMapping::SharedStorageURNMappingResult(
mapped_url, budget_metadata, std::move(fenced_frame_reporter)));
}
TestFencedFrameURLMappingResultObserver::
TestFencedFrameURLMappingResultObserver() = default;
TestFencedFrameURLMappingResultObserver::
~TestFencedFrameURLMappingResultObserver() = default;
void TestFencedFrameURLMappingResultObserver::OnFencedFrameURLMappingComplete(
const std::optional<FencedFrameProperties>& properties) {
mapping_complete_observed_ = true;
observed_fenced_frame_properties_ = properties;
}
bool FencedFrameURLMappingTestPeer::HasObserver(
const GURL& urn_uuid,
FencedFrameURLMapping::MappingResultObserver* observer) {
return fenced_frame_url_mapping_->IsPendingMapped(urn_uuid) &&
fenced_frame_url_mapping_->pending_urn_uuid_to_url_map_.at(urn_uuid)
.count(observer);
}
void FencedFrameURLMappingTestPeer::GetSharedStorageReportingMap(
const GURL& urn_uuid,
SharedStorageReportingMap* out_reporting_map) {
DCHECK(out_reporting_map);
auto urn_it = fenced_frame_url_mapping_->urn_uuid_to_url_map_.find(urn_uuid);
CHECK(urn_it != fenced_frame_url_mapping_->urn_uuid_to_url_map_.end());
scoped_refptr<FencedFrameReporter> fenced_frame_reporter =
urn_it->second.fenced_frame_reporter();
if (!fenced_frame_reporter) {
return;
}
const auto& metadata = fenced_frame_reporter->reporting_metadata();
auto data_it = metadata.find(
blink::FencedFrame::ReportingDestination::kSharedStorageSelectUrl);
if (data_it != metadata.end()) {
// No need to check if `reporting_url_map` is null - it never is for
// kSharedStorageSelectUrl reporting destinations.
*out_reporting_map = *data_it->second.reporting_url_map;
}
}
void FencedFrameURLMappingTestPeer::FillMap(const GURL& url) {
while (!fenced_frame_url_mapping_->IsFull()) {
auto it = fenced_frame_url_mapping_->AddMappingForUrl(url);
DCHECK(it.has_value());
}
DCHECK(fenced_frame_url_mapping_->IsFull());
}
bool PollUntilEvalToTrue(const std::string& script, RenderFrameHost* rfh) {
base::Time start_time = base::Time::Now();
base::TimeDelta timeout = TestTimeouts::action_max_timeout();
while (base::Time::Now() - start_time < timeout) {
EvalJsResult result = EvalJs(rfh, script);
if (!result.error.empty()) {
return false;
} else if (result.ExtractBool()) {
return true;
}
// Wait for a bit here to keep this loop from spinlocking too badly.
base::RunLoop run_loop;
base::SingleThreadTaskRunner::GetCurrentDefault()->PostDelayedTask(
FROM_HERE, run_loop.QuitClosure(), TestTimeouts::tiny_timeout());
run_loop.Run();
}
return false;
}
} // namespace content