Fix unit test.

infiniflow · Dec 11, 2024 · e66a36c · e66a36c
1 parent 8ba671b
commit e66a36c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/unit_test/storage/buffer/buffer_manager.cpp b/src/unit_test/storage/buffer/buffer_manager.cpp
@@ -116,6 +116,7 @@ TEST_F(BufferManagerTest, cleanup_test) {
             auto file_name = MakeShared<String>(fmt::format("file_{}", i));
             auto file_worker = MakeUnique<DataFileWorker>(data_dir_, temp_dir_, MakeShared<String>(""), file_name, file_size, buffer_mgr.persistence_manager());
             auto *buffer_obj = buffer_mgr.AllocateBufferObject(std::move(file_worker));
+            buffer_obj->AddObjRc();
             buffer_objs.push_back(buffer_obj);
             {
                 auto buffer_handle = buffer_obj->Load();
@@ -374,6 +375,7 @@ class Test1Obj : public TestObj {
             file_info.file_size_ = file_size;
             auto file_worker = MakeUnique<DataFileWorker>(data_dir_, temp_dir_, MakeShared<String>(""), file_name, file_size, nullptr);
             file_info.buffer_obj_ = buffer_mgr_->AllocateBufferObject(std::move(file_worker));
+            file_info.buffer_obj_->AddObjRc();
         } else {
             auto file_worker = MakeUnique<DataFileWorker>(data_dir_, temp_dir_, MakeShared<String>(""), file_name, file_info.file_size_, nullptr);
             file_info.buffer_obj_ = buffer_mgr_->GetBufferObject(std::move(file_worker));
@@ -449,6 +451,7 @@ class Test2Obj : public TestObj {
         if (alloc_new) {
             auto file_worker = MakeUnique<VarFileWorker>(data_dir_, temp_dir_, MakeShared<String>(""), file_name, 0, nullptr);
             file_info.buffer_obj_ = buffer_mgr_->AllocateBufferObject(std::move(file_worker));
+            file_info.buffer_obj_->AddObjRc();
         } else {
             auto file_worker = MakeUnique<VarFileWorker>(data_dir_, temp_dir_, MakeShared<String>(""), file_name, file_info.file_size_, nullptr);
             file_info.buffer_obj_ = buffer_mgr_->GetBufferObject(std::move(file_worker));