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
|
// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "base/memory/unsafe_shared_memory_pool.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace base {
TEST(UnsafeSharedMemoryPoolTest, CreatesRegion) {
scoped_refptr<UnsafeSharedMemoryPool> pool(
base::MakeRefCounted<UnsafeSharedMemoryPool>());
auto handle = pool->MaybeAllocateBuffer(1000);
ASSERT_TRUE(handle);
EXPECT_TRUE(handle->GetRegion().IsValid());
EXPECT_TRUE(handle->GetMapping().IsValid());
}
TEST(UnsafeSharedMemoryPoolTest, ReusesRegions) {
scoped_refptr<UnsafeSharedMemoryPool> pool(
base::MakeRefCounted<UnsafeSharedMemoryPool>());
auto handle = pool->MaybeAllocateBuffer(1000u);
ASSERT_TRUE(handle);
auto id1 = handle->GetRegion().GetGUID();
// Return memory to the pool.
handle.reset();
handle = pool->MaybeAllocateBuffer(1000u);
// Should reuse the freed region.
EXPECT_EQ(id1, handle->GetRegion().GetGUID());
}
TEST(UnsafeSharedMemoryPoolTest, RespectsSize) {
scoped_refptr<UnsafeSharedMemoryPool> pool(
base::MakeRefCounted<UnsafeSharedMemoryPool>());
auto handle = pool->MaybeAllocateBuffer(1000u);
ASSERT_TRUE(handle);
EXPECT_GE(handle->GetRegion().GetSize(), 1000u);
handle = pool->MaybeAllocateBuffer(100u);
ASSERT_TRUE(handle);
EXPECT_GE(handle->GetRegion().GetSize(), 100u);
handle = pool->MaybeAllocateBuffer(1100u);
ASSERT_TRUE(handle);
EXPECT_GE(handle->GetRegion().GetSize(), 1100u);
}
} // namespace base
|
