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


base / memory / unsafe_shared_memory_pool_unittest.cc [blame]

// 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