| Index: content/child/child_thread_impl_browsertest.cc
|
| diff --git a/content/child/child_thread_impl_browsertest.cc b/content/child/child_thread_impl_browsertest.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..ab21ddd453db3b28c0b4f633d8fea71ae7c32a29
|
| --- /dev/null
|
| +++ b/content/child/child_thread_impl_browsertest.cc
|
| @@ -0,0 +1,130 @@
|
| +// Copyright 2015 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/child/child_thread_impl.h"
|
| +
|
| +#include <stddef.h>
|
| +#include <string.h>
|
| +#include <utility>
|
| +
|
| +#include "base/bind.h"
|
| +#include "base/command_line.h"
|
| +#include "base/memory/discardable_memory.h"
|
| +#include "base/memory/scoped_vector.h"
|
| +#include "base/time/time.h"
|
| +#include "components/discardable_memory/client/client_discardable_shared_memory_manager.h"
|
| +#include "components/discardable_memory/service/discardable_shared_memory_manager.h"
|
| +#include "content/child/child_gpu_memory_buffer_manager.h"
|
| +#include "content/public/common/content_switches.h"
|
| +#include "content/public/test/content_browser_test.h"
|
| +#include "content/public/test/content_browser_test_utils.h"
|
| +#include "content/shell/browser/shell.h"
|
| +#include "gpu/ipc/client/gpu_memory_buffer_impl.h"
|
| +#include "ui/gfx/buffer_format_util.h"
|
| +#include "url/gurl.h"
|
| +
|
| +namespace content {
|
| +namespace {
|
| +
|
| +class ChildThreadImplBrowserTest : public ContentBrowserTest {
|
| + public:
|
| + ChildThreadImplBrowserTest()
|
| + : 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(&ChildThreadImplBrowserTest::SetUpOnChildThread,
|
| + base::Unretained(this)));
|
| + }
|
| +
|
| + discardable_memory::ClientDiscardableSharedMemoryManager*
|
| + child_discardable_shared_memory_manager() {
|
| + return child_discardable_shared_memory_manager_;
|
| + }
|
| +
|
| + private:
|
| + void SetUpOnChildThread() {
|
| + child_discardable_shared_memory_manager_ =
|
| + ChildThreadImpl::current()->discardable_shared_memory_manager();
|
| + }
|
| +
|
| + discardable_memory::ClientDiscardableSharedMemoryManager*
|
| + child_discardable_shared_memory_manager_;
|
| +};
|
| +
|
| +IN_PROC_BROWSER_TEST_F(ChildThreadImplBrowserTest, 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(ChildThreadImplBrowserTest,
|
| + 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(ChildThreadImplBrowserTest,
|
| + 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.
|
| + 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
|
|
|