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
|
// 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.
#include "content/browser/btm/btm_page_visit_observer_test_utils.h"
#include <iostream>
#include "content/browser/btm/btm_page_visit_observer.h"
namespace content {
std::ostream& operator<<(std::ostream& out, const BtmPageVisitInfo& page) {
return out << "BtmPageVisitInfo{url=" << page.url
<< ", source_id=" << page.source_id
<< ", had_active_storage_access=" << page.had_active_storage_access
<< ", received_user_activation=" << page.received_user_activation
<< ", had_successful_webauthn_assertion="
<< page.had_successful_web_authn_assertion
<< ", visit_duration=" << page.visit_duration << "}";
}
std::ostream& operator<<(std::ostream& out,
const BtmServerRedirectInfo& redirect) {
return out << "BtmServerRedirectInfo{url=" << redirect.url
<< ", source_id=" << redirect.source_id
<< ", did_write_cookies=" << redirect.did_write_cookies << "}";
}
std::ostream& operator<<(std::ostream& out, const BtmNavigationInfo& nav) {
out << "BtmNavigationInfo{server_redirects=[";
for (size_t i = 0; i < nav.server_redirects.size(); ++i) {
if (i > 0) {
out << ", ";
}
out << nav.server_redirects[i];
}
out << "], was_user_initiated=" << nav.was_user_initiated
<< ", was_renderer_initiated=" << nav.was_renderer_initiated
<< ", page_transition="
<< ui::PageTransitionGetCoreTransitionString(nav.page_transition) << "}";
return out;
}
std::ostream& operator<<(std::ostream& out,
const BtmPageVisitObserver::VisitTuple& visit) {
return out << "VisitTuple{prev_page=" << visit.prev_page
<< ", navigation=" << visit.navigation << "}";
}
BtmPageVisitRecorder::BtmPageVisitRecorder(WebContents* web_contents,
base::Clock* clock)
: observer_(web_contents,
base::BindRepeating(&BtmPageVisitRecorder::OnVisit,
base::Unretained(this)),
clock) {}
BtmPageVisitRecorder::~BtmPageVisitRecorder() = default;
[[nodiscard]] bool BtmPageVisitRecorder::WaitForSize(size_t n) {
if (visits_.size() >= n) {
return true;
}
if (wait_state_.has_value()) {
NOTREACHED()
<< "PageVisitWaiter::WaitForCount() called when already waiting";
}
wait_state_.emplace(n);
wait_state_->run_loop.Run();
wait_state_.reset();
return visits_.size() >= n;
}
void BtmPageVisitRecorder::OnVisit(BtmPageVisitInfo prev_page,
BtmNavigationInfo navigation) {
visits_.emplace_back(std::move(prev_page), std::move(navigation));
if (wait_state_.has_value() && wait_state_->wanted_count <= visits_.size()) {
wait_state_->run_loop.Quit();
}
}
} // namespace content
|
