Function TEST_F

be/src/runtime/bufferpool/suballocator-test.cc:133–167

Basic test to make sure that we can make multiple suballocations of the same size while using the expected number of buffers.

/// Basic test to make sure that we can make multiple suballocations of the same size
/// while using the expected number of buffers.
TEST_F(SuballocatorTest, SameSizeAllocations) {
  const int64_t TOTAL_MEM = TEST_BUFFER_LEN * 100;
  InitPool(TEST_BUFFER_LEN, TOTAL_MEM);
  BufferPool::ClientHandle* client;
  RegisterClient(&global_reservation_, &client);
  Suballocator allocator(buffer_pool(), client, TEST_BUFFER_LEN);
  vector<unique_ptr<Suballocation>> allocs;

  // Make suballocations smaller than the buffer size.
  const int64_t ALLOC_SIZE = TEST_BUFFER_LEN / 4;
  int64_t allocated_mem = 0;
  while (allocated_mem < TOTAL_MEM) {
    allocs.emplace_back();
    ASSERT_OK(allocator.Allocate(ALLOC_SIZE, &allocs.back()));
    ASSERT_TRUE(allocs.back() != nullptr) << ALLOC_SIZE << " " << allocated_mem << " "
                                          << global_reservation_.DebugString();
    allocated_mem += ALLOC_SIZE;
  }

  // Attempts to allocate more memory should fail gracefully.
  const int64_t MAX_ALLOC_SIZE = 1L << 24;
  for (int alloc_size = 1; alloc_size <= MAX_ALLOC_SIZE; alloc_size *= 2) {
    unique_ptr<Suballocation> failed_alloc;
    ASSERT_OK(allocator.Allocate(alloc_size, &failed_alloc));
    ASSERT_TRUE(failed_alloc == nullptr) << alloc_size << " " << allocated_mem << " "
                                         << global_reservation_.DebugString();
  }
  AssertMemoryValid(allocs);

  // Check that reservation usage matches the amount allocated.
  EXPECT_EQ(client->GetUsedReservation(), allocated_mem)
      << global_reservation_.DebugString();
  FreeAllocations(&allocator, &allocs);
  ExpectReservationUnused(client);
}

/// Check behaviour of zero-length allocation.
TEST_F(SuballocatorTest, ZeroLengthAllocation) {