1 /* 2 * Copyright 2022 Google LLC 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #ifndef FCP_CLIENT_HTTP_TESTING_HTTP_TEST_SERVER_H_ 18 #define FCP_CLIENT_HTTP_TESTING_HTTP_TEST_SERVER_H_ 19 20 #include <memory> 21 #include <string> 22 23 #include "absl/status/statusor.h" 24 #include "absl/strings/str_cat.h" 25 #include "tensorflow_serving/util/net_http/server/public/httpserver.h" 26 27 namespace fcp { 28 namespace client { 29 namespace http { 30 // Creates a test server where 31 // uri: the uri starting with "/" 32 // port: the port to listen 33 // num_threads: the number of parallel threads the server has 34 absl::StatusOr< 35 std::unique_ptr<::tensorflow::serving::net_http::HTTPServerInterface>> 36 CreateHttpTestServer(const std::string& uri, int port, int num_threads); 37 } // namespace http 38 } // namespace client 39 } // namespace fcp 40 41 #endif // FCP_CLIENT_HTTP_TESTING_HTTP_TEST_SERVER_H_ 42