Add a public interface for content::MemoryCoordinator

content/browser/memory/memory_coordinator_impl_unittest.cc

 // Copyright 2016 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 "content/browser/memory/memory_coordinator_impl.h"
 
 #include "base/memory/memory_coordinator_client_registry.h"
 #include "base/memory/memory_coordinator_proxy.h"
 #include "base/memory/memory_pressure_monitor.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
+#include "base/test/multiprocess_test.h"
 #include "base/test/scoped_feature_list.h"
 #include "base/test/test_mock_time_task_runner.h"
 #include "content/browser/memory/memory_monitor.h"
 #include "content/browser/memory/memory_state_updater.h"
 #include "content/public/common/content_features.h"
 #include "content/public/test/mock_render_process_host.h"
 #include "content/public/test/test_browser_context.h"
 #include "mojo/public/cpp/bindings/binding.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
@@ skipping 132 matching lines @@
     return result;
   }
 
   TestBrowserContext browser_context_;
   std::vector<std::unique_ptr<Child>> children_;
   std::map<int, std::unique_ptr<MockRenderProcessHost>> render_process_hosts_;
 };
 
 }  // namespace
 
-class MemoryCoordinatorImplTest : public testing::Test {
+class MemoryCoordinatorImplTest : public base::MultiProcessTest {
  public:
   using MemoryState = base::MemoryState;
 
   void SetUp() override {
+    base::MultiProcessTest::SetUp();
     scoped_feature_list_.InitAndEnableFeature(features::kMemoryCoordinator);
 
     task_runner_ = new base::TestMockTimeTaskRunner();
     coordinator_.reset(new TestMemoryCoordinatorImpl(task_runner_));
 
     base::MemoryCoordinatorProxy::GetInstance()->
         SetGetCurrentMemoryStateCallback(base::Bind(
             &MemoryCoordinatorImpl::GetCurrentMemoryState,
             base::Unretained(coordinator_.get())));
     base::MemoryCoordinatorProxy::GetInstance()->
@@ skipping 284 matching lines @@
 
   // Also make sure that the state is updated based on free avaiable memory.
   // Since the global state has changed in the previous task, we have to wait
   // for |minimum_transition|.
   GetMockMemoryMonitor()->SetFreeMemoryUntilCriticalMB(40);
   task_runner_->FastForwardBy(minimum_transition);
   task_runner_->RunUntilIdle();
   EXPECT_EQ(base::MemoryState::THROTTLED, coordinator_->GetGlobalMemoryState());
 }
 
+TEST_F(MemoryCoordinatorImplTest, GetStateForProcess) {
+  EXPECT_EQ(base::MemoryState::UNKNOWN,
+            coordinator_->GetStateForProcess(base::kNullProcessHandle));
+  EXPECT_EQ(base::MemoryState::NORMAL,
+            coordinator_->GetStateForProcess(base::GetCurrentProcessHandle()));
+
+  coordinator_->CreateChildMemoryCoordinator(1);
+  coordinator_->CreateChildMemoryCoordinator(2);
+  base::Process process1 = SpawnChild("process1");
+  base::Process process2 = SpawnChild("process2");
+  coordinator_->GetMockRenderProcessHost(1)->SetProcessHandle(
+      base::MakeUnique<base::ProcessHandle>(process1.Handle()));
+  coordinator_->GetMockRenderProcessHost(2)->SetProcessHandle(
+      base::MakeUnique<base::ProcessHandle>(process2.Handle()));
+
+  EXPECT_EQ(base::MemoryState::NORMAL,
+            coordinator_->GetStateForProcess(process1.Handle()));
+  EXPECT_EQ(base::MemoryState::NORMAL,
+            coordinator_->GetStateForProcess(process2.Handle()));
+
+  EXPECT_TRUE(
+      coordinator_->SetChildMemoryState(1, MemoryState::THROTTLED));
+  EXPECT_EQ(base::MemoryState::THROTTLED,
+            coordinator_->GetStateForProcess(process1.Handle()));
+  EXPECT_EQ(base::MemoryState::NORMAL,
+            coordinator_->GetStateForProcess(process2.Handle()));
+}
+
 }  // namespace content