Add BumpPtrAllocator::allocateCopy() utilities

Makes it easy to use BumpPtrAllocator to make a copy of StringRef strings.


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@200331 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Nick Kledzik
2014-01-28 19:21:27 +00:00
parent 3dcb2a2d92
commit 1c71a20f32
2 changed files with 64 additions and 0 deletions

View File

@ -147,4 +147,32 @@ TEST(AllocatorTest, TestBigAlignment) {
EXPECT_LE(Ptr + 3000, ((uintptr_t)Slab) + Slab->Size);
}
TEST(AllocatorTest, CopyStringRef) {
BumpPtrAllocator Alloc;
StringRef Str1 = "hello";
StringRef Str2 = "bye";
StringRef Str1c = Alloc.allocateCopy(Str1);
StringRef Str2c = Alloc.allocateCopy(Str2);
EXPECT_TRUE(Str1.equals(Str1c));
EXPECT_NE(Str1.data(), Str1c.data());
EXPECT_TRUE(Str2.equals(Str2c));
EXPECT_NE(Str2.data(), Str2c.data());
}
TEST(AllocatorTest, CopyArrayRef) {
BumpPtrAllocator Alloc;
static const uint16_t Words1[] = { 1, 4, 200, 37 };
ArrayRef<uint16_t> Array1 = makeArrayRef(Words1, 4);
static const uint16_t Words2[] = { 11, 4003, 67, 64000, 13 };
ArrayRef<uint16_t> Array2 = makeArrayRef(Words2, 5);
ArrayRef<uint16_t> Array1c = Alloc.allocateCopy(Array1);
ArrayRef<uint16_t> Array2c = Alloc.allocateCopy(Array2);
EXPECT_TRUE(Array1.equals(Array1c));
EXPECT_NE(Array1.data(), Array1c.data());
EXPECT_TRUE(Array2.equals(Array2c));
EXPECT_NE(Array2.data(), Array2c.data());
}
} // anonymous namespace