diff options
Diffstat (limited to 'test/cpp/interop/interop_client.cc')
-rw-r--r-- | test/cpp/interop/interop_client.cc | 47 |
1 files changed, 16 insertions, 31 deletions
diff --git a/test/cpp/interop/interop_client.cc b/test/cpp/interop/interop_client.cc index 5169945762..fd01ad0666 100644 --- a/test/cpp/interop/interop_client.cc +++ b/test/cpp/interop/interop_client.cc @@ -83,7 +83,7 @@ CompressionType GetInteropCompressionTypeFromCompressionAlgorithm( } // namespace InteropClient::InteropClient(std::shared_ptr<Channel> channel) - : channel_(channel) {} + : channel_(channel), stub_(TestService::NewStub(channel)) {} void InteropClient::AssertOkOrPrintErrorStatus(const Status& s) { if (s.ok()) { @@ -96,13 +96,12 @@ void InteropClient::AssertOkOrPrintErrorStatus(const Status& s) { void InteropClient::DoEmpty() { gpr_log(GPR_INFO, "Sending an empty rpc..."); - std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); Empty request = Empty::default_instance(); Empty response = Empty::default_instance(); ClientContext context; - Status s = stub->EmptyCall(&context, request, &response); + Status s = stub_->EmptyCall(&context, request, &response); AssertOkOrPrintErrorStatus(s); gpr_log(GPR_INFO, "Empty rpc done."); @@ -111,7 +110,6 @@ void InteropClient::DoEmpty() { // Shared code to set large payload, make rpc and check response payload. void InteropClient::PerformLargeUnary(SimpleRequest* request, SimpleResponse* response) { - std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); ClientContext context; InteropClientContextInspector inspector(context); @@ -121,7 +119,7 @@ void InteropClient::PerformLargeUnary(SimpleRequest* request, grpc::string payload(kLargeRequestSize, '\0'); request->mutable_payload()->set_body(payload.c_str(), kLargeRequestSize); - Status s = stub->UnaryCall(&context, *request, response); + Status s = stub_->UnaryCall(&context, *request, response); // Compression related checks. GPR_ASSERT(request->response_compression() == @@ -187,11 +185,10 @@ void InteropClient::DoOauth2AuthToken(const grpc::string& username, SimpleResponse response; request.set_fill_username(true); request.set_fill_oauth_scope(true); - std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); ClientContext context; - Status s = stub->UnaryCall(&context, request, &response); + Status s = stub_->UnaryCall(&context, request, &response); AssertOkOrPrintErrorStatus(s); GPR_ASSERT(!response.username().empty()); @@ -207,7 +204,6 @@ void InteropClient::DoPerRpcCreds(const grpc::string& json_key) { SimpleRequest request; SimpleResponse response; request.set_fill_username(true); - std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); ClientContext context; std::chrono::seconds token_lifetime = std::chrono::hours(1); @@ -216,7 +212,7 @@ void InteropClient::DoPerRpcCreds(const grpc::string& json_key) { context.set_credentials(creds); - Status s = stub->UnaryCall(&context, request, &response); + Status s = stub_->UnaryCall(&context, request, &response); AssertOkOrPrintErrorStatus(s); GPR_ASSERT(!response.username().empty()); @@ -269,14 +265,13 @@ void InteropClient::DoLargeCompressedUnary() { void InteropClient::DoRequestStreaming() { gpr_log(GPR_INFO, "Sending request steaming rpc ..."); - std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); ClientContext context; StreamingInputCallRequest request; StreamingInputCallResponse response; std::unique_ptr<ClientWriter<StreamingInputCallRequest>> stream( - stub->StreamingInputCall(&context, &response)); + stub_->StreamingInputCall(&context, &response)); int aggregated_payload_size = 0; for (unsigned int i = 0; i < request_stream_sizes.size(); ++i) { @@ -295,7 +290,6 @@ void InteropClient::DoRequestStreaming() { void InteropClient::DoResponseStreaming() { gpr_log(GPR_INFO, "Receiving response steaming rpc ..."); - std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); ClientContext context; StreamingOutputCallRequest request; @@ -305,7 +299,7 @@ void InteropClient::DoResponseStreaming() { } StreamingOutputCallResponse response; std::unique_ptr<ClientReader<StreamingOutputCallResponse>> stream( - stub->StreamingOutputCall(&context, request)); + stub_->StreamingOutputCall(&context, request)); unsigned int i = 0; while (stream->Read(&response)) { @@ -320,7 +314,6 @@ void InteropClient::DoResponseStreaming() { } void InteropClient::DoResponseCompressedStreaming() { - std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); const CompressionType compression_types[] = {NONE, GZIP, DEFLATE}; const PayloadType payload_types[] = {COMPRESSABLE, UNCOMPRESSABLE, RANDOM}; @@ -348,7 +341,7 @@ void InteropClient::DoResponseCompressedStreaming() { StreamingOutputCallResponse response; std::unique_ptr<ClientReader<StreamingOutputCallResponse>> stream( - stub->StreamingOutputCall(&context, request)); + stub_->StreamingOutputCall(&context, request)); size_t k = 0; while (stream->Read(&response)) { @@ -401,7 +394,6 @@ void InteropClient::DoResponseCompressedStreaming() { void InteropClient::DoResponseStreamingWithSlowConsumer() { gpr_log(GPR_INFO, "Receiving response steaming rpc with slow consumer ..."); - std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); ClientContext context; StreamingOutputCallRequest request; @@ -412,7 +404,7 @@ void InteropClient::DoResponseStreamingWithSlowConsumer() { } StreamingOutputCallResponse response; std::unique_ptr<ClientReader<StreamingOutputCallResponse>> stream( - stub->StreamingOutputCall(&context, request)); + stub_->StreamingOutputCall(&context, request)); int i = 0; while (stream->Read(&response)) { @@ -431,12 +423,11 @@ void InteropClient::DoResponseStreamingWithSlowConsumer() { void InteropClient::DoHalfDuplex() { gpr_log(GPR_INFO, "Sending half-duplex streaming rpc ..."); - std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); ClientContext context; std::unique_ptr<ClientReaderWriter<StreamingOutputCallRequest, StreamingOutputCallResponse>> - stream(stub->HalfDuplexCall(&context)); + stream(stub_->HalfDuplexCall(&context)); StreamingOutputCallRequest request; ResponseParameters* response_parameter = request.add_response_parameters(); @@ -461,12 +452,11 @@ void InteropClient::DoHalfDuplex() { void InteropClient::DoPingPong() { gpr_log(GPR_INFO, "Sending Ping Pong streaming rpc ..."); - std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); ClientContext context; std::unique_ptr<ClientReaderWriter<StreamingOutputCallRequest, StreamingOutputCallResponse>> - stream(stub->FullDuplexCall(&context)); + stream(stub_->FullDuplexCall(&context)); StreamingOutputCallRequest request; request.set_response_type(PayloadType::COMPRESSABLE); @@ -491,14 +481,13 @@ void InteropClient::DoPingPong() { void InteropClient::DoCancelAfterBegin() { gpr_log(GPR_INFO, "Sending request steaming rpc ..."); - std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); ClientContext context; StreamingInputCallRequest request; StreamingInputCallResponse response; std::unique_ptr<ClientWriter<StreamingInputCallRequest>> stream( - stub->StreamingInputCall(&context, &response)); + stub_->StreamingInputCall(&context, &response)); gpr_log(GPR_INFO, "Trying to cancel..."); context.TryCancel(); @@ -509,12 +498,11 @@ void InteropClient::DoCancelAfterBegin() { void InteropClient::DoCancelAfterFirstResponse() { gpr_log(GPR_INFO, "Sending Ping Pong streaming rpc ..."); - std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); ClientContext context; std::unique_ptr<ClientReaderWriter<StreamingOutputCallRequest, StreamingOutputCallResponse>> - stream(stub->FullDuplexCall(&context)); + stream(stub_->FullDuplexCall(&context)); StreamingOutputCallRequest request; request.set_response_type(PayloadType::COMPRESSABLE); @@ -534,7 +522,6 @@ void InteropClient::DoCancelAfterFirstResponse() { void InteropClient::DoTimeoutOnSleepingServer() { gpr_log(GPR_INFO, "Sending Ping Pong streaming rpc with a short deadline..."); - std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); ClientContext context; std::chrono::system_clock::time_point deadline = @@ -542,7 +529,7 @@ void InteropClient::DoTimeoutOnSleepingServer() { context.set_deadline(deadline); std::unique_ptr<ClientReaderWriter<StreamingOutputCallRequest, StreamingOutputCallResponse>> - stream(stub->FullDuplexCall(&context)); + stream(stub_->FullDuplexCall(&context)); StreamingOutputCallRequest request; request.mutable_payload()->set_body(grpc::string(27182, '\0')); @@ -555,12 +542,11 @@ void InteropClient::DoTimeoutOnSleepingServer() { void InteropClient::DoEmptyStream() { gpr_log(GPR_INFO, "Starting empty_stream."); - std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); ClientContext context; std::unique_ptr<ClientReaderWriter<StreamingOutputCallRequest, StreamingOutputCallResponse>> - stream(stub->FullDuplexCall(&context)); + stream(stub_->FullDuplexCall(&context)); stream->WritesDone(); StreamingOutputCallResponse response; GPR_ASSERT(stream->Read(&response) == false); @@ -571,7 +557,6 @@ void InteropClient::DoEmptyStream() { void InteropClient::DoStatusWithMessage() { gpr_log(GPR_INFO, "Sending RPC with a request for status code 2 and message"); - std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); ClientContext context; SimpleRequest request; @@ -581,7 +566,7 @@ void InteropClient::DoStatusWithMessage() { grpc::string test_msg = "This is a test message"; requested_status->set_message(test_msg); - Status s = stub->UnaryCall(&context, request, &response); + Status s = stub_->UnaryCall(&context, request, &response); GPR_ASSERT(s.error_code() == grpc::StatusCode::UNKNOWN); GPR_ASSERT(s.error_message() == test_msg); |