388af45f18
Use a more representative test of allocating small chunks for oddly-sized (small) objects from an allocator that has a page's worth of memory. llvm-svn: 347286
61 lines
1.5 KiB
C++
61 lines
1.5 KiB
C++
//===-- allocator_test.cc -------------------------------------------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file is a part of XRay, a function call tracing system.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "xray_allocator.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
namespace __xray {
|
|
namespace {
|
|
|
|
struct TestData {
|
|
s64 First;
|
|
s64 Second;
|
|
};
|
|
|
|
TEST(AllocatorTest, Construction) { Allocator<sizeof(TestData)> A(2 << 11); }
|
|
|
|
TEST(AllocatorTest, Allocate) {
|
|
Allocator<sizeof(TestData)> A(2 << 11);
|
|
auto B = A.Allocate();
|
|
ASSERT_NE(B.Data, nullptr);
|
|
}
|
|
|
|
TEST(AllocatorTest, OverAllocate) {
|
|
Allocator<sizeof(TestData)> A(sizeof(TestData));
|
|
auto B1 = A.Allocate();
|
|
ASSERT_NE(B1.Data, nullptr);
|
|
auto B2 = A.Allocate();
|
|
ASSERT_EQ(B2.Data, nullptr);
|
|
}
|
|
|
|
struct OddSizedData {
|
|
s64 A;
|
|
s32 B;
|
|
};
|
|
|
|
TEST(AllocatorTest, AllocateBoundaries) {
|
|
Allocator<sizeof(OddSizedData)> A(GetPageSizeCached());
|
|
|
|
// Keep allocating until we hit a nullptr block.
|
|
unsigned C = 0;
|
|
auto Expected =
|
|
GetPageSizeCached() / RoundUpTo(sizeof(OddSizedData), kCacheLineSize);
|
|
for (auto B = A.Allocate(); B.Data != nullptr; B = A.Allocate(), ++C)
|
|
;
|
|
|
|
ASSERT_EQ(C, Expected);
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace __xray
|