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
|
// 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.
#include "components/performance_manager/test_support/test_browser_child_process.h"
#include <utility>
#include "base/check.h"
#include "components/performance_manager/browser_child_process_watcher.h"
#include "components/performance_manager/performance_manager_registry_impl.h"
#include "content/public/browser/child_process_data.h"
#include "content/public/browser/child_process_host.h"
namespace performance_manager {
TestBrowserChildProcess::TestBrowserChildProcess(
content::ProcessType process_type)
: host_(content::BrowserChildProcessHost::Create(process_type, this)) {}
TestBrowserChildProcess::~TestBrowserChildProcess() {
if (is_connected) {
SimulateDisconnect();
}
}
BrowserChildProcessHostId TestBrowserChildProcess::GetId() const {
return BrowserChildProcessHostId(host_->GetData().id);
}
BrowserChildProcessHostProxy TestBrowserChildProcess::GetProxy() const {
return BrowserChildProcessHostProxy::CreateForTesting(GetId());
}
void TestBrowserChildProcess::SimulateLaunch(base::Process process) {
CHECK(!is_connected);
// Simulate a successful process launch with a copy of ChildProcessData that
// has an existing process attached. ChildProcessData is move-only, but
// GetData() returns a const ref, so must copy each field individually.
content::ChildProcessData data(host_->GetData().process_type,
host_->GetData().GetChildProcessId());
data.name = host_->GetData().name;
data.metrics_name = host_->GetData().metrics_name;
data.sandbox_type = host_->GetData().sandbox_type;
data.SetProcess(std::move(process));
PerformanceManagerRegistryImpl::GetInstance()
->GetBrowserChildProcessWatcherForTesting()
.CreateChildProcessNodeForTesting(std::move(data));
is_connected = true;
}
void TestBrowserChildProcess::SimulateDisconnect() {
CHECK(is_connected);
PerformanceManagerRegistryImpl::GetInstance()
->GetBrowserChildProcessWatcherForTesting()
.DeleteChildProcessNodeForTesting(host_->GetData());
is_connected = false;
}
void DeleteBrowserProcessNodeForTesting() {
PerformanceManagerRegistryImpl::GetInstance()
->GetBrowserChildProcessWatcherForTesting()
.DeleteBrowserProcessNodeForTesting();
}
} // namespace performance_manager
|
