diff --git a/cpp/unittest/db/MySQLMetaImpl_test.cpp b/cpp/unittest/db/MySQLMetaImpl_test.cpp index 93e0ce9b28ed198fbbe11351b84a41af241274f1..a370fd1cd9e0707577aed2afcedc349e173c523d 100644 --- a/cpp/unittest/db/MySQLMetaImpl_test.cpp +++ b/cpp/unittest/db/MySQLMetaImpl_test.cpp @@ -32,12 +32,19 @@ using namespace zilliz::milvus::engine; //} TEST_F(MySQLTest, core) { -// DBMetaOptions options; + DBMetaOptions options; // //dialect+driver://username:password@host:port/database // options.backend_uri = "mysql://root:1234@:/test"; // options.path = "/tmp/vecwise_test"; + try { + options = getDBMetaOptions(); + } catch(std::exception& ex) { + ASSERT_TRUE(false); + return; + } + int mode = Options::MODE::SINGLE; - meta::MySQLMetaImpl impl(getDBMetaOptions(), mode); + meta::MySQLMetaImpl impl(options, mode); // auto status = impl.Initialize(); // ASSERT_TRUE(status.ok()); @@ -192,9 +199,16 @@ TEST_F(MySQLTest, core) { } TEST_F(MySQLTest, GROUP_TEST) { + DBMetaOptions options; + try { + options = getDBMetaOptions(); + } catch(std::exception& ex) { + ASSERT_TRUE(false); + return; + } int mode = Options::MODE::SINGLE; - meta::MySQLMetaImpl impl(getDBMetaOptions(), mode); + meta::MySQLMetaImpl impl(options, mode); auto table_id = "meta_test_group"; @@ -228,9 +242,16 @@ TEST_F(MySQLTest, GROUP_TEST) { } TEST_F(MySQLTest, table_file_TEST) { + DBMetaOptions options; + try { + options = getDBMetaOptions(); + } catch(std::exception& ex) { + ASSERT_TRUE(false); + return; + } int mode = Options::MODE::SINGLE; - meta::MySQLMetaImpl impl(getDBMetaOptions(), mode); + meta::MySQLMetaImpl impl(options, mode); auto table_id = "meta_test_group"; @@ -296,7 +317,14 @@ TEST_F(MySQLTest, table_file_TEST) { TEST_F(MySQLTest, ARCHIVE_TEST_DAYS) { srand(time(0)); - DBMetaOptions options = getDBMetaOptions(); + DBMetaOptions options; + try { + options = getDBMetaOptions(); + } catch(std::exception& ex) { + ASSERT_TRUE(false); + return; + } + int days_num = rand() % 100; std::stringstream ss; ss << "days:" << days_num; @@ -350,7 +378,14 @@ TEST_F(MySQLTest, ARCHIVE_TEST_DAYS) { } TEST_F(MySQLTest, ARCHIVE_TEST_DISK) { - DBMetaOptions options = getDBMetaOptions(); + DBMetaOptions options; + try { + options = getDBMetaOptions(); + } catch(std::exception& ex) { + ASSERT_TRUE(false); + return; + } + options.archive_conf = ArchiveConf("delete", "disk:11"); int mode = Options::MODE::SINGLE; auto impl = meta::MySQLMetaImpl(options, mode); @@ -397,9 +432,16 @@ TEST_F(MySQLTest, ARCHIVE_TEST_DISK) { } TEST_F(MySQLTest, TABLE_FILES_TEST) { + DBMetaOptions options; + try { + options = getDBMetaOptions(); + } catch(std::exception& ex) { + ASSERT_TRUE(false); + return; + } int mode = Options::MODE::SINGLE; - auto impl = meta::MySQLMetaImpl(getDBMetaOptions(), mode); + auto impl = meta::MySQLMetaImpl(options, mode); auto table_id = "meta_test_group"; diff --git a/cpp/unittest/db/utils.cpp b/cpp/unittest/db/utils.cpp index 6b1fc1e407d7eea883cc5b7e2a79e1ac9486a2f1..70c071254907a2c21aab24d276cc070c1d91d5e3 100644 --- a/cpp/unittest/db/utils.cpp +++ b/cpp/unittest/db/utils.cpp @@ -91,6 +91,11 @@ zilliz::milvus::engine::DBMetaOptions MySQLTest::getDBMetaOptions() { zilliz::milvus::engine::DBMetaOptions options; options.path = "/tmp/milvus_test"; options.backend_uri = DBTestEnvironment::getURI(); + + if(options.backend_uri.empty()) { + throw std::exception(); + } + return options; }