Index: services/resource_coordinator/coordination_unit/coordination_unit_graph_observer_unittest.cc |
diff --git a/services/resource_coordinator/coordination_unit/coordination_unit_graph_observer_unittest.cc b/services/resource_coordinator/coordination_unit/coordination_unit_graph_observer_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..9d3e4be46bf3ccd9648cf57fd7ffccf65fbbcf96 |
--- /dev/null |
+++ b/services/resource_coordinator/coordination_unit/coordination_unit_graph_observer_unittest.cc |
@@ -0,0 +1,179 @@ |
+// 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 <utility> |
+ |
+#include "base/memory/ptr_util.h" |
+#include "base/process/process_handle.h" |
+#include "base/run_loop.h" |
+#include "services/resource_coordinator/coordination_unit/coordination_unit_factory.h" |
+#include "services/resource_coordinator/coordination_unit/coordination_unit_graph_observer.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/coordination_unit/coordination_unit_manager.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 CoordinationUnitGraphObserverTest : public CoordinationUnitImplTestBase { |
+}; |
+ |
+class TestCoordinationUnitGraphObserver : public CoordinationUnitGraphObserver { |
+ public: |
+ TestCoordinationUnitGraphObserver() |
+ : on_child_added_count_(0u), |
+ on_parent_added_count_(0u), |
+ on_coordination_unit_created_count_(0u), |
+ on_property_changed_count_(0u), |
+ on_child_removed_count_(0u), |
+ on_parent_removed_count_(0u), |
+ on_coordination_unit_will_be_destroyed_count_(0u) {} |
+ |
+ size_t on_child_added_count() { return on_child_added_count_; } |
+ size_t on_parent_added_count() { return on_parent_added_count_; } |
+ size_t on_coordination_unit_created_count() { |
+ return on_coordination_unit_created_count_; |
+ } |
+ size_t on_property_changed_count() { return on_property_changed_count_; } |
+ size_t on_child_removed_count() { return on_child_removed_count_; } |
+ size_t on_parent_removed_count() { return on_parent_removed_count_; } |
+ size_t on_coordination_unit_will_be_destroyed_count() { |
+ return on_coordination_unit_will_be_destroyed_count_; |
+ } |
+ |
+ // Overridden from CoordinationUnitGraphObserver. |
+ bool ShouldObserve(const CoordinationUnitImpl* coordination_unit) override { |
+ return coordination_unit->id().type == CoordinationUnitType::kFrame; |
+ } |
+ void OnCoordinationUnitCreated( |
+ const CoordinationUnitImpl* coordination_unit) override { |
+ ++on_coordination_unit_created_count_; |
+ } |
+ void OnChildAdded( |
+ const CoordinationUnitImpl* coordination_unit, |
+ const CoordinationUnitImpl* child_coordination_unit) override { |
+ ++on_child_added_count_; |
+ } |
+ void OnParentAdded( |
+ const CoordinationUnitImpl* coordination_unit, |
+ const CoordinationUnitImpl* parent_coordination_unit) override { |
+ ++on_parent_added_count_; |
+ } |
+ void OnPropertyChanged(const CoordinationUnitImpl* coordination_unit, |
+ mojom::PropertyType property) override { |
+ ++on_property_changed_count_; |
+ } |
+ void OnChildRemoved( |
+ const CoordinationUnitImpl* coordination_unit, |
+ const CoordinationUnitImpl* former_child_coordination_unit) override { |
+ ++on_child_removed_count_; |
+ } |
+ void OnParentRemoved( |
+ const CoordinationUnitImpl* coordination_unit, |
+ const CoordinationUnitImpl* former_parent_coordination_unit) override { |
+ ++on_parent_removed_count_; |
+ } |
+ void OnCoordinationUnitWillBeDestroyed( |
+ const CoordinationUnitImpl* coordination_unit) override { |
+ ++on_coordination_unit_will_be_destroyed_count_; |
+ } |
+ |
+ private: |
+ size_t on_child_added_count_; |
+ size_t on_parent_added_count_; |
+ size_t on_coordination_unit_created_count_; |
+ size_t on_property_changed_count_; |
+ size_t on_child_removed_count_; |
+ size_t on_parent_removed_count_; |
+ size_t on_coordination_unit_will_be_destroyed_count_; |
+}; |
+ |
+} // namespace |
+ |
+TEST_F(CoordinationUnitGraphObserverTest, CallbacksInvokedNoFilters) { |
Zhen Wang
2017/06/21 03:48:36
nit: CallbacksInvokedNoFilters/CallbacksInvoked
T
|
+ EXPECT_TRUE(coordination_unit_manager().observers_for_testing().empty()); |
+ coordination_unit_manager().RegisterObserver( |
+ base::MakeUnique<TestCoordinationUnitGraphObserver>()); |
+ EXPECT_EQ(1u, coordination_unit_manager().observers_for_testing().size()); |
+ |
+ TestCoordinationUnitGraphObserver* observer = |
+ static_cast<TestCoordinationUnitGraphObserver*>( |
+ coordination_unit_manager().observers_for_testing()[0].get()); |
+ |
+ CoordinationUnitID process_cu_id(CoordinationUnitType::kProcess, |
+ std::string()); |
+ CoordinationUnitID root_frame_cu_id(CoordinationUnitType::kFrame, |
+ std::string()); |
+ CoordinationUnitID frame_cu_id(CoordinationUnitType::kFrame, 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> root_frame_coordination_unit = |
+ coordination_unit_factory::CreateCoordinationUnit( |
+ root_frame_cu_id, service_context_ref_factory()->CreateRef()); |
+ std::unique_ptr<CoordinationUnitImpl> frame_coordination_unit = |
+ coordination_unit_factory::CreateCoordinationUnit( |
+ frame_cu_id, service_context_ref_factory()->CreateRef()); |
+ |
+ coordination_unit_manager().OnCoordinationUnitCreated( |
+ process_coordination_unit.get()); |
+ coordination_unit_manager().OnCoordinationUnitCreated( |
+ root_frame_coordination_unit.get()); |
+ coordination_unit_manager().OnCoordinationUnitCreated( |
+ frame_coordination_unit.get()); |
+ EXPECT_EQ(2u, observer->on_coordination_unit_created_count()); |
+ |
+ // The registered observer will only observe the events that happen to |
+ // |root_frame_coordination_unit| and |frame_coordination_unit| because |
+ // they are CoordinationUnitType::kFrame. |
+ // OnAddParent will called for |root_frame_coordination_unit|. |
+ process_coordination_unit->AddChild(root_frame_coordination_unit->id()); |
+ // OnAddParent will called for |frame_coordination_unit|. |
+ process_coordination_unit->AddChild(frame_coordination_unit->id()); |
+ // OnAddChild will called for |root_frame_coordination_unit| and |
+ // OnAddParent will called for |frame_coordination_unit|. |
+ root_frame_coordination_unit->AddChild(frame_coordination_unit->id()); |
+ EXPECT_EQ(1u, observer->on_child_added_count()); |
+ EXPECT_EQ(3u, observer->on_parent_added_count()); |
+ |
+ // The registered observer will only observe the events that happen to |
+ // |root_frame_coordination_unit| and |frame_coordination_unit| because |
+ // they are CoordinationUnitType::kFrame. |
+ // OnRemoveParent will called for |root_frame_coordination_unit|. |
+ process_coordination_unit->RemoveChild(root_frame_coordination_unit->id()); |
+ // OnRemoveParent will called for |frame_coordination_unit|. |
+ process_coordination_unit->RemoveChild(frame_coordination_unit->id()); |
+ // OnRemoveChild will called for |root_frame_coordination_unit| and |
+ // OnRemoveParent will called for |frame_coordination_unit|. |
+ root_frame_coordination_unit->RemoveChild(frame_coordination_unit->id()); |
+ EXPECT_EQ(1u, observer->on_child_removed_count()); |
+ EXPECT_EQ(3u, observer->on_parent_removed_count()); |
+ |
+ // The registered observer will only observe the events that happen to |
+ // |root_frame_coordination_unit| and |frame_coordination_unit| because |
+ // they are CoordinationUnitType::kFrame, so OnPropertyChanged |
+ // will only be called for |root_frame_coordination_unit|. |
+ root_frame_coordination_unit->SetProperty(mojom::PropertyType::kTest, |
+ base::Value(42)); |
+ process_coordination_unit->SetProperty(mojom::PropertyType::kTest, |
+ base::Value(42)); |
+ EXPECT_EQ(1u, observer->on_property_changed_count()); |
+ |
+ coordination_unit_manager().OnCoordinationUnitWillBeDestroyed( |
+ process_coordination_unit.get()); |
+ coordination_unit_manager().OnCoordinationUnitWillBeDestroyed( |
+ root_frame_coordination_unit.get()); |
+ coordination_unit_manager().OnCoordinationUnitWillBeDestroyed( |
+ frame_coordination_unit.get()); |
+ EXPECT_EQ(2u, observer->on_coordination_unit_will_be_destroyed_count()); |
+} |
+ |
+} // namespace resource_coordinator |