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
|
// 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"
#include "services/device/public/mojom/pressure_update.mojom.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::UpdateVirtualPressureSourceData(
device::mojom::PressureSource source,
device::mojom::PressureState state,
double own_contribution_estimate,
UpdateVirtualPressureSourceDataCallback callback) {
auto it = pressure_source_overrides_.find(source);
if (it == pressure_source_overrides_.end()) {
std::move(callback).Run(
blink::test::mojom::UpdateVirtualPressureSourceDataResult::
kSourceTypeNotOverridden);
return;
}
it->second->UpdateVirtualPressureSourceData(
state, own_contribution_estimate,
base::BindOnce(
std::move(callback),
blink::test::mojom::UpdateVirtualPressureSourceDataResult::kSuccess));
}
WEB_CONTENTS_USER_DATA_KEY_IMPL(WebTestPressureManager);
} // namespace content
|