[GRC] Tab CPU Usage Observer

services/resource_coordinator/coordination_unit/web_contents_coordination_unit_impl_unittest.cc

Index: services/resource_coordinator/coordination_unit/web_contents_coordination_unit_impl_unittest.cc
diff --git a/services/resource_coordinator/coordination_unit/web_contents_coordination_unit_impl_unittest.cc b/services/resource_coordinator/coordination_unit/web_contents_coordination_unit_impl_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..34c8f4c97ca6d2c4c67e1ecb2a6c31fb1fc8b61d
--- /dev/null
+++ b/services/resource_coordinator/coordination_unit/web_contents_coordination_unit_impl_unittest.cc
@@ -0,0 +1,151 @@
+// Copyright 2017 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include <string>
+
+#include "base/process/process_handle.h"
+#include "base/run_loop.h"
+#include "base/values.h"
+#include "services/resource_coordinator/coordination_unit/coordination_unit_factory.h"
+#include "services/resource_coordinator/coordination_unit/coordination_unit_impl.h"
+#include "services/resource_coordinator/coordination_unit/coordination_unit_impl_unittest_util.h"
+#include "services/resource_coordinator/public/cpp/coordination_unit_id.h"
+#include "services/resource_coordinator/public/cpp/coordination_unit_types.h"
+#include "services/resource_coordinator/public/interfaces/coordination_unit.mojom.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace resource_coordinator {
+
+namespace {
+
+class WebContentsCoordinationUnitImplTest : public CoordinationUnitImplTestBase {};
+
+}  // namespace
+
+TEST_F(WebContentsCoordinationUnitImplTest, SingleProcessSingleTabCPUUsageAttribution) {
+  base::ProcessId current_pid = base::GetCurrentProcId();
+  CoordinationUnitID process_cu_id(CoordinationUnitType::kProcess, current_pid);
+  CoordinationUnitID tab_cu_id(CoordinationUnitType::kWebContents, std::string());
+
+  std::unique_ptr<CoordinationUnitImpl> process_coordination_unit =
+      coordination_unit_factory::CreateCoordinationUnit(
+          process_cu_id, service_context_ref_factory()->CreateRef());
+  std::unique_ptr<CoordinationUnitImpl> tab_coordination_unit =
+      coordination_unit_factory::CreateCoordinationUnit(
+          tab_cu_id, service_context_ref_factory()->CreateRef());
+
+  process_coordination_unit->AddChild(tab_coordination_unit->id());
+  base::RunLoop().RunUntilIdle();
+
+  process_coordination_unit->SetProperty(mojom::PropertyType::kCPUUsage, base::Value(40.0));
+
+
+  base::Value tab_cpu_usage_value =
+      tab_coordination_unit->GetProperty(mojom::PropertyType::kCPUUsage);
+  EXPECT_EQ(40.0, tab_cpu_usage_value.GetDouble());
+}
+
+/*
+TEST_F(WebContentsCoordinationUnitImplTest, SingleProcessMultipleTabsCPUUsageAttribution) {
+  CoordinationUnitID process_cu_id(CoordinationUnitType::kProcess, std::string());
+  CoordinationUnitID first_tab_cu_id(CoordinationUnitType::kWebContents, std::string());
+  CoordinationUnitID second_tab_cu_id(CoordinationUnitType::kWebContents, std::string());
+
+  std::unique_ptr<CoordinationUnitImpl> process_coordination_unit =
+      coordination_unit_factory::CreateCoordinationUnit(
+          process_cu_id, service_context_ref_factory()->CreateRef());
+  std::unique_ptr<CoordinationUnitImpl> first_tab_coordination_unit =
+      coordination_unit_factory::CreateCoordinationUnit(
+          first_tab_cu_id, service_context_ref_factory()->CreateRef());
+  std::unique_ptr<CoordinationUnitImpl> second_tab_coordination_unit =
+      coordination_unit_factory::CreateCoordinationUnit(
+          second_tab_cu_id, service_context_ref_factory()->CreateRef());
+
+  process_coordination_unit->AddChild(first_tab_coordination_unit->id());
+  process_coordination_unit->AddChild(second_tab_coordination_unit->id());
+
+  base::RunLoop().RunUntilIdle();
+
+  process_coordination_unit->SetProperty(mojom::PropertyType::kCPUUsage, base::Value(40.0));
+
+  base::Value first_tab_cpu_usage_value =
+      first_tab_coordination_unit->GetProperty(mojom::PropertyType::kCPUUsage);
+  EXPECT_EQ(20.0, first_tab_cpu_usage_value.GetDouble());
+  base::Value second_tab_cpu_usage_value =
+      second_tab_coordination_unit->GetProperty(mojom::PropertyType::kCPUUsage);
+  EXPECT_EQ(20.0, second_tab_cpu_usage_value.GetDouble());
+
+  base::RunLoop().RunUntilIdle();
+}
+
+TEST_F(WebContentsCoordinationUnitImplTest, MultipleProcessesSingleTabCPUUsageAttribution) {
+  CoordinationUnitID first_process_cu_id(CoordinationUnitType::kProcess, std::string());
+  CoordinationUnitID second_process_cu_id(CoordinationUnitType::kProcess, std::string());
+  CoordinationUnitID tab_cu_id(CoordinationUnitType::kWebContents, std::string());
+
+  std::unique_ptr<CoordinationUnitImpl> first_process_coordination_unit =
+      coordination_unit_factory::CreateCoordinationUnit(
+          first_process_cu_id, service_context_ref_factory()->CreateRef());
+  std::unique_ptr<CoordinationUnitImpl> second_process_coordination_unit =
+      coordination_unit_factory::CreateCoordinationUnit(
+          second_process_cu_id, service_context_ref_factory()->CreateRef());
+  std::unique_ptr<CoordinationUnitImpl> tab_coordination_unit =
+      coordination_unit_factory::CreateCoordinationUnit(
+          tab_cu_id, service_context_ref_factory()->CreateRef());
+
+  first_process_coordination_unit->AddChild(tab_coordination_unit->id());
+  second_process_coordination_unit->AddChild(tab_coordination_unit->id());
+
+  base::RunLoop().RunUntilIdle();
+
+  first_process_coordination_unit->SetProperty(mojom::PropertyType::kCPUUsage, base::Value(40.0));
+  second_process_coordination_unit->SetProperty(mojom::PropertyType::kCPUUsage, base::Value(30.0));
+
+  base::Value tab_cpu_usage_value =
+      tab_coordination_unit->GetProperty(mojom::PropertyType::kCPUUsage);
+  EXPECT_EQ(70.0, tab_cpu_usage_value.GetDouble());
+
+  base::RunLoop().RunUntilIdle();
+}
+
+TEST_F(WebContentsCoordinationUnitImplTest, MultipleProcessesMultipleTabCPUUsageAttribution) {
+  CoordinationUnitID first_process_cu_id(CoordinationUnitType::kProcess, std::string());
+  CoordinationUnitID second_process_cu_id(CoordinationUnitType::kProcess, std::string());
+  CoordinationUnitID first_tab_cu_id(CoordinationUnitType::kWebContents, std::string());
+  CoordinationUnitID second_tab_cu_id(CoordinationUnitType::kWebContents, std::string());
+
+  std::unique_ptr<CoordinationUnitImpl> first_process_coordination_unit =
+      coordination_unit_factory::CreateCoordinationUnit(
+          first_process_cu_id, service_context_ref_factory()->CreateRef());
+  std::unique_ptr<CoordinationUnitImpl> second_process_coordination_unit =
+      coordination_unit_factory::CreateCoordinationUnit(
+          second_process_cu_id, service_context_ref_factory()->CreateRef());
+  std::unique_ptr<CoordinationUnitImpl> first_tab_coordination_unit =
+      coordination_unit_factory::CreateCoordinationUnit(
+          first_tab_cu_id, service_context_ref_factory()->CreateRef());
+  std::unique_ptr<CoordinationUnitImpl> second_tab_coordination_unit =
+      coordination_unit_factory::CreateCoordinationUnit(
+          second_tab_cu_id, service_context_ref_factory()->CreateRef());
+
+  first_process_coordination_unit->AddChild(first_tab_coordination_unit->id());
+  first_process_coordination_unit->AddChild(second_tab_coordination_unit->id());
+  second_process_coordination_unit->AddChild(first_tab_coordination_unit->id());
+
+  base::RunLoop().RunUntilIdle();
+
+  first_process_coordination_unit->SetProperty(mojom::PropertyType::kCPUUsage, base::Value(40.0));
+  second_process_coordination_unit->SetProperty(mojom::PropertyType::kCPUUsage, base::Value(30.0));
+
+  base::Value first_tab_cpu_usage_value =
+      first_tab_coordination_unit->GetProperty(mojom::PropertyType::kCPUUsage);
+  EXPECT_EQ(50.0, first_tab_cpu_usage_value.GetDouble());
+  base::Value second_tab_cpu_usage_value =
+      second_tab_coordination_unit->GetProperty(mojom::PropertyType::kCPUUsage);
+  EXPECT_EQ(30.0, second_tab_cpu_usage_value.GetDouble());
+
+  base::RunLoop().RunUntilIdle();
+}
+*/
+
+}  // namespace resource_coordinator