/*
- * Copyright 2012 Facebook, Inc.
+ * Copyright 2017 Facebook, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* limitations under the License.
*/
-#include "folly/Arena.h"
-#include "folly/StlAllocator.h"
+#include <folly/Arena.h>
+#include <folly/Memory.h>
+#include <folly/portability/GTest.h>
#include <set>
#include <vector>
#include <glog/logging.h>
-#include <gtest/gtest.h>
using namespace folly;
+static_assert(IsArenaAllocator<SysArena>::value, "");
+
TEST(Arena, SizeSanity) {
std::set<size_t*> allocatedItems;
size_t* ptr = static_cast<size_t*>(arena.allocate(sizeof(long)));
allocatedItems.insert(ptr);
minimum_size += requestedBlockSize;
- maximum_size += goodMallocSize(requestedBlockSize + 1);
+ maximum_size += goodMallocSize(requestedBlockSize + SysArena::kBlockOverhead);
EXPECT_TRUE(arena.totalSize() >= minimum_size);
EXPECT_TRUE(arena.totalSize() <= maximum_size);
VLOG(4) << minimum_size << " < " << arena.totalSize() << " < "
allocatedItems.insert(ptr);
}
minimum_size += 10 * requestedBlockSize;
- maximum_size += 10 * goodMallocSize(requestedBlockSize + 1);
+ maximum_size += 10 * goodMallocSize(requestedBlockSize
+ + SysArena::kBlockOverhead);
EXPECT_TRUE(arena.totalSize() >= minimum_size);
EXPECT_TRUE(arena.totalSize() <= maximum_size);
VLOG(4) << minimum_size << " < " << arena.totalSize() << " < "
ptr = static_cast<size_t*>(arena.allocate(10 * requestedBlockSize));
allocatedItems.insert(ptr);
minimum_size += 10 * requestedBlockSize;
- maximum_size += goodMallocSize(10 * requestedBlockSize + 1);
+ maximum_size += goodMallocSize(10 * requestedBlockSize
+ + SysArena::kBlockOverhead);
EXPECT_TRUE(arena.totalSize() >= minimum_size);
EXPECT_TRUE(arena.totalSize() <= maximum_size);
VLOG(4) << minimum_size << " < " << arena.totalSize() << " < "
}
}
+TEST(Arena, SizeLimit) {
+ static const size_t requestedBlockSize = sizeof(size_t);
+ static const size_t maxSize = 10 * requestedBlockSize;
+
+ SysArena arena(requestedBlockSize, maxSize);
+
+ void* a = arena.allocate(sizeof(size_t));
+ EXPECT_TRUE(a != nullptr);
+ EXPECT_THROW(arena.allocate(maxSize + 1), std::bad_alloc);
+}
+
+TEST(Arena, MoveArena) {
+ SysArena arena(sizeof(size_t) * 2);
+ arena.allocate(sizeof(size_t));
+ auto totalSize = arena.totalSize();
+ auto bytesUsed = arena.bytesUsed();
+
+ SysArena moved(std::move(arena));
+ EXPECT_EQ(totalSize, moved.totalSize());
+ EXPECT_EQ(bytesUsed, moved.bytesUsed());
+}
+
int main(int argc, char *argv[]) {
testing::InitGoogleTest(&argc, argv);
- google::ParseCommandLineFlags(&argc, &argv, true);
+ gflags::ParseCommandLineFlags(&argc, &argv, true);
auto ret = RUN_ALL_TESTS();
return ret;
}