From 7e00f351e1e51f96a4b129931f1b3aa44e37c6e5 Mon Sep 17 00:00:00 2001
From: Xu Peng <xupeng3112@163.com>
Date: Fri, 26 Apr 2019 17:59:51 +0800
Subject: [PATCH] test(db): change log format for unittest

Former-commit-id: a364d3800fbefbd9287df906b5e69f4ec89d1aa7
---
 cpp/unittest/db/db_tests.cpp | 25 +++++++++++++++++++------
 1 file changed, 19 insertions(+), 6 deletions(-)

diff --git a/cpp/unittest/db/db_tests.cpp b/cpp/unittest/db/db_tests.cpp
index b2252c22..16236f6f 100644
--- a/cpp/unittest/db/db_tests.cpp
+++ b/cpp/unittest/db/db_tests.cpp
@@ -15,6 +15,18 @@
 
 using namespace zilliz::vecwise;
 
+class DBTest : public ::testing::Test {
+protected:
+    virtual void SetUp() {
+        el::Configurations defaultConf;
+        defaultConf.setToDefault();
+        defaultConf.set(el::Level::Debug,
+                el::ConfigurationType::Format, "[%thread-%datetime-%level]: %msg (%fbase:%line)");
+        el::Loggers::reconfigureLogger("default", defaultConf);
+    }
+
+};
+
 namespace {
     void ASSERT_STATS(engine::Status& stat) {
         ASSERT_TRUE(stat.ok());
@@ -24,7 +36,8 @@ namespace {
     }
 }
 
-TEST(DBTest, DB_TEST) {
+TEST_F(DBTest, DB_TEST) {
+
     static const std::string group_name = "test_group";
     static const int group_dim = 256;
 
@@ -52,7 +65,7 @@ TEST(DBTest, DB_TEST) {
     engine::IDNumbers target_ids;
 
     int d = 256;
-    int nb = 2;
+    int nb = 10;
     float *xb = new float[d * nb];
     for(int i = 0; i < nb; i++) {
         for(int j = 0; j < d; j++) xb[d * i + j] = drand48();
@@ -66,7 +79,7 @@ TEST(DBTest, DB_TEST) {
         qxb[d * i] += i / 2000.;
     }
 
-    int loop = 1000000;
+    int loop = 500000;
 
     for (auto i=0; i<loop; ++i) {
         if (i==40) {
@@ -98,9 +111,9 @@ TEST(DBTest, DB_TEST) {
     engine::DB::Open(opt, &db);
     db->drop_all();
     delete db;
-}
+};
 
-TEST(SearchTest, DB_TEST) {
+TEST_F(DBTest, SEARCH_TEST) {
     static const std::string group_name = "test_group";
     static const int group_dim = 256;
 
@@ -181,4 +194,4 @@ TEST(SearchTest, DB_TEST) {
     engine::DB::Open(opt, &db);
     db->drop_all();
     delete db;
-}
+};
-- 
GitLab