admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-07 19:31:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

MS-532 Add grpc server unittest

Browse Source 
Former-commit-id: 031714ba8543429be0862e5878a20cdfdbb3981c
This commit is contained in:
Yu Kun 2019-09-11 11:02:29 +08:00
parent c98c46d3f0
commit ca61c45f09
1 changed files with 22 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -227,9 +227,25 @@ TEST_F(RpcHandlerTest, SearchTest) {
//test search with range
::milvus::grpc::Range *range = request.mutable_query_range_array()->Add();
range->set_start_value(CurrentTmDate(-2));
range->set_end_value(CurrentTmDate(-3));
std::vector<std::vector<float>> record_array;
BuildVectors(0, VECTOR_COUNT, record_array);
::milvus::grpc::InsertParam insert_param;
for (auto &record : record_array) {
::milvus::grpc::RowRecord *grpc_record = insert_param.add_row_record_array();
for (size_t i = 0; i < record.size(); i++) {
grpc_record->add_vector_data(record[i]);
}
}
//insert vectors
insert_param.set_table_name(TABLE_NAME);
::milvus::grpc::VectorIds vector_ids;
handler->Insert(&context, &insert_param, &vector_ids);
record_array.clear();
BuildVectors(0, 10, record_array);
for (auto &record : record_array) {
::milvus::grpc::RowRecord *row_record = request.add_query_record_array();
for (auto &rec : record) {
@ -241,10 +257,15 @@ TEST_F(RpcHandlerTest, SearchTest) {
request.set_table_name(TABLE_NAME);
handler->Search(&context, &request, &response);
request.mutable_query_range_array()->Clear();
handler->Search(&context, &request, &response);
::milvus::grpc::SearchInFilesParam search_in_files_param;
std::string *file_id = search_in_files_param.add_file_id_array();
*file_id = "test_tbl";
handler->SearchInFiles(&context, &search_in_files_param, &response);
delete file_id;
delete range;
}
TEST_F(RpcHandlerTest, TablesTest) {