mirror of https://github.com/grpc/grpc.git
Yash Tibrewal
6 years ago
parent
66cc56bb03
commit
62280b42c7
10 changed files with 301 additions and 149 deletions
@ -0,0 +1,178 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2018 gRPC authors. |
||||
* |
||||
* 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 "src/proto/grpc/testing/echo.grpc.pb.h" |
||||
#include "test/cpp/util/string_ref_helper.h" |
||||
|
||||
#include <gtest/gtest.h> |
||||
|
||||
namespace grpc { |
||||
namespace testing { |
||||
class EchoTestServiceStreamingImpl : public EchoTestService::Service { |
||||
public: |
||||
~EchoTestServiceStreamingImpl() override {} |
||||
|
||||
Status BidiStream( |
||||
ServerContext* context, |
||||
grpc::ServerReaderWriter<EchoResponse, EchoRequest>* stream) override { |
||||
EchoRequest req; |
||||
EchoResponse resp; |
||||
auto client_metadata = context->client_metadata(); |
||||
for (const auto& pair : client_metadata) { |
||||
context->AddTrailingMetadata(ToString(pair.first), ToString(pair.second)); |
||||
} |
||||
|
||||
while (stream->Read(&req)) { |
||||
resp.set_message(req.message()); |
||||
EXPECT_TRUE(stream->Write(resp, grpc::WriteOptions())); |
||||
} |
||||
return Status::OK; |
||||
} |
||||
|
||||
Status RequestStream(ServerContext* context, |
||||
ServerReader<EchoRequest>* reader, |
||||
EchoResponse* resp) override { |
||||
auto client_metadata = context->client_metadata(); |
||||
for (const auto& pair : client_metadata) { |
||||
context->AddTrailingMetadata(ToString(pair.first), ToString(pair.second)); |
||||
} |
||||
|
||||
EchoRequest req; |
||||
string response_str = ""; |
||||
while (reader->Read(&req)) { |
||||
response_str += req.message(); |
||||
} |
||||
resp->set_message(response_str); |
||||
return Status::OK; |
||||
} |
||||
|
||||
Status ResponseStream(ServerContext* context, const EchoRequest* req, |
||||
ServerWriter<EchoResponse>* writer) override { |
||||
auto client_metadata = context->client_metadata(); |
||||
for (const auto& pair : client_metadata) { |
||||
context->AddTrailingMetadata(ToString(pair.first), ToString(pair.second)); |
||||
} |
||||
|
||||
EchoResponse resp; |
||||
resp.set_message(req->message()); |
||||
for (int i = 0; i < 10; i++) { |
||||
EXPECT_TRUE(writer->Write(resp)); |
||||
} |
||||
return Status::OK; |
||||
} |
||||
}; |
||||
|
||||
void MakeCall(const std::shared_ptr<Channel>& channel) { |
||||
auto stub = grpc::testing::EchoTestService::NewStub(channel); |
||||
ClientContext ctx; |
||||
EchoRequest req; |
||||
req.mutable_param()->set_echo_metadata(true); |
||||
ctx.AddMetadata("testkey", "testvalue"); |
||||
req.set_message("Hello"); |
||||
EchoResponse resp; |
||||
Status s = stub->Echo(&ctx, req, &resp); |
||||
EXPECT_EQ(s.ok(), true); |
||||
EXPECT_EQ(resp.message(), "Hello"); |
||||
} |
||||
|
||||
void MakeClientStreamingCall(const std::shared_ptr<Channel>& channel) { |
||||
auto stub = grpc::testing::EchoTestService::NewStub(channel); |
||||
ClientContext ctx; |
||||
EchoRequest req; |
||||
req.mutable_param()->set_echo_metadata(true); |
||||
ctx.AddMetadata("testkey", "testvalue"); |
||||
req.set_message("Hello"); |
||||
EchoResponse resp; |
||||
string expected_resp = ""; |
||||
auto writer = stub->RequestStream(&ctx, &resp); |
||||
for (int i = 0; i < 10; i++) { |
||||
writer->Write(req); |
||||
expected_resp += "Hello"; |
||||
} |
||||
writer->WritesDone(); |
||||
Status s = writer->Finish(); |
||||
EXPECT_EQ(s.ok(), true); |
||||
EXPECT_EQ(resp.message(), expected_resp); |
||||
} |
||||
|
||||
void MakeServerStreamingCall(const std::shared_ptr<Channel>& channel) { |
||||
auto stub = grpc::testing::EchoTestService::NewStub(channel); |
||||
ClientContext ctx; |
||||
EchoRequest req; |
||||
req.mutable_param()->set_echo_metadata(true); |
||||
ctx.AddMetadata("testkey", "testvalue"); |
||||
req.set_message("Hello"); |
||||
EchoResponse resp; |
||||
string expected_resp = ""; |
||||
auto reader = stub->ResponseStream(&ctx, req); |
||||
int count = 0; |
||||
while (reader->Read(&resp)) { |
||||
EXPECT_EQ(resp.message(), "Hello"); |
||||
count++; |
||||
} |
||||
ASSERT_EQ(count, 10); |
||||
Status s = reader->Finish(); |
||||
EXPECT_EQ(s.ok(), true); |
||||
} |
||||
|
||||
void MakeBidiStreamingCall(const std::shared_ptr<Channel>& channel) { |
||||
auto stub = grpc::testing::EchoTestService::NewStub(channel); |
||||
ClientContext ctx; |
||||
EchoRequest req; |
||||
EchoResponse resp; |
||||
ctx.AddMetadata("testkey", "testvalue"); |
||||
auto stream = stub->BidiStream(&ctx); |
||||
for (auto i = 0; i < 10; i++) { |
||||
req.set_message("Hello" + std::to_string(i)); |
||||
stream->Write(req); |
||||
stream->Read(&resp); |
||||
EXPECT_EQ(req.message(), resp.message()); |
||||
} |
||||
ASSERT_TRUE(stream->WritesDone()); |
||||
Status s = stream->Finish(); |
||||
EXPECT_EQ(s.ok(), true); |
||||
} |
||||
|
||||
void MakeCallbackCall(const std::shared_ptr<Channel>& channel) { |
||||
auto stub = grpc::testing::EchoTestService::NewStub(channel); |
||||
ClientContext ctx; |
||||
EchoRequest req; |
||||
std::mutex mu; |
||||
std::condition_variable cv; |
||||
bool done = false; |
||||
req.mutable_param()->set_echo_metadata(true); |
||||
ctx.AddMetadata("testkey", "testvalue"); |
||||
req.set_message("Hello"); |
||||
EchoResponse resp; |
||||
stub->experimental_async()->Echo(&ctx, &req, &resp, |
||||
[&resp, &mu, &done, &cv](Status s) { |
||||
// gpr_log(GPR_ERROR, "got the callback");
|
||||
EXPECT_EQ(s.ok(), true); |
||||
EXPECT_EQ(resp.message(), "Hello"); |
||||
std::lock_guard<std::mutex> l(mu); |
||||
done = true; |
||||
cv.notify_one(); |
||||
}); |
||||
std::unique_lock<std::mutex> l(mu); |
||||
while (!done) { |
||||
cv.wait(l); |
||||
} |
||||
} |
||||
|
||||
} // namespace testing
|
||||
} // namespace grpc
|
||||
Loading…
Reference in new issue