test(db): add more meta test

Former-commit-id: b8078e972d1c08b38cd6ff272ec2cd6ab1d895b2

cpp/unittest/db/CMakeLists.txt

@@ -27,8 +27,8 @@ set(db_test_src
     ${wrapper_src}
     ${require_files}
     utils.cpp
+    db_tests.cpp
     meta_tests.cpp)
-    # db_tests.cpp)
 
 cuda_add_executable(db_test ${db_test_src})
 

cpp/unittest/db/db_tests.cpp

@@ -65,7 +65,7 @@ TEST_F(DBTest, DB_TEST) {
         std::stringstream ss;
         long count = 0;
 
-        for (auto j=0; j<8; ++j) {
+        for (auto j=0; j<15; ++j) {
             ss.str("");
             db->count(group_name, count);
 

cpp/unittest/db/meta_tests.cpp

@@ -13,22 +13,22 @@
 
 using namespace zilliz::vecwise::engine;
 
-TEST_F(DBTest, META_TEST) {
-    auto impl = DBMetaImplFactory::Build();
+TEST_F(MetaTest, META_TEST) {
+    /* auto impl = DBMetaImplFactory::Build(); */
 
     auto group_id = "meta_test_group";
 
     meta::GroupSchema group;
     group.group_id = group_id;
-    auto status = impl->add_group(group);
+    auto status = impl_->add_group(group);
     ASSERT_TRUE(status.ok());
 
-    status = impl->get_group(group);
+    status = impl_->get_group(group);
     ASSERT_TRUE(status.ok());
 
     group.group_id = "not_found";
-    status = impl->get_group(group);
+    status = impl_->get_group(group);
     ASSERT_TRUE(!status.ok());
 
-    impl->drop_all();
+    /* impl->drop_all(); */
 }

cpp/unittest/db/utils.cpp

@@ -8,6 +8,7 @@
 #include <easylogging++.h>
 
 #include "utils.h"
+#include "db/Factories.h"
 
 using namespace zilliz::vecwise;
 
@@ -25,3 +26,12 @@ void DBTest::SetUp() {
             el::ConfigurationType::Format, "[%thread-%datetime-%level]: %msg (%fbase:%line)");
     el::Loggers::reconfigureLogger("default", defaultConf);
 }
+
+void MetaTest::SetUp() {
+    DBTest::SetUp();
+    impl_ = engine::DBMetaImplFactory::Build();
+}
+
+void MetaTest::TearDown() {
+    impl_->drop_all();
+}

cpp/unittest/db/utils.h

@@ -10,6 +10,7 @@
 #include <chrono>
 
 #include "db/DB.h"
+#include "db/DBMetaImpl.h"
 
 
 #define TIMING
@@ -35,3 +36,12 @@ class DBTest : public ::testing::Test {
 protected:
     virtual void SetUp() override;
 };
+
+
+class MetaTest : public DBTest {
+protected:
+    std::shared_ptr<zilliz::vecwise::engine::meta::DBMetaImpl> impl_;
+
+    virtual void SetUp() override;
+    virtual void TearDown() override;
+};