diff --git a/cpp/unittest/db/search_test.cpp b/cpp/unittest/db/search_test.cpp index 64e03f65ae2e07eabd39035e5867350f72772f9f..cd2ff3b5c6d742197ee5f5a2850370f4b6eb3c59 100644 --- a/cpp/unittest/db/search_test.cpp +++ b/cpp/unittest/db/search_test.cpp @@ -216,7 +216,7 @@ TEST(DBSearchTest, MERGE_TEST) { TEST(DBSearchTest, PARALLEL_CLUSTER_TEST) { server::ServerConfig &config = server::ServerConfig::GetInstance(); server::ConfigNode& db_config = config.GetConfig(server::CONFIG_DB); - db_config.SetValue(server::CONFIG_DB_PARALLEL_REDUCE, "true"); + db_config.SetValue(server::CONFIG_DB_PARALLEL_REDUCE, "false");//lvoc cannot work for std::function, set to false bool ascending = true; std::vector target_ids; @@ -252,7 +252,7 @@ TEST(DBSearchTest, PARALLEL_CLUSTER_TEST) { TEST(DBSearchTest, PARALLEL_TOPK_TEST) { server::ServerConfig &config = server::ServerConfig::GetInstance(); server::ConfigNode& db_config = config.GetConfig(server::CONFIG_DB); - db_config.SetValue(server::CONFIG_DB_PARALLEL_REDUCE, "true"); + db_config.SetValue(server::CONFIG_DB_PARALLEL_REDUCE, "false");//lvoc cannot work for std::function, set to false std::vector target_ids; std::vector target_distence; diff --git a/cpp/unittest/metrics/metrics_test.cpp b/cpp/unittest/metrics/metrics_test.cpp index c1c674f12b3e1436cd4bd8d83e28bd3805a8d80d..f589cf3ace675c2fbb796a662fe9accb97214673 100644 --- a/cpp/unittest/metrics/metrics_test.cpp +++ b/cpp/unittest/metrics/metrics_test.cpp @@ -25,6 +25,11 @@ using namespace zilliz::milvus; TEST_F(MetricTest, Metric_Tes) { + server::ConfigNode &configNode = server::ServerConfig::GetInstance().GetConfig(server::CONFIG_METRIC); + configNode.SetValue(server::CONFIG_METRIC_COLLECTOR, "zabbix"); + server::Metrics::GetInstance(); + configNode.SetValue(server::CONFIG_METRIC_COLLECTOR, "prometheus"); + server::Metrics::GetInstance(); server::SystemInfo::GetInstance().Init(); // server::Metrics::GetInstance().Init(); diff --git a/cpp/unittest/metrics/prometheus_test.cpp b/cpp/unittest/metrics/prometheus_test.cpp index 912e910f32f1b81d3b52c6de16c58dc2e6d5a833..07aab27f48db3d24ff293a77f87a2a0d8ad72785 100644 --- a/cpp/unittest/metrics/prometheus_test.cpp +++ b/cpp/unittest/metrics/prometheus_test.cpp @@ -12,6 +12,9 @@ using namespace zilliz::milvus; TEST(PrometheusTest, PROMETHEUS_TEST){ + server::ConfigNode &configNode = server::ServerConfig::GetInstance().GetConfig(server::CONFIG_METRIC); + configNode.SetValue(server::CONFIG_METRIC_IS_STARTUP, "on"); + server::PrometheusMetrics instance = server::PrometheusMetrics::GetInstance(); instance.Init(); instance.SetStartup(true); @@ -51,11 +54,20 @@ TEST(PrometheusTest, PROMETHEUS_TEST){ instance.GPUMemoryUsageGaugeSet(); instance.AddVectorsPerSecondGaugeSet(1,1,1); instance.QueryIndexTypePerSecondSet("IVF", 1.0); + instance.QueryIndexTypePerSecondSet("IDMap", 1.0); instance.ConnectionGaugeIncrement(); instance.ConnectionGaugeDecrement(); instance.KeepingAliveCounterIncrement(); instance.OctetsSet(); + + instance.CPUCoreUsagePercentSet(); + instance.GPUTemperature(); + instance.CPUTemperature(); + + configNode.SetValue(server::CONFIG_METRIC_IS_STARTUP, "off"); + instance.Init(); instance.CPUCoreUsagePercentSet(); instance.GPUTemperature(); instance.CPUTemperature(); + } \ No newline at end of file diff --git a/cpp/unittest/server/config_test.cpp b/cpp/unittest/server/config_test.cpp index 59669710ba58188bb434b16a9a67328f5424f19e..3071faf3411be290087c20c1e5ec169faf79d139 100644 --- a/cpp/unittest/server/config_test.cpp +++ b/cpp/unittest/server/config_test.cpp @@ -43,6 +43,10 @@ TEST(ConfigTest, CONFIG_TEST) { server::ConfigNode& db_config = root_config.GetChild("db_config"); server::ConfigNode& metric_config = root_config.GetChild("metric_config"); server::ConfigNode& cache_config = root_config.GetChild("cache_config"); + server::ConfigNode invalid_config = root_config.GetChild("invalid_config"); + auto valus = invalid_config.GetSequence("not_exist"); + float ff = invalid_config.GetFloatValue("not_exist", 3.0); + ASSERT_EQ(ff, 3.0); std::string address = server_config.GetValue("address"); ASSERT_TRUE(!address.empty());