blob: cc73fb0cbbfb115d8cde3b543767967ffd9a24c5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
// 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 "components/html_viewer/discardable_memory_allocator.h"
#include "base/memory/discardable_memory.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace html_viewer {
namespace {
const size_t kOneKilobyte = 1024;
const size_t kAlmostOneMegabyte = 1023 * kOneKilobyte;
const size_t kOneMegabyte = 1024 * kOneKilobyte;
TEST(DiscardableMemoryAllocator, Basic) {
scoped_ptr<base::DiscardableMemory> chunk;
{
DiscardableMemoryAllocator allocator(kOneMegabyte);
// Make sure the chunk is locked when allocated. In debug mode, we will
// dcheck.
chunk = allocator.AllocateLockedDiscardableMemory(kOneKilobyte);
chunk->Unlock();
// Make sure we can lock a chunk.
EXPECT_TRUE(chunk->Lock());
chunk->Unlock();
}
// The chunk's backing should have disappeared with the allocator.
EXPECT_FALSE(chunk->Lock());
}
TEST(DiscardableMemoryAllocator, DiscardChunks) {
DiscardableMemoryAllocator allocator(kOneMegabyte);
scoped_ptr<base::DiscardableMemory> chunk_to_remove =
allocator.AllocateLockedDiscardableMemory(kAlmostOneMegabyte);
chunk_to_remove->Unlock();
// Allocating a second chunk should deallocate the first one due to memory
// pressure, since we only have one megabyte available.
scoped_ptr<base::DiscardableMemory> chunk_to_keep =
allocator.AllocateLockedDiscardableMemory(kAlmostOneMegabyte);
// Fail to get a lock because allocating the second chunk removed the first.
EXPECT_FALSE(chunk_to_remove->Lock());
chunk_to_keep->Unlock();
}
TEST(DiscardableMemoryAllocator, DontDiscardLiveChunks) {
DiscardableMemoryAllocator allocator(kOneMegabyte);
scoped_ptr<base::DiscardableMemory> chunk_one =
allocator.AllocateLockedDiscardableMemory(kAlmostOneMegabyte);
scoped_ptr<base::DiscardableMemory> chunk_two =
allocator.AllocateLockedDiscardableMemory(kAlmostOneMegabyte);
scoped_ptr<base::DiscardableMemory> chunk_three =
allocator.AllocateLockedDiscardableMemory(kAlmostOneMegabyte);
// These accesses will fail if the underlying weak ptr has been deallocated.
EXPECT_NE(nullptr, chunk_one->data());
EXPECT_NE(nullptr, chunk_two->data());
EXPECT_NE(nullptr, chunk_three->data());
chunk_one->Unlock();
chunk_two->Unlock();
chunk_three->Unlock();
}
} // namespace
} // namespace html_viewer
|
