diff options
author | Craig Tiller <ctiller@google.com> | 2015-02-23 12:53:39 -0800 |
---|---|---|
committer | Craig Tiller <ctiller@google.com> | 2015-02-23 12:53:39 -0800 |
commit | fd1b49b7572049989eef0b70d8a328b60cfa23f2 (patch) | |
tree | eb46d461856b860b9929e15384cdf81251778e00 /test/cpp | |
parent | 7aef9e60f373c899604363e617baaed5b37a154f (diff) |
Move to unique_ptr for all GRPC returned objects
Diffstat (limited to 'test/cpp')
-rw-r--r-- | test/cpp/end2end/async_end2end_test.cc | 2 | ||||
-rw-r--r-- | test/cpp/end2end/end2end_test.cc | 25 |
2 files changed, 7 insertions, 20 deletions
diff --git a/test/cpp/end2end/async_end2end_test.cc b/test/cpp/end2end/async_end2end_test.cc index 248e054e49..6dca0aa348 100644 --- a/test/cpp/end2end/async_end2end_test.cc +++ b/test/cpp/end2end/async_end2end_test.cc @@ -106,7 +106,7 @@ class AsyncEnd2endTest : public ::testing::Test { void ResetStub() { std::shared_ptr<ChannelInterface> channel = CreateChannel(server_address_.str(), ChannelArguments()); - stub_.reset(grpc::cpp::test::util::TestService::NewStub(channel)); + stub_ = std::move(grpc::cpp::test::util::TestService::NewStub(channel)); } void server_ok(int i) { diff --git a/test/cpp/end2end/end2end_test.cc b/test/cpp/end2end/end2end_test.cc index d4ca3ef49e..8b3fb4d485 100644 --- a/test/cpp/end2end/end2end_test.cc +++ b/test/cpp/end2end/end2end_test.cc @@ -162,7 +162,7 @@ class End2endTest : public ::testing::Test { void ResetStub() { std::shared_ptr<ChannelInterface> channel = CreateChannel(server_address_.str(), ChannelArguments()); - stub_.reset(grpc::cpp::test::util::TestService::NewStub(channel)); + stub_ = std::move(grpc::cpp::test::util::TestService::NewStub(channel)); } std::unique_ptr<grpc::cpp::test::util::TestService::Stub> stub_; @@ -292,15 +292,13 @@ TEST_F(End2endTest, RequestStreamOneRequest) { EchoResponse response; ClientContext context; - ClientWriter<EchoRequest>* stream = stub_->RequestStream(&context, &response); + auto stream = stub_->RequestStream(&context, &response); request.set_message("hello"); EXPECT_TRUE(stream->Write(request)); stream->WritesDone(); Status s = stream->Finish(); EXPECT_EQ(response.message(), request.message()); EXPECT_TRUE(s.IsOk()); - - delete stream; } TEST_F(End2endTest, RequestStreamTwoRequests) { @@ -309,7 +307,7 @@ TEST_F(End2endTest, RequestStreamTwoRequests) { EchoResponse response; ClientContext context; - ClientWriter<EchoRequest>* stream = stub_->RequestStream(&context, &response); + auto stream = stub_->RequestStream(&context, &response); request.set_message("hello"); EXPECT_TRUE(stream->Write(request)); EXPECT_TRUE(stream->Write(request)); @@ -317,8 +315,6 @@ TEST_F(End2endTest, RequestStreamTwoRequests) { Status s = stream->Finish(); EXPECT_EQ(response.message(), "hellohello"); EXPECT_TRUE(s.IsOk()); - - delete stream; } TEST_F(End2endTest, ResponseStream) { @@ -328,8 +324,7 @@ TEST_F(End2endTest, ResponseStream) { ClientContext context; request.set_message("hello"); - ClientReader<EchoResponse>* stream = - stub_->ResponseStream(&context, request); + auto stream = stub_->ResponseStream(&context, request); EXPECT_TRUE(stream->Read(&response)); EXPECT_EQ(response.message(), request.message() + "0"); EXPECT_TRUE(stream->Read(&response)); @@ -340,8 +335,6 @@ TEST_F(End2endTest, ResponseStream) { Status s = stream->Finish(); EXPECT_TRUE(s.IsOk()); - - delete stream; } TEST_F(End2endTest, BidiStream) { @@ -351,8 +344,7 @@ TEST_F(End2endTest, BidiStream) { ClientContext context; grpc::string msg("hello"); - ClientReaderWriter<EchoRequest, EchoResponse>* stream = - stub_->BidiStream(&context); + auto stream = stub_->BidiStream(&context); request.set_message(msg + "0"); EXPECT_TRUE(stream->Write(request)); @@ -374,8 +366,6 @@ TEST_F(End2endTest, BidiStream) { Status s = stream->Finish(); EXPECT_TRUE(s.IsOk()); - - delete stream; } // Talk to the two services with the same name but different package names. @@ -426,14 +416,11 @@ TEST_F(End2endTest, BadCredentials) { EXPECT_EQ("Rpc sent on a lame channel.", s.details()); ClientContext context2; - ClientReaderWriter<EchoRequest, EchoResponse>* stream = - stub->BidiStream(&context2); + auto stream = stub->BidiStream(&context2); s = stream->Finish(); EXPECT_FALSE(s.IsOk()); EXPECT_EQ(StatusCode::UNKNOWN, s.code()); EXPECT_EQ("Rpc sent on a lame channel.", s.details()); - - delete stream; } } // namespace testing |