test(db): add utils in db test

Former-commit-id: 0e65a0a885cc3ead0fb233bd041bb4f8b8645898

cpp/unittest/db/CMakeLists.txt

@@ -26,6 +26,7 @@ set(db_test_src
     ${db_srcs}
     ${wrapper_src}
     ${require_files}
+    utils.cpp
     db_tests.cpp)
 
 cuda_add_executable(db_test ${db_test_src})

cpp/unittest/db/db_tests.cpp

@@ -6,48 +6,12 @@
 #include <gtest/gtest.h>
 #include <thread>
 #include <easylogging++.h>
-#include <chrono>
 
+#include "utils.h"
 #include "db/DB.h"
 
 using namespace zilliz::vecwise;
 
-#define TIMING
-
-#ifdef TIMING
-#define INIT_TIMER auto start = std::chrono::high_resolution_clock::now();
-#define START_TIMER  start = std::chrono::high_resolution_clock::now();
-#define STOP_TIMER(name)  LOG(DEBUG) << "RUNTIME of " << name << ": " << \
-    std::chrono::duration_cast<std::chrono::milliseconds>( \
-            std::chrono::high_resolution_clock::now()-start \
-    ).count() << " ms ";
-#else
-#define INIT_TIMER
-#define START_TIMER
-#define STOP_TIMER(name)
-#endif
-
-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());
-        if(!stat.ok()) {
-            std::cout << stat.ToString() << std::endl;
-        }
-    }
-}
-
 TEST_F(DBTest, DB_TEST) {
 
     static const std::string group_name = "test_group";

cpp/unittest/db/utils.cpp

+////////////////////////////////////////////////////////////////////////////////
+// Copyright 上海赜睿信息科技有限公司(Zilliz) - All Rights Reserved
+// Unauthorized copying of this file, via any medium is strictly prohibited.
+// Proprietary and confidential.
+////////////////////////////////////////////////////////////////////////////////
+
+#include <iostream>
+#include <easylogging++.h>
+
+#include "utils.h"
+
+using namespace zilliz::vecwise;
+
+void ASSERT_STATS(engine::Status& stat) {
+    ASSERT_TRUE(stat.ok());
+    if(!stat.ok()) {
+        std::cout << stat.ToString() << std::endl;
+    }
+}
+
+void DBTest::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);
+}

cpp/unittest/db/utils.h

+////////////////////////////////////////////////////////////////////////////////
+// Copyright 上海赜睿信息科技有限公司(Zilliz) - All Rights Reserved
+// Unauthorized copying of this file, via any medium is strictly prohibited.
+// Proprietary and confidential.
+////////////////////////////////////////////////////////////////////////////////
+
+#pragma once
+
+#include <gtest/gtest.h>
+#include <chrono>
+
+#include "db/DB.h"
+
+
+#define TIMING
+
+#ifdef TIMING
+#define INIT_TIMER auto start = std::chrono::high_resolution_clock::now();
+#define START_TIMER  start = std::chrono::high_resolution_clock::now();
+#define STOP_TIMER(name)  LOG(DEBUG) << "RUNTIME of " << name << ": " << \
+    std::chrono::duration_cast<std::chrono::milliseconds>( \
+            std::chrono::high_resolution_clock::now()-start \
+    ).count() << " ms ";
+#else
+#define INIT_TIMER
+#define START_TIMER
+#define STOP_TIMER(name)
+#endif
+
+
+void ASSERT_STATS(zilliz::vecwise::engine::Status& stat);
+
+
+class DBTest : public ::testing::Test {
+protected:
+    virtual void SetUp() override;
+};