Add tests for MemoryCoordinator::Can{Throttle,Suspend}Renderer()

content/browser/memory/memory_coordinator.h

 // 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.
 
 #ifndef CONTENT_BROWSER_MEMORY_MEMORY_COORDINATOR_H_
 #define CONTENT_BROWSER_MEMORY_MEMORY_COORDINATOR_H_
 
 #include "base/memory/memory_coordinator_client_registry.h"
 #include "base/memory/memory_pressure_monitor.h"
 #include "base/process/process_handle.h"
@@ skipping 20 matching lines @@
   static MemoryCoordinator* GetInstance();
 
   // Starts monitoring memory usage. After calling this method, memory
   // coordinator will start dispatching state changes.
   virtual void Start() {}
 
   // Creates a handle to the provided child process.
   void CreateHandle(int render_process_id,
                     mojom::MemoryCoordinatorHandleRequest request);
 
-  // Returns number of children. Only used for testing.
-  size_t NumChildrenForTesting();
-
   // Dispatches a memory state change to the provided process. Returns true if
   // the process is tracked by this coordinator and successfully dispatches,
   // returns false otherwise.
   bool SetChildMemoryState(
       int render_process_id, mojom::MemoryState memory_state);
 
   // Returns the memory state of the specified render process. Returns UNKNOWN
   // if the process is not tracked by this coordinator.
   mojom::MemoryState GetChildMemoryState(int render_process_id) const;
 
@@ skipping 45 matching lines @@
     mojom::MemoryState memory_state;
     std::unique_ptr<MemoryCoordinatorHandleImpl> handle;
   };
 
   // A map from process ID (RenderProcessHost::GetID()) to child process info.
   using ChildInfoMap = std::map<int, ChildInfo>;
 
   ChildInfoMap& children() { return children_; }
 
 private:
+#if !defined(OS_MACOSX)
+  FRIEND_TEST_ALL_PREFIXES(MemoryCoordinatorTest, HandleAdded);
+  FRIEND_TEST_ALL_PREFIXES(MemoryCoordinatorTest, CanSuspendRenderer);
+  FRIEND_TEST_ALL_PREFIXES(MemoryCoordinatorTest, CanThrottleRenderer);
+#endif
+  FRIEND_TEST_ALL_PREFIXES(MemoryCoordinatorTest,
+                           ChildRemovedOnConnectionError);
+
+  void SetDelegateForTesting(
+      std::unique_ptr<MemoryCoordinatorDelegate> delegate);
+
   // Helper function of CreateHandle and AddChildForTesting.
   void CreateChildInfoMapEntry(
       int render_process_id,
       std::unique_ptr<MemoryCoordinatorHandleImpl> handle);
 
   // Tracks child processes. An entry is added when a renderer connects to
   // MemoryCoordinator and removed automatically when an underlying binding is
   // disconnected.
   ChildInfoMap children_;
 
   std::unique_ptr<MemoryCoordinatorDelegate> delegate_;
 
   DISALLOW_COPY_AND_ASSIGN(MemoryCoordinator);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_MEMORY_MEMORY_COORDINATOR_H_