|
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205 |
- /**
- * Copyright 2020 Huawei Technologies Co., Ltd
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
- #include "common/common_test.h"
- #include "master/server.h"
- #include "common/tensor_base.h"
- #define private public
- #include "master/restful/http_process.h"
- #undef private
-
- using std::string;
- using std::vector;
- namespace mindspore {
- namespace serving {
- class TestParseInput : public UT::Common {
- public:
- TestParseInput() = default;
- };
-
- class TestParseReply : public UT::Common {
- public:
- TestParseReply() = default;
- };
-
- TEST_F(TestParseInput, test_parse_SUCCESS) {
- nlohmann::json js = R"(
- {"instances":[
- {
- "key_tag":"scalar",
- "key_int": 1,
- "key_bool": false,
- "key_float": 2.3,
- "key_str": "ut_test",
- "key_bytes": {"b64": "dXRfdGVzdA==", "type": "bytes"}
- },
- {
- "key_tag":"tensor",
- "key_int": [1,2,3],
- "key_bool":[[true, false], [false, true]],
- "key_float":[[1.1, 2.2]],
- "key_str":["ut_test"],
- "key_bytes":{"b64":"dXRfdGVzdA=="}
- },
- {
- "key_tag":"b64",
- "key_str_format1":"ut_test",
- "key_str_foramt2":{"b64":"dXRfdGVzdA==", "type":"str"},
- "key_bytes_int16":{"b64":"AQACAAIAAwADAAQA", "type":"int16", "shape":[3,2]},
- "key_bytes_fp16":{"b64":"ZjxmQJpCZkQ=", "type":"fp16", "shape":[2,2]},
- "key_bytes_bool":{"b64":"AQA=", "type":"bool", "shape":[1,2]}
- }
- ]
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_EQ(status.StatusCode(), SUCCESS);
- ASSERT_EQ(predict_request.instances().size(), 3);
- for (int32_t i = 0; i < predict_request.instances().size(); i++) {
- auto &cur_instance = predict_request.instances(i);
- auto &items = cur_instance.items();
- if (i == 0) {
- ASSERT_EQ(items.size(), 6);
- for (const auto &item : items) {
- ProtoTensor pb_tensor(const_cast<proto::Tensor *>(&item.second));
- if (item.first == "key_int") {
- ASSERT_EQ(pb_tensor.data_type(), DataType::kMSI_Int32);
- const int32_t *data = reinterpret_cast<const int32_t *>(pb_tensor.data());
- ASSERT_EQ(*data, 1);
- } else if (item.first == "key_bool") {
- ASSERT_EQ(pb_tensor.data_type(), DataType::kMSI_Bool);
- const bool *data = reinterpret_cast<const bool *>(pb_tensor.data());
- ASSERT_EQ(*data, false);
- } else if (item.first == "key_float") {
- ASSERT_EQ(pb_tensor.data_type(), DataType::kMSI_Float32);
- const float *data = reinterpret_cast<const float *>(pb_tensor.data());
- ASSERT_FLOAT_EQ(*data, 2.3);
- } else if (item.first == "key_str") {
- ASSERT_EQ(pb_tensor.data_type(), DataType::kMSI_String);
- auto str_nums = pb_tensor.bytes_data_size();
- ASSERT_EQ(str_nums, 1);
- std::string value;
- size_t length;
- const uint8_t *ptr = nullptr;
- pb_tensor.get_bytes_data(0, &ptr, &length);
- value.resize(length);
- memcpy_s(value.data(), length, reinterpret_cast<const char *>(ptr), length);
- ASSERT_EQ(value, "ut_test");
- } else if (item.first == "key_bytes") {
- ASSERT_EQ(pb_tensor.data_type(), DataType::kMSI_Bytes);
- auto str_nums = pb_tensor.bytes_data_size();
- ASSERT_EQ(str_nums, 1);
- std::string value;
- size_t length;
- const uint8_t *ptr = nullptr;
- pb_tensor.get_bytes_data(0, &ptr, &length);
- value.resize(length);
- memcpy_s(value.data(), length, reinterpret_cast<const char *>(ptr), length);
- ASSERT_EQ(value, "ut_test");
- }
- }
- } else if (i == 1) {
- ASSERT_EQ(items.size(), 6);
- for (const auto &item : items) {
- ProtoTensor pb_tensor(const_cast<proto::Tensor *>(&item.second));
- auto shape = pb_tensor.shape();
- if (item.first == "key_int") {
- ASSERT_EQ(pb_tensor.data_type(), DataType::kMSI_Int32);
- ASSERT_EQ(shape.size(), 1);
- ASSERT_EQ(shape[0], 3);
- vector<int32_t> expected_value = {1, 2, 3};
- for (int i = 0; i < 3; i++) {
- const int32_t *data = reinterpret_cast<const int32_t *>(pb_tensor.data()) + i;
- ASSERT_EQ(*data, expected_value[i]);
- }
- } else if (item.first == "key_bool") {
- ASSERT_EQ(pb_tensor.data_type(), DataType::kMSI_Bool);
- ASSERT_EQ(shape.size(), 2);
- ASSERT_EQ(shape[0], 2);
- ASSERT_EQ(shape[1], 2);
- vector<vector<bool>> expected_value = {{true, false}, {false, true}};
- for (int i = 0; i < 2; i++) {
- for (int j = 0; j < 2; j++) {
- const bool *data = reinterpret_cast<const bool *>(pb_tensor.data()) + i * 2 + j;
- ASSERT_EQ(*data, expected_value[i][j]);
- }
- }
- } else if (item.first == "key_float") {
- ASSERT_EQ(pb_tensor.data_type(), DataType::kMSI_Float32);
- ASSERT_EQ(shape.size(), 2);
- ASSERT_EQ(shape[0], 1);
- ASSERT_EQ(shape[1], 2);
- vector<vector<float>> expected_value = {{1.1, 2.2}};
- for (int i = 0; i < 1; i++) {
- for (int j = 0; j < 2; j++) {
- const float *data = reinterpret_cast<const float *>(pb_tensor.data()) + i * 1 + j;
- ASSERT_FLOAT_EQ(*data, expected_value[i][j]);
- }
- }
- } else if (item.first == "key_str") {
- ASSERT_EQ(pb_tensor.data_type(), DataType::kMSI_String);
- ASSERT_EQ(shape.size(), 1);
- ASSERT_EQ(shape[0], 1);
- auto str_nums = pb_tensor.bytes_data_size();
- ASSERT_EQ(str_nums, 1);
- std::string value;
- size_t length;
- const uint8_t *ptr = nullptr;
- pb_tensor.get_bytes_data(0, &ptr, &length);
- value.resize(length);
- memcpy_s(value.data(), length, reinterpret_cast<const char *>(ptr), length);
- ASSERT_EQ(value, "ut_test");
- } else if (item.first == "key_bytes") {
- ASSERT_EQ(pb_tensor.data_type(), DataType::kMSI_Bytes);
- auto str_nums = pb_tensor.bytes_data_size();
- ASSERT_EQ(str_nums, 1);
- std::string value;
- size_t length;
- const uint8_t *ptr = nullptr;
- pb_tensor.get_bytes_data(0, &ptr, &length);
- value.resize(length);
- memcpy_s(value.data(), length, reinterpret_cast<const char *>(ptr), length);
- ASSERT_EQ(value, "ut_test");
- }
- }
- } else if (i == 2) {
- ASSERT_EQ(items.size(), 6);
- for (const auto &item : items) {
- ProtoTensor pb_tensor(const_cast<proto::Tensor *>(&item.second));
- auto shape = pb_tensor.shape();
- if (item.first == "key_str_format1") {
- ASSERT_EQ(pb_tensor.data_type(), DataType::kMSI_String);
- auto str_nums = pb_tensor.bytes_data_size();
- ASSERT_EQ(str_nums, 1);
- std::string value;
- size_t length;
- const uint8_t *ptr = nullptr;
- pb_tensor.get_bytes_data(0, &ptr, &length);
- value.resize(length);
- memcpy_s(value.data(), length, reinterpret_cast<const char *>(ptr), length);
- ASSERT_EQ(value, "ut_test");
- } else if (item.first == "key_str_format2") {
- ASSERT_EQ(pb_tensor.data_type(), DataType::kMSI_String);
- auto str_nums = pb_tensor.bytes_data_size();
- ASSERT_EQ(str_nums, 1);
- std::string value;
- size_t length;
- const uint8_t *ptr = nullptr;
- pb_tensor.get_bytes_data(0, &ptr, &length);
- value.resize(length);
- memcpy_s(value.data(), length, reinterpret_cast<const char *>(ptr), length);
- ASSERT_EQ(value, "ut_test");
- } else if (item.first == "key_bytes_int16") {
- ASSERT_EQ(pb_tensor.data_type(), DataType::kMSI_Int16);
- ASSERT_EQ(shape.size(), 2);
- ASSERT_EQ(shape[0], 3);
- ASSERT_EQ(shape[1], 2);
- vector<vector<int16_t>> expected_value = {{1, 2}, {2, 3}, {3, 4}};
- for (int i = 0; i < 3; i++) {
- for (int j = 0; j < 2; j++) {
- const int16_t *data = reinterpret_cast<const int16_t *>(pb_tensor.data()) + i * 2 + j;
- ASSERT_FLOAT_EQ(*data, expected_value[i][j]);
- }
- }
- } else if (item.first == "key_bytes_fp16") {
- ASSERT_EQ(pb_tensor.data_type(), DataType::kMSI_Float16);
- ASSERT_EQ(shape.size(), 2);
- ASSERT_EQ(shape[0], 2);
- ASSERT_EQ(shape[1], 2);
- } else if (item.first == "key_bytes_bool") {
- ASSERT_EQ(pb_tensor.data_type(), DataType::kMSI_Bool);
- ASSERT_EQ(shape.size(), 2);
- ASSERT_EQ(shape[0], 1);
- ASSERT_EQ(shape[1], 2);
- vector<vector<bool>> expected_value = {{true, false}};
- for (int i = 0; i < 1; i++) {
- for (int j = 0; j < 2; j++) {
- const bool *data = reinterpret_cast<const bool *>(pb_tensor.data()) + i * 2 + j;
- ASSERT_FLOAT_EQ(*data, expected_value[i][j]);
- }
- }
- }
- }
- }
- }
- }
-
- TEST_F(TestParseInput, test_instances_empty_FAIL) {
- nlohmann::json js = R"(
- {"":
- {
- "key_tag":"scalar",
- "key_int": 1,
- "key_bool": false,
- "key_float": 2.3,
- "key_str": "ut_test",
- "key_bytes": {"b64": "dXRfdGVzdA==", "type": "bytes"}
- }
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_NE(status.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseInput, test_instances_incorrect_FAIL) {
- nlohmann::json js = R"(
- {"instance":
- {
- "key_tag":"scalar",
- "key_int": 1,
- "key_bool": false,
- "key_float": 2.3,
- "key_str": "ut_test",
- "key_bytes": {"b64": "dXRfdGVzdA==", "type": "bytes"}
- }
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_NE(status.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseInput, test_key_empty_FAIL) {
- nlohmann::json js = R"(
- {"instances":
- {
- "":"scalar",
- "key_int": 1,
- "key_bool": false,
- "key_float": 2.3,
- "key_str": "ut_test",
- "key_bytes": {"b64": "dXRfdGVzdA==", "type": "bytes"}
- }
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_NE(status.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseInput, test_value_empty_FAIL) {
- nlohmann::json js = R"(
- {"instances":
- {
- "key_tag":"",
- "key_int": 1,
- "key_bool": false,
- "key_float": 2.3,
- "key_str": "ut_test",
- "key_bytes": {"b64": "dXRfdGVzdA==", "type": "bytes"}
- }
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_NE(status.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseInput, test_obj_unknown_key_FAIL) {
- nlohmann::json js = R"(
- {"instances":
- {
- "key_tag":"",
- "key_int": 1,
- "key_bool": false,
- "key_float": 2.3,
- "key_str": "ut_test",
- "key_bytes": {"b64": "dXRfdGVzdA==", "type": "bytes", "type1":"bytes"}
- }
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_NE(status.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseInput, test_obj_nob64_key_FAIL) {
- nlohmann::json js = R"(
- {"instances":
- {
- "key_tag":"",
- "key_int": 1,
- "key_bool": false,
- "key_float": 2.3,
- "key_str": "ut_test",
- "key_bytes": {"base64": "dXRfdGVzdA==", "type": "bytes"}
- }
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_NE(status.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseInput, test_obj_illegal_b64value_FAIL) {
- nlohmann::json js = R"(
- {"instances":
- {
- "key_tag":"",
- "key_int": 1,
- "key_bool": false,
- "key_float": 2.3,
- "key_str": "ut_test",
- "key_bytes": {"base64": "dXRfdGVzdA", "type": "bytes"}
- }
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_NE(status.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseInput, test_obj_unknown_type_FAIL) {
- nlohmann::json js = R"(
- {"instances":
- {
- "key_tag":"",
- "key_int": 1,
- "key_bool": false,
- "key_float": 2.3,
- "key_str": "ut_test",
- "key_bytes": {"base64": "dXRfdGVzdA==", "type": "INt"}
- }
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_NE(status.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseInput, test_obj_error_shape_format_FAIL) {
- nlohmann::json js = R"(
- {"instances":
- {
- "key_tag":"",
- "key_int": 1,
- "key_bool": false,
- "key_float": 2.3,
- "key_str": "ut_test",
- "key_bytes_int16":{"b64":"AQACAAIAAwADAAQA", "type":"int16", "shape":3}
- }
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_NE(status.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseInput, test_obj_error_shape_format2_FAIL) {
- nlohmann::json js = R"(
- {"instances":
- {
- "key_tag":"",
- "key_int": 1,
- "key_bool": false,
- "key_float": 2.3,
- "key_str": "ut_test",
- "key_bytes_int16":{"b64":"AQACAAIAAwADAAQA", "type":"int16", "shape":[[3],[2]]}
- }
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_NE(status.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseInput, test_obj_error_shape_value_FAIL) {
- nlohmann::json js = R"(
- {"instances":
- {
- "key_tag":"",
- "key_int": 1,
- "key_bool": false,
- "key_float": 2.3,
- "key_str": "ut_test",
- "key_bytes_int16":{"b64":"AQACAAIAAwADAAQA", "type":"int16", "shape":[3.0,2.0]}
- }
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_NE(status.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseInput, test_obj_error_shape_value2_FAIL) {
- nlohmann::json js = R"(
- {"instances":
- {
- "key_tag":"",
- "key_int": 1,
- "key_bool": false,
- "key_float": 2.3,
- "key_str": "ut_test",
- "key_bytes_int16":{"b64":"AQACAAIAAwADAAQA", "type":"int16", "shape":[3,3]}
- }
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_NE(status.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseInput, test_obj_error_shape_value3_FAIL) {
- nlohmann::json js = R"(
- {"instances":
- {
- "key_tag":"",
- "key_int": 1,
- "key_bool": false,
- "key_float": 2.3,
- "key_str": "ut_test",
- "key_bytes_int16":{"b64":"AQACAAIAAwADAAQA", "type":"int16", "shape":[3,-2]}
- }
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_NE(status.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseInput, test_tensor_value_empty_FAIL) {
- nlohmann::json js = R"(
- {"instances":
- {
- "key_tag":"tensor",
- "key_int": [],
- "key_bool":[[true, false], [false, true]],
- "key_float":[[1.1, 2.2]],
- "key_str":["ut_test"],
- "key_bytes":{"b64":"dXRfdGVzdA=="}
- }
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_NE(status.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseInput, test_tensor_value_diff_type_FAIL) {
- nlohmann::json js = R"(
- {"instances":
- {
- "key_tag":"tensor",
- "key_int": [1, 2.0],
- "key_bool":[[true, false], [false, true]],
- "key_float":[[1.1, 2.2]],
- "key_str":["ut_test"],
- "key_bytes":{"b64":"dXRfdGVzdA=="}
- }
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_NE(status.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseInput, test_tensor_value_diff_dimention_FAIL) {
- nlohmann::json js = R"(
- {"instances":
- {
- "key_tag":"tensor",
- "key_int": [1, 2],
- "key_bool":[[true, false], [false]],
- "key_float":[[1.1, 2.2]],
- "key_str":["ut_test"],
- "key_bytes":{"b64":"dXRfdGVzdA=="}
- }
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_NE(status.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseInput, test_tensor_multi_object_FAIL) {
- nlohmann::json js = R"(
- {"instances":
- {
- "key_tag":"tensor",
- "key_int": [1, 2],
- "key_bool":[[true, false], [false, true]],
- "key_float":[[1.1, 2.2]],
- "key_str":["ut_test"],
- "key_bytes":[{"b64":"dXRfdGVzdA=="}, {"b64":"dXRfdGVzdA=="}]
- }
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_NE(status.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseReply, test_reply_SUCCESS) {
- nlohmann::json js = R"(
- {"instances":[
- {
- "key_tag":"scalar",
- "key_int": 1,
- "key_bool": false,
- "key_float": 2.3,
- "key_str": "ut_test",
- "key_bytes": {"b64": "dXRfdGVzdA==", "type": "bytes"}
- },
- {
- "key_tag":"tensor",
- "key_int": [1,2,3],
- "key_bool":[[true, false], [false, true]],
- "key_float":[[1.1, 2.2]],
- "key_str":["ut_test"]
- }
- ]
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status(INVALID_INPUTS);
- status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_EQ(status.StatusCode(), SUCCESS);
-
- nlohmann::json out_js;
- proto::PredictReply reply;
- auto instance_ptr = reply.add_instances();
- auto &map_item = *(instance_ptr->mutable_items());
- // test scalar:
- // scalar:key_int
- proto::Tensor tensor_int;
- ProtoTensor pb_tensor_int(&tensor_int);
- DataType type_int = kMSI_Int32;
- pb_tensor_int.set_data_type(type_int);
- pb_tensor_int.set_shape({1});
- pb_tensor_int.resize_data(pb_tensor_int.GetTypeSize(type_int));
- auto data_int = reinterpret_cast<int32_t *>(pb_tensor_int.mutable_data());
- *data_int = 1;
- map_item["key_int"] = tensor_int;
-
- // scalar: key_bool
- proto::Tensor tensor_bool;
- ProtoTensor pb_tensor_bool(&tensor_bool);
- DataType type_bool = kMSI_Bool;
- pb_tensor_bool.set_data_type(type_bool);
- pb_tensor_bool.resize_data(pb_tensor_bool.GetTypeSize(type_bool));
- auto data_bool = reinterpret_cast<bool *>(pb_tensor_bool.mutable_data());
- *data_bool = false;
- map_item["key_bool"] = tensor_bool;
-
- // scalar: key_float
- proto::Tensor tensor_float;
- ProtoTensor pb_tensor_float(&tensor_float);
- DataType type_float = kMSI_Float32;
- pb_tensor_float.set_data_type(type_float);
- pb_tensor_float.set_shape({1});
- pb_tensor_float.resize_data(pb_tensor_float.GetTypeSize(type_float));
- auto data_float = reinterpret_cast<float *>(pb_tensor_float.mutable_data());
- *data_float = 2.3;
- map_item["key_float"] = tensor_float;
-
- // scalar: key_str
- string value = "ut_test";
- proto::Tensor tensor_str;
- ProtoTensor pb_tensor_str(&tensor_str);
- DataType type_str = kMSI_String;
- pb_tensor_str.set_data_type(type_str);
- pb_tensor_str.add_bytes_data(reinterpret_cast<uint8_t *>(value.data()), value.length());
- map_item["key_str"] = tensor_str;
-
- // scalar: key_bytes
- string value_bytes = "ut_test";
- proto::Tensor tensor_bytes;
- ProtoTensor pb_tensor_bytes(&tensor_bytes);
- DataType type_bytes = kMSI_Bytes;
- pb_tensor_bytes.set_data_type(type_bytes);
- pb_tensor_bytes.add_bytes_data(reinterpret_cast<uint8_t *>(value_bytes.data()), value_bytes.length());
- map_item["key_bytes"] = tensor_bytes;
-
- // test tensor:
- auto instance_ptr2 = reply.add_instances();
- auto &map_item2 = *(instance_ptr2->mutable_items());
-
- // tensor int:
- vector<int32_t> tensor_value_int = {1, 2, 3};
- proto::Tensor tensor_int2;
- ProtoTensor pb_tensor_int2(&tensor_int2);
- DataType type_int2 = kMSI_Int32;
- pb_tensor_int2.set_data_type(type_int2);
- pb_tensor_int2.set_shape({3});
- pb_tensor_int2.resize_data(pb_tensor_int2.GetTypeSize(type_int2) * 3);
- for (int i = 0; i < 3; i++) {
- auto data_int2 = reinterpret_cast<int32_t *>(pb_tensor_int2.mutable_data()) + i;
- *data_int2 = tensor_value_int[i];
- }
- map_item2["key_int"] = tensor_int2;
-
- // tensor: key_bool
- vector<vector<bool>> tensor_value_bool = {{true, false}, {false, true}};
- proto::Tensor tensor_bool2;
- ProtoTensor pb_tensor_bool2(&tensor_bool2);
- DataType type_bool2 = kMSI_Bool;
- pb_tensor_bool2.set_data_type(type_bool2);
- pb_tensor_bool2.set_shape({2, 2});
- pb_tensor_bool2.resize_data(pb_tensor_bool2.GetTypeSize(type_bool2) * 4);
- for (int i = 0; i < 2; i++) {
- for (int j = 0; j < 2; j++) {
- auto data_bool2 = reinterpret_cast<bool *>(pb_tensor_bool2.mutable_data()) + i * 2 + j;
- *data_bool2 = tensor_value_bool[i][j];
- }
- }
- map_item2["key_bool"] = tensor_bool2;
-
- // tensor: key_float
- vector<vector<float>> tensor_value_float = {{1.1, 2.2}};
- proto::Tensor tensor_float2;
- ProtoTensor pb_tensor_float2(&tensor_float2);
- DataType type_float2 = kMSI_Float32;
- pb_tensor_float2.set_data_type(type_float2);
- pb_tensor_float2.set_shape({1, 2});
- pb_tensor_float2.resize_data(pb_tensor_float2.GetTypeSize(type_float2) * 2);
- for (int i = 0; i < 1; i++) {
- for (int j = 0; j < 2; j++) {
- auto data_float2 = reinterpret_cast<float *>(pb_tensor_float2.mutable_data()) + i * 1 + j;
- *data_float2 = tensor_value_float[i][j];
- }
- }
- map_item2["key_float"] = tensor_float2;
-
- // tensor: key_str
- vector<string> tensor_value_str = {"ut_test", "ut_test2"};
- proto::Tensor tensor_str2;
- ProtoTensor pb_tensor_str2(&tensor_str2);
- DataType type_str2 = kMSI_String;
- pb_tensor_str2.set_data_type(type_str2);
- pb_tensor_str2.set_shape({2});
- for (int i = 0; i < 2; i++) {
- pb_tensor_str2.add_bytes_data(reinterpret_cast<uint8_t *>(tensor_value_str[i].data()),
- tensor_value_str[i].length());
- }
- map_item2["key_str"] = tensor_str2;
-
- Status status2 = restful_service.ParseReply(reply, &out_js);
- ASSERT_EQ(status2.StatusCode(), SUCCESS);
- string out_str = out_js.dump();
- std::cout << "Parse reply out:" << out_str << std::endl;
-
- ASSERT_TRUE(out_js.is_object());
- for (auto &item : out_js.items()) {
- ASSERT_EQ(item.key(), "instances");
- ASSERT_TRUE(item.value().is_array());
- ASSERT_EQ(item.value().size(), 2);
- int sum = 0;
- // array
- for (auto &element : item.value()) {
- ASSERT_TRUE(element.is_object());
- if (element.size() == 5) {
- int count = 0;
- // object
- std::cout << "===start====" << std::endl;
- for (auto &it : element.items()) {
- if (it.key() == "key_int") {
- ASSERT_EQ(it.value(), 1);
- count++;
- } else if (it.key() == "key_bool") {
- ASSERT_EQ(it.value(), false);
- count++;
- } else if (it.key() == "key_float") {
- ASSERT_FLOAT_EQ(it.value(), 2.3);
- count++;
- } else if (it.key() == "key_str") {
- ASSERT_EQ(it.value(), "ut_test");
- count++;
- } else if (it.key() == "key_bytes") {
- ASSERT_TRUE(it.value().is_object());
- ASSERT_EQ(it.value()["b64"], "dXRfdGVzdA==");
- count++;
- }
- }
- ASSERT_EQ(count, 5);
- sum++;
- } else if (element.size() == 4) {
- int count = 0;
- // object
- for (auto &it : element.items()) {
- if (it.key() == "key_int") {
- ASSERT_TRUE(it.value().is_array());
- ASSERT_EQ(it.value().size(), 3);
- ASSERT_EQ(it.value()[0], 1);
- ASSERT_EQ(it.value()[1], 2);
- ASSERT_EQ(it.value()[2], 3);
- count++;
- } else if (it.key() == "key_bool") {
- ASSERT_TRUE(it.value().is_array());
- ASSERT_EQ(it.value().size(), 2);
- ASSERT_TRUE(it.value()[0].is_array());
- ASSERT_EQ(it.value()[0].size(), 2);
- ASSERT_EQ(it.value()[0][0], true);
- ASSERT_EQ(it.value()[0][1], false);
- ASSERT_EQ(it.value()[1].size(), 2);
- ASSERT_EQ(it.value()[1][0], false);
- ASSERT_EQ(it.value()[1][1], true);
- count++;
- } else if (it.key() == "key_float") {
- ASSERT_TRUE(it.value().is_array());
- ASSERT_EQ(it.value().size(), 1);
- ASSERT_TRUE(it.value()[0].is_array());
- ASSERT_EQ(it.value()[0].size(), 2);
- ASSERT_FLOAT_EQ(it.value()[0][0], 1.1);
- ASSERT_FLOAT_EQ(it.value()[0][1], 2.2);
- count++;
- } else if (it.key() == "key_str") {
- ASSERT_TRUE(it.value().is_array());
- ASSERT_EQ(it.value().size(), 2);
- ASSERT_EQ(it.value()[0], "ut_test");
- ASSERT_EQ(it.value()[1], "ut_test2");
- count++;
- }
- }
- ASSERT_EQ(count, 4);
- sum++;
- }
- }
- ASSERT_EQ(sum, 2);
- }
- }
-
- TEST_F(TestParseReply, test_reply_instances_num_not_match_FAIL) {
- nlohmann::json js = R"(
- {"instances":[
- {
- "key_tag":"scalar",
- "key_int": 1,
- "key_bool": false,
- "key_float": 2.3,
- "key_str": "ut_test",
- "key_bytes": {"b64": "dXRfdGVzdA==", "type": "bytes"}
- }
- ]
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status(INVALID_INPUTS);
- status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_EQ(status.StatusCode(), SUCCESS);
-
- nlohmann::json out_js;
- proto::PredictReply reply;
- auto instance_ptr = reply.add_instances();
- auto &map_item = *(instance_ptr->mutable_items());
- // test scalar:
- // scalar:key_int
- proto::Tensor tensor_int;
- ProtoTensor pb_tensor_int(&tensor_int);
- DataType type_int = kMSI_Int32;
- pb_tensor_int.set_data_type(type_int);
- pb_tensor_int.set_shape({1});
- pb_tensor_int.resize_data(pb_tensor_int.GetTypeSize(type_int));
- auto data_int = reinterpret_cast<int32_t *>(pb_tensor_int.mutable_data());
- *data_int = 1;
- map_item["key_int"] = tensor_int;
-
- // scalar: key_bool
- proto::Tensor tensor_bool;
- ProtoTensor pb_tensor_bool(&tensor_bool);
- DataType type_bool = kMSI_Bool;
- pb_tensor_bool.set_data_type(type_bool);
- pb_tensor_bool.resize_data(pb_tensor_bool.GetTypeSize(type_bool));
- auto data_bool = reinterpret_cast<bool *>(pb_tensor_bool.mutable_data());
- *data_bool = false;
- map_item["key_bool"] = tensor_bool;
-
- // scalar: key_float
- proto::Tensor tensor_float;
- ProtoTensor pb_tensor_float(&tensor_float);
- DataType type_float = kMSI_Float32;
- pb_tensor_float.set_data_type(type_float);
- pb_tensor_float.set_shape({1});
- pb_tensor_float.resize_data(pb_tensor_float.GetTypeSize(type_float));
- auto data_float = reinterpret_cast<float *>(pb_tensor_float.mutable_data());
- *data_float = 2.3;
- map_item["key_float"] = tensor_float;
-
- // scalar: key_str
- string value = "ut_test";
- proto::Tensor tensor_str;
- ProtoTensor pb_tensor_str(&tensor_str);
- DataType type_str = kMSI_String;
- pb_tensor_str.set_data_type(type_str);
- pb_tensor_str.add_bytes_data(reinterpret_cast<uint8_t *>(value.data()), value.length());
- map_item["key_str"] = tensor_str;
-
- // scalar: key_bytes
- string value_bytes = "ut_test";
- proto::Tensor tensor_bytes;
- ProtoTensor pb_tensor_bytes(&tensor_bytes);
- DataType type_bytes = kMSI_Bytes;
- pb_tensor_bytes.set_data_type(type_bytes);
- pb_tensor_bytes.add_bytes_data(reinterpret_cast<uint8_t *>(value_bytes.data()), value_bytes.length());
- map_item["key_bytes"] = tensor_bytes;
-
- // test tensor:
- auto instance_ptr2 = reply.add_instances();
- auto &map_item2 = *(instance_ptr2->mutable_items());
-
- // tensor int:
- vector<int32_t> tensor_value_int = {1, 2, 3};
- proto::Tensor tensor_int2;
- ProtoTensor pb_tensor_int2(&tensor_int2);
- DataType type_int2 = kMSI_Int32;
- pb_tensor_int2.set_data_type(type_int2);
- pb_tensor_int2.set_shape({3});
- pb_tensor_int2.resize_data(pb_tensor_int2.GetTypeSize(type_int2) * 3);
- for (int i = 0; i < 3; i++) {
- auto data_int2 = reinterpret_cast<int32_t *>(pb_tensor_int2.mutable_data()) + i;
- *data_int2 = tensor_value_int[i];
- }
- map_item2["key_int"] = tensor_int2;
-
- // tensor: key_bool
- vector<vector<bool>> tensor_value_bool = {{true, false}, {false, true}};
- proto::Tensor tensor_bool2;
- ProtoTensor pb_tensor_bool2(&tensor_bool2);
- DataType type_bool2 = kMSI_Bool;
- pb_tensor_bool2.set_data_type(type_bool2);
- pb_tensor_bool2.set_shape({2, 2});
- pb_tensor_bool2.resize_data(pb_tensor_bool2.GetTypeSize(type_bool2) * 4);
- for (int i = 0; i < 2; i++) {
- for (int j = 0; j < 2; j++) {
- auto data_bool2 = reinterpret_cast<bool *>(pb_tensor_bool2.mutable_data()) + i * 2 + j;
- *data_bool2 = tensor_value_bool[i][j];
- }
- }
- map_item2["key_bool"] = tensor_bool2;
-
- // tensor: key_float
- vector<vector<float>> tensor_value_float = {{1.1, 2.2}};
- proto::Tensor tensor_float2;
- ProtoTensor pb_tensor_float2(&tensor_float2);
- DataType type_float2 = kMSI_Float32;
- pb_tensor_float2.set_data_type(type_float2);
- pb_tensor_float2.set_shape({1, 2});
- pb_tensor_float2.resize_data(pb_tensor_float2.GetTypeSize(type_float2) * 2);
- for (int i = 0; i < 1; i++) {
- for (int j = 0; j < 2; j++) {
- auto data_float2 = reinterpret_cast<float *>(pb_tensor_float2.mutable_data()) + i * 1 + j;
- *data_float2 = tensor_value_float[i][j];
- }
- }
- map_item2["key_float"] = tensor_float2;
-
- // tensor: key_str
- vector<string> tensor_value_str = {"ut_test", "ut_test2"};
- proto::Tensor tensor_str2;
- ProtoTensor pb_tensor_str2(&tensor_str2);
- DataType type_str2 = kMSI_String;
- pb_tensor_str2.set_data_type(type_str2);
- pb_tensor_str2.set_shape({2});
- for (int i = 0; i < 2; i++) {
- pb_tensor_str2.add_bytes_data(reinterpret_cast<uint8_t *>(tensor_value_str[i].data()),
- tensor_value_str[i].length());
- }
- map_item2["key_str"] = tensor_str2;
-
- Status status2 = restful_service.ParseReply(reply, &out_js);
- ASSERT_NE(status2.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseReply, test_reply_error_num_not_match_FAIL) {
- nlohmann::json js = R"(
- {"instances":[
- {
- "key_tag":"scalar",
- "key_int": 1,
- "key_bool": false,
- "key_float": 2.3,
- "key_str": "ut_test",
- "key_bytes": {"b64": "dXRfdGVzdA==", "type": "bytes"}
- }
- ]
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status(INVALID_INPUTS);
- status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_EQ(status.StatusCode(), SUCCESS);
-
- nlohmann::json out_js;
- proto::PredictReply reply;
- auto error_msg = reply.add_error_msg();
- error_msg->set_error_msg("error1");
-
- auto error_msg2 = reply.add_error_msg();
- error_msg2->set_error_msg("error2");
-
- Status status2 = restful_service.ParseReply(reply, &out_js);
- ASSERT_NE(status2.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseReply, test_reply_type_not_set_FAIL) {
- nlohmann::json js = R"(
- {"instances":[
- {
- "key_tag":"scalar",
- "key_int": 1,
- "key_bool": false,
- "key_float": 2.3,
- "key_str": "ut_test",
- "key_bytes": {"b64": "dXRfdGVzdA==", "type": "bytes"}
- }
- ]
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status(INVALID_INPUTS);
- status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_EQ(status.StatusCode(), SUCCESS);
-
- nlohmann::json out_js;
- proto::PredictReply reply;
- auto instance_ptr = reply.add_instances();
- auto &map_item = *(instance_ptr->mutable_items());
- // test scalar:
- // scalar:key_int
- proto::Tensor tensor_int;
- ProtoTensor pb_tensor_int(&tensor_int);
- pb_tensor_int.set_shape({1});
- pb_tensor_int.resize_data(pb_tensor_int.GetTypeSize(kMSI_Int32));
- auto data_int = reinterpret_cast<int32_t *>(pb_tensor_int.mutable_data());
- *data_int = 1;
- map_item["key_int"] = tensor_int;
-
- Status status2 = restful_service.ParseReply(reply, &out_js);
- ASSERT_NE(status2.StatusCode(), SUCCESS);
- }
-
- TEST_F(TestParseReply, test_reply_type_fp16_FAIL) {
- nlohmann::json js = R"(
- {"instances":[
- {
- "key_tag":"scalar",
- "key_int": 1,
- "key_bool": false,
- "key_float": 2.3,
- "key_str": "ut_test",
- "key_bytes": {"b64": "dXRfdGVzdA==", "type": "bytes"}
- }
- ]
- }
- )"_json;
-
- struct evhttp_request *request = new evhttp_request();
- int size = 100;
- std::shared_ptr<DecomposeEvRequest> request_msg = std::make_shared<DecomposeEvRequest>(request, size);
- request_msg->request_message_ = js;
- std::shared_ptr<RestfulRequest> restful_request = std::make_shared<RestfulRequest>(request_msg);
- proto::PredictRequest predict_request;
- std::shared_ptr<Dispatcher> dispatcher_ = Server::Instance().GetDispatcher();
- RestfulService restful_service(dispatcher_);
- Status status(INVALID_INPUTS);
- status = restful_service.ParseRequest(restful_request, &predict_request);
- ASSERT_EQ(status.StatusCode(), SUCCESS);
-
- nlohmann::json out_js;
- proto::PredictReply reply;
- auto instance_ptr = reply.add_instances();
- auto &map_item = *(instance_ptr->mutable_items());
- // test scalar:
- // scalar: key_float
- proto::Tensor tensor_float;
- ProtoTensor pb_tensor_float(&tensor_float);
- DataType type_float = kMSI_Float16;
- pb_tensor_float.set_data_type(type_float);
- pb_tensor_float.set_shape({1});
- pb_tensor_float.resize_data(pb_tensor_float.GetTypeSize(type_float));
- map_item["key_float16"] = tensor_float;
-
- Status status2 = restful_service.ParseReply(reply, &out_js);
- ASSERT_NE(status2.StatusCode(), SUCCESS);
- }
- } // namespace serving
- } // namespace mindspore
|