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
content / web_test / browser / web_test_pressure_manager.cc [blame]
// Copyright 2024 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/web_test/browser/web_test_pressure_manager.h"
#include "content/browser/compute_pressure/web_contents_pressure_manager_proxy.h"
#include "content/browser/web_contents/web_contents_impl.h"
#include "content/public/browser/web_contents.h"
namespace content {
WebTestPressureManager::WebTestPressureManager(WebContents* web_contents)
: WebContentsUserData<WebTestPressureManager>(*web_contents) {}
WebTestPressureManager::~WebTestPressureManager() = default;
// static
WebTestPressureManager* WebTestPressureManager::GetOrCreate(
WebContents* web_contents) {
WebContentsUserData<WebTestPressureManager>::CreateForWebContents(
web_contents);
return WebContentsUserData<WebTestPressureManager>::FromWebContents(
web_contents);
}
void WebTestPressureManager::Bind(
mojo::PendingReceiver<blink::test::mojom::WebPressureManagerAutomation>
receiver) {
receivers_.Add(this, std::move(receiver));
}
void WebTestPressureManager::CreateVirtualPressureSource(
device::mojom::PressureSource source,
device::mojom::VirtualPressureSourceMetadataPtr metadata,
CreateVirtualPressureSourceCallback callback) {
if (pressure_source_overrides_.contains(source)) {
std::move(callback).Run(
blink::test::mojom::CreateVirtualPressureSourceResult::
kSourceTypeAlreadyOverridden);
return;
}
auto virtual_pressure_source =
WebContentsPressureManagerProxy::GetOrCreate(&GetWebContents())
->CreateVirtualPressureSourceForDevTools(source, std::move(metadata));
CHECK(virtual_pressure_source);
pressure_source_overrides_[source] = std::move(virtual_pressure_source);
std::move(callback).Run(
blink::test::mojom::CreateVirtualPressureSourceResult::kSuccess);
}
void WebTestPressureManager::RemoveVirtualPressureSource(
device::mojom::PressureSource source,
RemoveVirtualPressureSourceCallback callback) {
pressure_source_overrides_.erase(source);
std::move(callback).Run();
}
void WebTestPressureManager::UpdateVirtualPressureSourceState(
device::mojom::PressureSource source,
device::mojom::PressureState state,
UpdateVirtualPressureSourceStateCallback callback) {
auto it = pressure_source_overrides_.find(source);
if (it == pressure_source_overrides_.end()) {
std::move(callback).Run(
blink::test::mojom::UpdateVirtualPressureSourceStateResult::
kSourceTypeNotOverridden);
return;
}
it->second->UpdateVirtualPressureSourceState(
state,
base::BindOnce(std::move(callback),
blink::test::mojom::
UpdateVirtualPressureSourceStateResult::kSuccess));
}
WEB_CONTENTS_USER_DATA_KEY_IMPL(WebTestPressureManager);
} // namespace content