Add MemoryCoordinatorProxy::SetCurrentMemoryStateForTesting

content/browser/memory/memory_coordinator_impl.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/threading/thread_task_runner_handle.h"
 #include "content/browser/memory/memory_monitor.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/notification_types.h"
@@ skipping 96 matching lines @@
 
 void MemoryCoordinatorImpl::OnChildAdded(int render_process_id) {
   // Populate the global state as an initial state of a newly created process.
   SetMemoryState(render_process_id, ToMojomMemoryState(current_state_));
 }
 
 base::MemoryState MemoryCoordinatorImpl::GetCurrentMemoryState() const {
   return current_state_;
 }
 
+void MemoryCoordinatorImpl::SetCurrentMemoryStateForTesting(
+    base::MemoryState memory_state) {
+  // This changes the current state temporariy for testing. The state will be
+  // updated later by the task posted at ScheduleUpdateState.
+  base::MemoryState prev_state = current_state_;
+  current_state_ = memory_state;
+  if (prev_state != current_state_) {
+    NotifyStateToClients();
+    NotifyStateToChildren();
+  }
+}
+
 void MemoryCoordinatorImpl::Observe(int type,
                                     const NotificationSource& source,
                                     const NotificationDetails& details) {
   DCHECK(type == NOTIFICATION_RENDER_WIDGET_VISIBILITY_CHANGED);
   RenderWidgetHost* render_widget_host = Source<RenderWidgetHost>(source).ptr();
   RenderProcessHost* process = render_widget_host->GetProcess();
   if (!process)
     return;
   auto iter = children().find(process->GetID());
   if (iter == children().end())
@@ skipping 82 matching lines @@
 }
 
 bool MemoryCoordinatorImpl::ValidateParameters() {
   return (new_renderers_until_throttled_ > new_renderers_until_suspended_) &&
       (new_renderers_back_to_normal_ > new_renderers_back_to_throttled_) &&
       (new_renderers_back_to_normal_ > new_renderers_until_throttled_) &&
       (new_renderers_back_to_throttled_ > new_renderers_until_suspended_);
 }
 
 }  // namespace content