MS-653 fix config unittest

Former-commit-id: bd20d3caff96edc92625dbc333feaa94c5e3b841
pull/191/head
yudong.cai 2019-10-15 16:41:08 +08:00
parent b95f3f6dc8
commit 38747d9fae
1 changed files with 10 additions and 8 deletions

View File

@ -18,11 +18,13 @@
#include <gtest/gtest.h>
#include <gtest/gtest-death-test.h>
#include "config/ConfigMgr.h"
#include "config/YamlConfigMgr.h"
#include "utils/CommonUtil.h"
#include "utils/ValidationUtil.h"
#include "server/Config.h"
using namespace milvus;
namespace {
static const char *CONFIG_FILE_PATH = "./milvus/conf/server_config.yaml";
@ -35,16 +37,16 @@ static constexpr uint64_t GB = MB * 1024;
} // namespace
TEST(ConfigTest, CONFIG_TEST) {
milvus::server::ConfigMgr *config_mgr = milvus::server::ConfigMgr::GetInstance();
server::ConfigMgr *config_mgr = server::YamlConfigMgr::GetInstance();
milvus::ErrorCode err = config_mgr->LoadConfigFile("");
ASSERT_EQ(err, milvus::SERVER_UNEXPECTED_ERROR);
milvus::Status s = config_mgr->LoadConfigFile("");
ASSERT_FALSE(s.ok());
err = config_mgr->LoadConfigFile(LOG_FILE_PATH);
ASSERT_EQ(err, milvus::SERVER_UNEXPECTED_ERROR);
s = config_mgr->LoadConfigFile(LOG_FILE_PATH);
ASSERT_FALSE(s.ok());
err = config_mgr->LoadConfigFile(CONFIG_FILE_PATH);
ASSERT_EQ(err, milvus::SERVER_SUCCESS);
s = config_mgr->LoadConfigFile(CONFIG_FILE_PATH);
ASSERT_TRUE(s.ok());
config_mgr->Print();