1 files changed, 80 insertions, 0 deletions
diff --git a/lib/entityx/entityx/help/Pool_test.cc b/lib/entityx/entityx/help/Pool_test.cc
new file mode 100644
index 0000000..56ca85a
--- /dev/null
+++ b/lib/entityx/entityx/help/Pool_test.cc
@@ -0,0 +1,80 @@
+/*
+ * Copyright (C) 2012-2014 Alec Thomas <alec@swapoff.org>
+ * All rights reserved.
+ *
+ * This software is licensed as described in the file COPYING, which
+ * you should have received as part of this distribution.
+ *
+ * Author: Alec Thomas <alec@swapoff.org>
+ */
+
+#define CATCH_CONFIG_MAIN
+
+#include <vector>
+#include "entityx/3rdparty/catch.hpp"
+#include "entityx/help/Pool.h"
+
+struct Position {
+  explicit Position(int *ptr = nullptr) : ptr(ptr) {
+    if (ptr) (*ptr)++;
+  }
+  ~Position() {
+    if (ptr) (*ptr)++;
+  }
+
+  float x, y;
+  int *ptr;
+};
+
+
+TEST_CASE("TestPoolReserve") {
+  entityx::Pool<Position, 8> pool;
+  REQUIRE(0 ==  pool.capacity());
+  REQUIRE(0 ==  pool.chunks());
+  pool.reserve(8);
+  REQUIRE(0 ==  pool.size());
+  REQUIRE(8 ==  pool.capacity());
+  REQUIRE(1 ==  pool.chunks());
+  pool.reserve(16);
+  REQUIRE(0 ==  pool.size());
+  REQUIRE(16 ==  pool.capacity());
+  REQUIRE(2 ==  pool.chunks());
+}
+
+TEST_CASE("TestPoolPointers") {
+  entityx::Pool<Position, 8> pool;
+  std::vector<char*> ptrs;
+  for (int i = 0; i < 4; i++) {
+    pool.expand(i * 8 + 8);
+    // NOTE: This is an attempt to ensure non-contiguous allocations from
+    // arena allocators.
+    ptrs.push_back(new char[8 * sizeof(Position)]);
+  }
+  char *p0 = static_cast<char*>(pool.get(0));
+  char *p7 = static_cast<char*>(pool.get(7));
+  char *p8 = static_cast<char*>(pool.get(8));
+  char *p16 = static_cast<char*>(pool.get(16));
+  char *p24 = static_cast<char*>(pool.get(24));
+
+  void *expected_p7 = p0 + 7 * sizeof(Position);
+  REQUIRE(expected_p7 == static_cast<void*>(p7));
+  void *extrapolated_p8 = p0 + 8 * sizeof(Position);
+  REQUIRE(extrapolated_p8 !=  static_cast<void*>(p8));
+  void *extrapolated_p16 = p8 + 8 * sizeof(Position);
+  REQUIRE(extrapolated_p16 !=  static_cast<void*>(p16));
+  void *extrapolated_p24 = p16 + 8 * sizeof(Position);
+  REQUIRE(extrapolated_p24 !=  static_cast<void*>(p24));
+}
+
+TEST_CASE("TestDeconstruct") {
+  entityx::Pool<Position, 8> pool;
+  pool.expand(8);
+
+  void *p0 = pool.get(0);
+
+  int counter = 0;
+  new(p0) Position(&counter);
+  REQUIRE(1 ==  counter);
+  pool.destroy(0);
+  REQUIRE(2 ==  counter);
+}