diff --git a/internal/core/unittest/test_sealed.cpp b/internal/core/unittest/test_sealed.cpp index 0f86c79dd0..5372847808 100644 --- 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(sc); - // mcm->CreateBucket(sc.bucket_name); + auto sc = milvus::storage::MmapConfig{}; + milvus::storage::MmapManager::GetInstance().Init(sc); auto schema = std::make_shared(); 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(sc); + auto sc = milvus::storage::MmapConfig{}; + milvus::storage::MmapManager::GetInstance().Init(sc); auto schema = std::make_shared(); 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();