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