admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-07 19:31:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

enhance: Reopen chunk cache cpp ut (#33622)

Browse Source 
issue: https://github.com/milvus-io/milvus/issues/33210

---------

Signed-off-by: bigsheeper <yihao.dai@zilliz.com>
This commit is contained in:
yihao.dai 2024-09-28 18:19:15 +08:00 committed by GitHub
parent 9e3efa06be
commit 8ed34dce84
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
internal/core/unittest/test_sealed.cpp
View File

@ -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();