From a3cc7581b198d14f1ba8ed2da4945333fea61658 Mon Sep 17 00:00:00 2001
From: Axel Kohlmeyer <akohlmey@gmail.com>
Date: Tue, 8 Sep 2020 15:12:08 -0400
Subject: [PATCH] rename test for memory pools

---
 unittest/utils/CMakeLists.txt                 |  6 +-
 .../{test_my_page.cpp => test_mempool.cpp}    | 64 +++++++++++++++++++
 2 files changed, 67 insertions(+), 3 deletions(-)
 rename unittest/utils/{test_my_page.cpp => test_mempool.cpp} (81%)

diff --git a/unittest/utils/CMakeLists.txt b/unittest/utils/CMakeLists.txt
index 30a834a126..5b5b931210 100644
--- a/unittest/utils/CMakeLists.txt
+++ b/unittest/utils/CMakeLists.txt
@@ -2,9 +2,9 @@ add_executable(test_tokenizer test_tokenizer.cpp)
 target_link_libraries(test_tokenizer PRIVATE lammps GTest::GMockMain GTest::GMock GTest::GTest)
 add_test(Tokenizer test_tokenizer)
 
-add_executable(test_my_page test_my_page.cpp)
-target_link_libraries(test_my_page PRIVATE lammps GTest::GMockMain GTest::GMock GTest::GTest)
-add_test(MyPage test_my_page)
+add_executable(test_mempool test_mempool.cpp)
+target_link_libraries(test_mempool PRIVATE lammps GTest::GMockMain GTest::GMock GTest::GTest)
+add_test(MemPool test_mempool)
 
 add_executable(test_utils test_utils.cpp)
 target_link_libraries(test_utils PRIVATE lammps GTest::GMockMain GTest::GMock GTest::GTest)
diff --git a/unittest/utils/test_my_page.cpp b/unittest/utils/test_mempool.cpp
similarity index 81%
rename from unittest/utils/test_my_page.cpp
rename to unittest/utils/test_mempool.cpp
index 7104157012..11c5be50ee 100644
--- a/unittest/utils/test_my_page.cpp
+++ b/unittest/utils/test_mempool.cpp
@@ -13,6 +13,7 @@
 
 #include "lmptype.h"
 #include "my_page.h"
+#include "my_pool_chunk.h"
 #include "gmock/gmock.h"
 #include "gtest/gtest.h"
 
@@ -281,3 +282,66 @@ TEST(MyPage, bigint) {
     ASSERT_EQ(p.ndatum,133);
     ASSERT_EQ(p.nchunk,10);
 }
+
+TEST(MyPoolChunk, int) {
+    // defaults to minchunk=1, maxchunk=1, nbin=1, chunksperpage=1024, pagedelta=1
+    MyPoolChunk<int> p;
+
+    ASSERT_EQ(p.ndatum,0);
+    ASSERT_EQ(p.nchunk,0);
+    ASSERT_EQ(p.size(),0.0);
+
+    int idx=~0x0000;
+    int *iptr = p.get(idx);
+    ASSERT_NE(iptr,nullptr);
+    ASSERT_EQ(idx,0);
+
+    iptr = p.get(1,idx);
+    ASSERT_NE(iptr,nullptr);
+    ASSERT_EQ(idx,1);
+    // we have only one page allocated
+    ASSERT_EQ(p.size(),1024*sizeof(int)+1024*sizeof(int)+sizeof(void *)+sizeof(int));
+    ASSERT_EQ(p.ndatum,2);
+    ASSERT_EQ(p.nchunk,2);
+
+    p.put(0);
+    ASSERT_EQ(p.ndatum,1);
+    ASSERT_EQ(p.nchunk,1);
+
+    iptr = p.get(2,idx);
+    ASSERT_EQ(iptr,nullptr);
+    ASSERT_EQ(p.status(),3);
+    ASSERT_EQ(p.ndatum,1);
+    ASSERT_EQ(p.nchunk,1);
+}
+
+TEST(MyPoolChunk, double) {
+    // defaults to minchunk=1, maxchunk=1, nbin=1, chunksperpage=1024, pagedelta=1
+    MyPoolChunk<double> p;
+
+    ASSERT_EQ(p.ndatum,0);
+    ASSERT_EQ(p.nchunk,0);
+    ASSERT_EQ(p.size(),0.0);
+
+    int idx=~0x0000;
+    double *dptr = p.get(idx);
+    ASSERT_NE(dptr,nullptr);
+    ASSERT_EQ(idx,0);
+
+    dptr = p.get(1,idx);
+    ASSERT_NE(dptr,nullptr);
+    ASSERT_EQ(idx,1);
+    // we have only one page allocated
+    ASSERT_EQ(p.size(),1024*sizeof(int)+1024*sizeof(double)+sizeof(void *)+sizeof(int));
+
+    p.put(0);
+    ASSERT_EQ(p.ndatum,1);
+    ASSERT_EQ(p.nchunk,1);
+
+    dptr = p.get(2,idx);
+    ASSERT_EQ(dptr,nullptr);
+    ASSERT_EQ(p.status(),3);
+    ASSERT_EQ(p.ndatum,1);
+    ASSERT_EQ(p.nchunk,1);
+}
+