discardable_memory: Using mojo IPC to replace Chrome IPC

content/renderer/render_thread_impl_browsertest.cc

 // Copyright 2013 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/renderer/render_thread_impl.h"
 
 #include <stddef.h>
 #include <stdint.h>
 #include <utility>
 
 #include "base/callback.h"
 #include "base/command_line.h"
 #include "base/location.h"
 #include "base/memory/discardable_memory.h"
 #include "base/run_loop.h"
 #include "base/single_thread_task_runner.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "cc/output/buffer_to_texture_target_map.h"
+#include "components/discardable_memory/client/client_discardable_shared_memory_manager.h"
+#include "components/discardable_memory/service/discardable_shared_memory_manager.h"
 #include "content/app/mojo/mojo_init.h"
 #include "content/child/child_gpu_memory_buffer_manager.h"
 #include "content/common/in_process_child_thread_params.h"
 #include "content/common/resource_messages.h"
 #include "content/common/service_manager/child_connection.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/content_browser_client.h"
 #include "content/public/common/content_client.h"
 #include "content/public/common/content_switches.h"
 #include "content/public/common/service_manager_connection.h"
@@ skipping 342 matching lines @@
     ::testing::Combine(::testing::Values(kDisableNativeBuffers,
                                          kEnableNativeBuffers),
                        // These formats are guaranteed to work on all platforms.
                        ::testing::Values(gfx::BufferFormat::R_8,
                                          gfx::BufferFormat::BGR_565,
                                          gfx::BufferFormat::RGBA_4444,
                                          gfx::BufferFormat::RGBA_8888,
                                          gfx::BufferFormat::BGRA_8888,
                                          gfx::BufferFormat::YVU_420)));
 
+class RenderThreadImplDiscardableMemoryBrowserTest : public ContentBrowserTest {
+ public:
+  RenderThreadImplDiscardableMemoryBrowserTest()
+      : child_discardable_shared_memory_manager_(nullptr) {}
+
+  // Overridden from BrowserTestBase:
+  void SetUpCommandLine(base::CommandLine* command_line) override {
+    command_line->AppendSwitch(switches::kSingleProcess);
+  }
+  void SetUpOnMainThread() override {
+    NavigateToURL(shell(), GURL(url::kAboutBlankURL));
+    PostTaskToInProcessRendererAndWait(base::Bind(
+        &RenderThreadImplDiscardableMemoryBrowserTest::SetUpOnRenderThread,
+        base::Unretained(this)));
+  }
+
+  discardable_memory::ClientDiscardableSharedMemoryManager*
+  child_discardable_shared_memory_manager() {
+    return child_discardable_shared_memory_manager_;
+  }
+
+ private:
+  void SetUpOnRenderThread() {
+    child_discardable_shared_memory_manager_ =
+        RenderThreadImpl::current()->discardable_shared_memory_manager();
+  }
+
+  discardable_memory::ClientDiscardableSharedMemoryManager*
+      child_discardable_shared_memory_manager_;
+};
+
+IN_PROC_BROWSER_TEST_F(RenderThreadImplDiscardableMemoryBrowserTest,
+                       LockDiscardableMemory) {
+  const size_t kSize = 1024 * 1024;  // 1MiB.
+
+  std::unique_ptr<base::DiscardableMemory> memory =
+      child_discardable_shared_memory_manager()
+          ->AllocateLockedDiscardableMemory(kSize);
+
+  ASSERT_TRUE(memory);
+  void* addr = memory->data();
+  ASSERT_NE(nullptr, addr);
+
+  memory->Unlock();
+
+  // Purge all unlocked memory.
+  discardable_memory::DiscardableSharedMemoryManager::current()->SetMemoryLimit(
+      0);
+
+  // Should fail as memory should have been purged.
+  EXPECT_FALSE(memory->Lock());
+}
+
+IN_PROC_BROWSER_TEST_F(RenderThreadImplDiscardableMemoryBrowserTest,
+                       DiscardableMemoryAddressSpace) {
+  const size_t kLargeSize = 4 * 1024 * 1024;   // 4MiB.
+  const size_t kNumberOfInstances = 1024 + 1;  // >4GiB total.
+
+  ScopedVector<base::DiscardableMemory> instances;
+  for (size_t i = 0; i < kNumberOfInstances; ++i) {
+    std::unique_ptr<base::DiscardableMemory> memory =
+        child_discardable_shared_memory_manager()
+            ->AllocateLockedDiscardableMemory(kLargeSize);
+    ASSERT_TRUE(memory);
+    void* addr = memory->data();
+    ASSERT_NE(nullptr, addr);
+    memory->Unlock();
+    instances.push_back(std::move(memory));
+  }
+}
+
+IN_PROC_BROWSER_TEST_F(RenderThreadImplDiscardableMemoryBrowserTest,
+                       ReleaseFreeDiscardableMemory) {
+  const size_t kSize = 1024 * 1024;  // 1MiB.
+
+  std::unique_ptr<base::DiscardableMemory> memory =
+      child_discardable_shared_memory_manager()
+          ->AllocateLockedDiscardableMemory(kSize);
+
+  EXPECT_TRUE(memory);
+  memory.reset();
+
+  EXPECT_GE(discardable_memory::DiscardableSharedMemoryManager::current()
+                ->GetBytesAllocated(),
+            kSize);
+
+  child_discardable_shared_memory_manager()->ReleaseFreeMemory();
+
+  // Busy wait for host memory usage to be reduced.

[dcheng, 2016/11/25 00:07:08]

Is there a way to do this? This is generally a testing anti-pattern

[Peng, 2016/11/25 16:41:53]

I don't see a easy way to avoid this busy waiting.

+  base::TimeTicks end =
+      base::TimeTicks::Now() + base::TimeDelta::FromSeconds(5);
+  while (base::TimeTicks::Now() < end) {
+    if (!discardable_memory::DiscardableSharedMemoryManager::current()
+             ->GetBytesAllocated())
+      break;
+  }
+
+  EXPECT_LT(base::TimeTicks::Now(), end);
+}
+
 }  // namespace
 }  // namespace content