fix unittest

infiniflow · Dec 22, 2023 · f984773 · f984773
1 parent 5a35f24
commit f984773
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 2 deletions.
diff --git a/src/unit_test/storage/buffer/buffer_handle.cpp b/src/unit_test/storage/buffer/buffer_handle.cpp
@@ -20,8 +20,26 @@ import buffer_manager;
 import data_file_worker;
 import buffer_obj;
 import infinity_exception;
+import global_resource_usage;
+import infinity_context;
+
+class BufferHandleTest : public BaseTest {
+    void SetUp() override {
+        BaseTest::SetUp();
+        system("rm -rf /tmp/infinity/log /tmp/infinity/data /tmp/infinity/wal");
+        infinity::GlobalResourceUsage::Init();
+        std::shared_ptr<std::string> config_path = nullptr;
+        infinity::InfinityContext::instance().Init(config_path);
+    }
 
-class BufferHandleTest : public BaseTest {};
+    void TearDown() override {
+        infinity::InfinityContext::instance().UnInit();
+        EXPECT_EQ(infinity::GlobalResourceUsage::GetObjectCount(), 0);
+        EXPECT_EQ(infinity::GlobalResourceUsage::GetRawMemoryCount(), 0);
+        infinity::GlobalResourceUsage::UnInit();
+        BaseTest::TearDown();
+    }
+};
 
 TEST_F(BufferHandleTest, test1) {
     using namespace infinity;

diff --git a/src/unit_test/storage/buffer/buffer_obj.cpp b/src/unit_test/storage/buffer/buffer_obj.cpp
@@ -22,9 +22,25 @@ import buffer_manager;
 import buffer_handle;
 import buffer_obj;
 import data_file_worker;
+import global_resource_usage;
+import infinity_context;
 
 class BufferObjTest : public BaseTest {
-    void SetUp() override { system("rm -rf /tmp/infinity"); }
+    void SetUp() override {
+        BaseTest::SetUp();
+        system("rm -rf /tmp/infinity/log /tmp/infinity/data /tmp/infinity/wal");
+        infinity::GlobalResourceUsage::Init();
+        std::shared_ptr<std::string> config_path = nullptr;
+        infinity::InfinityContext::instance().Init(config_path);
+    }
+
+    void TearDown() override {
+        infinity::InfinityContext::instance().UnInit();
+        EXPECT_EQ(infinity::GlobalResourceUsage::GetObjectCount(), 0);
+        EXPECT_EQ(infinity::GlobalResourceUsage::GetRawMemoryCount(), 0);
+        infinity::GlobalResourceUsage::UnInit();
+        BaseTest::TearDown();
+    }
 };
 
 // Test status transfer of buffer handle.