From 57783827fb93e651b89b62e08bec1edbcfab87da Mon Sep 17 00:00:00 2001
From: groot <yihua.mo@zilliz.com>
Date: Thu, 28 Oct 2021 18:54:35 +0800
Subject: [PATCH] Add cpp coverage (#10801)

Signed-off-by: yhmo <yihua.mo@zilliz.com>
---
 internal/core/unittest/test_index_wrapper.cpp | 21 +++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/internal/core/unittest/test_index_wrapper.cpp b/internal/core/unittest/test_index_wrapper.cpp
index c3f32bede8..b53593c426 100644
--- a/internal/core/unittest/test_index_wrapper.cpp
+++ b/internal/core/unittest/test_index_wrapper.cpp
@@ -19,6 +19,7 @@
 #include "indexbuilder/utils.h"
 #include "index/knowhere/knowhere/index/vector_index/helpers/IndexParameter.h"
 #include "index/knowhere/knowhere/index/vector_index/adapter/VectorAdapter.h"
+#include "index/knowhere/knowhere/archive/KnowhereConfig.h"
 #include "pb/index_cgo_msg.pb.h"
 #include "test_utils/DataGen.h"
 #include "test_utils/indexbuilder_test_utils.h"
@@ -32,6 +33,8 @@ class IndexWrapperTest : public ::testing::TestWithParam<Param> {
  protected:
     void
     SetUp() override {
+        milvus::engine::KnowhereConfig::SetStatisticsLevel(3);
+
         auto param = GetParam();
         index_type = param.first;
         metric_type = param.second;
@@ -124,6 +127,8 @@ TEST(IVFFLATNM, Build) {
 }
 
 TEST(IVFFLATNM, Query) {
+    milvus::engine::KnowhereConfig::SetStatisticsLevel(3);
+
     auto index_type = milvus::knowhere::IndexEnum::INDEX_FAISS_IVFFLAT;
     auto metric_type = milvus::knowhere::Metric::L2;
     auto conf = generate_conf(index_type, metric_type);
@@ -142,6 +147,13 @@ TEST(IVFFLATNM, Query) {
     auto xq_data = dataset.get_col<float>(0);
     auto xq_dataset = milvus::knowhere::GenDataset(NQ, DIM, xq_data.data());
     auto result = index->Query(xq_dataset, conf, nullptr);
+
+    index->UpdateIndexSize();
+    ASSERT_GT(index->IndexSize(), 0);
+
+    auto stats = index->GetStatistics();
+    ASSERT_TRUE(stats != nullptr);
+    index->ClearStatistics();
 }
 
 TEST(NSG, Query) {
@@ -190,6 +202,8 @@ print_query_result(const std::unique_ptr<milvus::indexbuilder::IndexWrapper::Que
 
 // test for: https://github.com/milvus-io/milvus/issues/6569
 TEST(BinIVFFlat, Build_and_Query) {
+    milvus::engine::KnowhereConfig::SetStatisticsLevel(2);
+    
     auto index_type = milvus::knowhere::IndexEnum::INDEX_FAISS_BIN_IVFFLAT;
     auto metric_type = milvus::knowhere::Metric::TANIMOTO;
     auto conf = generate_conf(index_type, metric_type);
@@ -222,6 +236,13 @@ TEST(BinIVFFlat, Build_and_Query) {
     memcpy(query_res->distances.data(), distances, sizeof(float) * nq * topk);
 
     print_query_result(query_res);
+
+    index->UpdateIndexSize();
+    ASSERT_GT(index->IndexSize(), 0);
+
+    auto stats = index->GetStatistics();
+    ASSERT_TRUE(stats != nullptr);
+    index->ClearStatistics();
 }
 
 TEST(BINIDMAP, Build) {