diff --git a/core/unittest/server/test_rpc.cpp b/core/unittest/server/test_rpc.cpp index 8560ad9db40b4492c84a4fa18c294648d5af0508..6fd5a5da837166c022cc8fc26ca632078a125e5f 100644 --- a/core/unittest/server/test_rpc.cpp +++ b/core/unittest/server/test_rpc.cpp @@ -514,6 +514,7 @@ TEST_F(RpcHandlerTest, COMBINE_SEARCH_TEST) { int64_t index = 0; for (auto& result_ptr : result_array) { ASSERT_NE(result_ptr->ids_size(), 0); + ASSERT_NE(result_ptr->row_num(), 0); std::string msg = "Result no." + std::to_string(index) + ": \n"; for (int64_t i = 0; i < NQ; i++) { for (int64_t k = 0; k < TOPK; k++) { @@ -615,6 +616,7 @@ TEST_F(RpcHandlerTest, COMBINE_SEARCH_BINARY_TEST) { int64_t index = 0; for (auto& result_ptr : result_array) { ASSERT_NE(result_ptr->ids_size(), 0); + ASSERT_NE(result_ptr->row_num(), 0); std::string msg = "Result no." + std::to_string(++index) + ": \n"; for (int64_t i = 0; i < NQ; i++) { for (int64_t k = 0; k < TOPK; k++) { diff --git a/sdk/examples/qps/src/ClientTest.cpp b/sdk/examples/qps/src/ClientTest.cpp index df84df01f6cdb57273e70449510d2afab239e025..de23be35da78cd2d6acc1ac2d49c527f09a9c3d3 100644 --- a/sdk/examples/qps/src/ClientTest.cpp +++ b/sdk/examples/qps/src/ClientTest.cpp @@ -58,6 +58,8 @@ InsertEntities(std::shared_ptr& conn) { milvus::Status stat = conn->Insert(COLLECTION_NAME, "", entity_array, record_ids); std::cout << "InsertEntities function call status: " << stat.message() << std::endl; std::cout << "Returned id array count: " << record_ids.size() << std::endl; + + stat = conn->FlushCollection(COLLECTION_NAME); } return true;