enhance: Reopen chunk cache cpp ut (#33622)

issue: #33210 --------- Signed-off-by: bigsheeper <yihao.dai@zilliz.com>
milvus-io · Sep 28, 2024 · 8ed34dc · 8ed34dc
1 parent 9e3efa0
commit 8ed34dc
Showing 1 changed file with 8 additions and 15 deletions.
diff --git a/internal/core/unittest/test_sealed.cpp b/internal/core/unittest/test_sealed.cpp
@@ -1429,9 +1429,6 @@ TEST(Sealed, GetVector) {
 }
 
 TEST(Sealed, GetVectorFromChunkCache) {
-    // skip test due to mem leak from AWS::InitSDK
-    return;
-
     auto dim = 16;
     auto topK = 5;
     auto N = ROW_COUNT;
@@ -1441,10 +1438,8 @@ TEST(Sealed, GetVectorFromChunkCache) {
     auto file_name = std::string(
         "sealed_test_get_vector_from_chunk_cache/insert_log/1/101/1000000");
 
-    auto sc = milvus::storage::StorageConfig{};
-    milvus::storage::RemoteChunkManagerSingleton::GetInstance().Init(sc);
-    auto mcm = std::make_unique<milvus::storage::MinioChunkManager>(sc);
-    // mcm->CreateBucket(sc.bucket_name);
+    auto sc = milvus::storage::MmapConfig{};
+    milvus::storage::MmapManager::GetInstance().Init(sc);
 
     auto schema = std::make_shared<Schema>();
     auto fakevec_id = schema->AddDebugField(
@@ -1465,7 +1460,8 @@ TEST(Sealed, GetVectorFromChunkCache) {
                                         fakevec_id,
                                         milvus::DataType::VECTOR_FLOAT,
                                         dim,
-                                        metric_type);
+                                        metric_type,
+                                        false);
 
     auto rcm = milvus::storage::RemoteChunkManagerSingleton::GetInstance()
                    .GetRemoteChunkManager();
@@ -1640,9 +1636,6 @@ TEST(Sealed, GetSparseVectorFromChunkCache) {
 }
 
 TEST(Sealed, WarmupChunkCache) {
-    // skip test due to mem leak from AWS::InitSDK
-    return;
-
     auto dim = 16;
     auto topK = 5;
     auto N = ROW_COUNT;
@@ -1653,9 +1646,8 @@ TEST(Sealed, WarmupChunkCache) {
     auto file_name = std::string(
         "sealed_test_get_vector_from_chunk_cache/insert_log/1/101/1000000");
 
-    auto sc = milvus::storage::StorageConfig{};
-    milvus::storage::RemoteChunkManagerSingleton::GetInstance().Init(sc);
-    auto mcm = std::make_unique<milvus::storage::MinioChunkManager>(sc);
+    auto sc = milvus::storage::MmapConfig{};
+    milvus::storage::MmapManager::GetInstance().Init(sc);
 
     auto schema = std::make_shared<Schema>();
     auto fakevec_id = schema->AddDebugField(
@@ -1676,7 +1668,8 @@ TEST(Sealed, WarmupChunkCache) {
                                         fakevec_id,
                                         milvus::DataType::VECTOR_FLOAT,
                                         dim,
-                                        metric_type);
+                                        metric_type,
+                                        false);
 
     auto rcm = milvus::storage::RemoteChunkManagerSingleton::GetInstance()
                    .GetRemoteChunkManager();