Add MemoryCoordinatorProxy::SetCurrentMemoryStateForTesting

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_proxy.h"
 #include "base/run_loop.h"
 #include "content/browser/memory/memory_monitor.h"
 #include "testing/gtest/include/gtest/gtest.h"
@@ skipping 42 matching lines @@
   void SetUp() override {
     MemoryCoordinator::EnableFeaturesForTesting();
 
     coordinator_.reset(new MemoryCoordinatorImpl(
         message_loop_.task_runner(), base::WrapUnique(new MockMemoryMonitor)));
 
     base::MemoryCoordinatorProxy::GetInstance()->
         SetGetCurrentMemoryStateCallback(base::Bind(
             &MemoryCoordinator::GetCurrentMemoryState,
             base::Unretained(coordinator_.get())));
+    base::MemoryCoordinatorProxy::GetInstance()->
+        SetSetCurrentMemoryStateForTestingCallback(base::Bind(
+            &MemoryCoordinator::SetCurrentMemoryStateForTesting,
+            base::Unretained(coordinator_.get())));
   }
 
   MockMemoryMonitor* GetMockMemoryMonitor() {
     return static_cast<MockMemoryMonitor*>(coordinator_->memory_monitor());
   }
 
  protected:
   std::unique_ptr<MemoryCoordinatorImpl> coordinator_;
   base::MessageLoop message_loop_;
 };
@@ skipping 87 matching lines @@
     GetMockMemoryMonitor()->SetFreeMemoryUntilCriticalMB(50);
     coordinator_->UpdateState();
     base::RunLoop loop;
     loop.RunUntilIdle();
     EXPECT_FALSE(client.is_called());
     EXPECT_EQ(base::MemoryState::NORMAL, client.state());
     base::MemoryCoordinatorClientRegistry::GetInstance()->Unregister(&client);
   }
 }
 
+TEST_F(MemoryCoordinatorImplTest, SetMemoryStateForTesting) {
+  coordinator_->expected_renderer_size_ = 10;
+  coordinator_->new_renderers_until_throttled_ = 4;
+  coordinator_->new_renderers_until_suspended_ = 2;
+  coordinator_->new_renderers_back_to_normal_ = 5;
+  coordinator_->new_renderers_back_to_throttled_ = 3;
+  DCHECK(coordinator_->ValidateParameters());
+
+  MockMemoryCoordinatorClient client;
+  base::MemoryCoordinatorClientRegistry::GetInstance()->Register(&client);
+  EXPECT_EQ(base::MemoryState::NORMAL, coordinator_->GetCurrentMemoryState());
+  EXPECT_EQ(base::MemoryState::NORMAL,
+            base::MemoryCoordinatorProxy::GetInstance()->
+                GetCurrentMemoryState());
+  EXPECT_EQ(base::MemoryState::NORMAL, client.state());
+
+  base::MemoryCoordinatorProxy::GetInstance()->SetCurrentMemoryStateForTesting(
+      base::MemoryState::THROTTLED);
+  EXPECT_EQ(base::MemoryState::THROTTLED,
+            coordinator_->GetCurrentMemoryState());
+  EXPECT_EQ(base::MemoryState::THROTTLED,
+            base::MemoryCoordinatorProxy::GetInstance()->
+                GetCurrentMemoryState());
+  base::RunLoop loop;
+  loop.RunUntilIdle();
+  EXPECT_TRUE(client.is_called());
+  EXPECT_EQ(base::MemoryState::THROTTLED, client.state());
+  base::MemoryCoordinatorClientRegistry::GetInstance()->Unregister(&client);
+}
+
 }  // namespace content