admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-08 01:58:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

MS-532 Add grpc server unittest

Browse Source 
Former-commit-id: 92368a067cc2a7295f3fd027b576f78789297b52
This commit is contained in:
Yu Kun 2019-09-11 20:17:43 +08:00
parent 6afd997b7a
commit cc0ce2e49f
2 changed files with 7 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cpp/src/sdk/examples/grpcsimple/src/ClientTest.cpp
View File

@ -164,7 +164,7 @@ void DoSearch(std::shared_ptr<Connection> conn,
const std::string& phase_name) { const std::string& phase_name) {
std::vector<Range> query_range_array; std::vector<Range> query_range_array;
Range rg; Range rg;
rg.start_value = CurrentTmDate(1); rg.start_value = CurrentTmDate();
rg.end_value = CurrentTmDate(1); rg.end_value = CurrentTmDate(1);
query_range_array.emplace_back(rg); query_range_array.emplace_back(rg);

16
cpp/unittest/server/rpc_test.cpp
View File

@ -18,7 +18,6 @@
#include "server/DBWrapper.h" #include "server/DBWrapper.h"
#include "server/ServerConfig.h" #include "server/ServerConfig.h"
#include "db/Factories.h"
#include "scheduler/SchedInst.h" #include "scheduler/SchedInst.h"
#include "scheduler/ResourceFactory.h" #include "scheduler/ResourceFactory.h"
#include "utils/CommonUtil.h" #include "utils/CommonUtil.h"
@ -40,8 +39,6 @@ class RpcHandlerTest : public testing::Test {
protected: protected:
void void
SetUp() override { SetUp() override {
server::ConfigNode &config = server::ServerConfig::GetInstance().GetConfig(server::CONFIG_CACHE);
config.AddSequenceItem(server::CONFIG_GPU_IDS, "0");
auto res_mgr = engine::ResMgrInst::GetInstance(); auto res_mgr = engine::ResMgrInst::GetInstance();
res_mgr->Clear(); res_mgr->Clear();
@ -264,8 +261,8 @@ TEST_F(RpcHandlerTest, SearchTest) {
std::string *file_id = search_in_files_param.add_file_id_array(); std::string *file_id = search_in_files_param.add_file_id_array();
*file_id = "test_tbl"; *file_id = "test_tbl";
handler->SearchInFiles(&context, &search_in_files_param, &response); handler->SearchInFiles(&context, &search_in_files_param, &response);
delete file_id; // delete file_id;
delete range; // delete range;
} }
TEST_F(RpcHandlerTest, TablesTest) { TEST_F(RpcHandlerTest, TablesTest) {
@ -380,13 +377,12 @@ TEST_F(RpcHandlerTest, CmdTest) {
command.set_cmd("version"); command.set_cmd("version");
::milvus::grpc::StringReply reply; ::milvus::grpc::StringReply reply;
handler->Cmd(&context, &command, &reply); handler->Cmd(&context, &command, &reply);
ASSERT_EQ(reply.string_reply(), MILVUS_VERSION);
command.set_cmd("tasktable"); command.set_cmd("tasktable");
handler->Cmd(&context, &command, &reply); handler->Cmd(&context, &command, &reply);
command.set_cmd("test"); command.set_cmd("test");
handler->Cmd(&context, &command, &reply); handler->Cmd(&context, &command, &reply);
ASSERT_EQ(reply.string_reply(), MILVUS_VERSION);
} }
TEST_F(RpcHandlerTest, DeleteByRangeTest) { TEST_F(RpcHandlerTest, DeleteByRangeTest) {
@ -404,10 +400,10 @@ TEST_F(RpcHandlerTest, DeleteByRangeTest) {
int error_code = status.error_code(); int error_code = status.error_code();
ASSERT_EQ(error_code, ::milvus::grpc::ErrorCode::SUCCESS); ASSERT_EQ(error_code, ::milvus::grpc::ErrorCode::SUCCESS);
request.mutable_range()->set_start_value("aaa"); request.mutable_range()->set_start_value("test6");
grpc_status = handler->DeleteByRange(&context, &request, &status); grpc_status = handler->DeleteByRange(&context, &request, &status);
request.mutable_range()->set_start_value(CurrentTmDate(-2)); request.mutable_range()->set_start_value(CurrentTmDate(-2));
request.mutable_range()->set_end_value("aaa"); request.mutable_range()->set_end_value("test6");
grpc_status = handler->DeleteByRange(&context, &request, &status); grpc_status = handler->DeleteByRange(&context, &request, &status);
request.mutable_range()->set_end_value(CurrentTmDate(-2)); request.mutable_range()->set_end_value(CurrentTmDate(-2));
grpc_status = handler->DeleteByRange(&context, &request, &status); grpc_status = handler->DeleteByRange(&context, &request, &status);