Commit 5f05f8b3 authored by groot's avatar groot
Browse files

#48 Config unittest failed 


Former-commit-id: 528fa91b59745822d81e3e9ee7cd23fff2382292
parent 49e05252

core/unittest/server/test_config.cpp

+5 −3
Original line number Diff line number Diff line
@@ -38,10 +38,11 @@ TEST_F(ConfigTest, CONFIG_TEST) { 
    milvus::Status s = config_mgr->LoadConfigFile("");

    ASSERT_FALSE(s.ok());



    s = config_mgr->LoadConfigFile(INVALID_CONFIG_PATH);

    std::string config_path(CONFIG_PATH);

    s = config_mgr->LoadConfigFile(config_path+ INVALID_CONFIG_FILE);

    ASSERT_FALSE(s.ok());



    s = config_mgr->LoadConfigFile(VALID_CONFIG_PATH);

    s = config_mgr->LoadConfigFile(config_path + VALID_CONFIG_FILE);

    ASSERT_TRUE(s.ok());



    config_mgr->Print();
@@ -98,8 +99,9 @@ TEST_F(ConfigTest, CONFIG_TEST) { 
}



TEST_F(ConfigTest, SERVER_CONFIG_TEST) {

    std::string config_path(CONFIG_PATH);

    milvus::server::Config &config = milvus::server::Config::GetInstance();

    milvus::Status s = config.LoadConfigFile(VALID_CONFIG_PATH);

    milvus::Status s = config.LoadConfigFile(config_path + VALID_CONFIG_FILE);

    ASSERT_TRUE(s.ok());



    s = config.ValidateConfig();

core/unittest/server/utils.cpp

+9 −7
Original line number Diff line number Diff line
@@ -16,12 +16,12 @@ 
// under the License.



#include "server/utils.h"

#include "utils/CommonUtil.h"



#include <fstream>

#include <iostream>

#include <thread>

#include <string>

#include <boost/filesystem.hpp>



namespace {
@@ -67,8 +67,8 @@ static const char 
static const char* INVALID_CONFIG_STR = "*INVALID*";



void

WriteToFile(const char* file_path, const char* content) {

    std::fstream fs(file_path, std::ios_base::out);

WriteToFile(const std::string& file_path, const char* content) {

    std::fstream fs(file_path.c_str(), std::ios_base::out);



    //write data to file

    fs << content;
@@ -80,12 +80,14 @@ WriteToFile(const char* file_path, const char* content) { 


void

ConfigTest::SetUp() {

    WriteToFile(VALID_CONFIG_PATH, VALID_CONFIG_STR);

    WriteToFile(INVALID_CONFIG_PATH, INVALID_CONFIG_STR);

    std::string config_path(CONFIG_PATH);

    milvus::server::CommonUtil::CreateDirectory(config_path);

    WriteToFile(config_path + VALID_CONFIG_FILE, VALID_CONFIG_STR);

    WriteToFile(config_path+ INVALID_CONFIG_FILE, INVALID_CONFIG_STR);

}



void

ConfigTest::TearDown() {

    boost::filesystem::remove(VALID_CONFIG_PATH);

    boost::filesystem::remove(INVALID_CONFIG_PATH);

    std::string config_path(CONFIG_PATH);

    milvus::server::CommonUtil::DeleteDirectory(config_path);

}

core/unittest/server/utils.h

+3 −2
Original line number Diff line number Diff line
@@ -21,8 +21,9 @@ 
#include <gtest/gtest.h>

#include <chrono>



static const char *VALID_CONFIG_PATH = "/tmp/milvus_test/valid_config.yaml";

static const char *INVALID_CONFIG_PATH = "/tmp/milvus_test/invalid_config.conf";

static const char *CONFIG_PATH = "/tmp/milvus_test/";

static const char *VALID_CONFIG_FILE = "valid_config.yaml";

static const char *INVALID_CONFIG_FILE = "invalid_config.conf";



class ConfigTest : public ::testing::Test {

 protected: