mirror of https://github.com/grpc/grpc.git
The C based gRPC (C++, Python, Ruby, Objective-C, PHP, C#)
https://grpc.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
3.1 KiB
100 lines
3.1 KiB
/* |
|
* |
|
* 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 <memory> |
|
#include <vector> |
|
|
|
#include <grpcpp/channel.h> |
|
#include <grpcpp/client_context.h> |
|
#include <grpcpp/create_channel.h> |
|
#include <grpcpp/generic/generic_stub.h> |
|
#include <grpcpp/impl/codegen/delegating_channel.h> |
|
#include <grpcpp/impl/codegen/proto_utils.h> |
|
#include <grpcpp/server.h> |
|
#include <grpcpp/server_builder.h> |
|
#include <grpcpp/server_context.h> |
|
#include <grpcpp/support/client_interceptor.h> |
|
|
|
#include "src/proto/grpc/testing/echo.grpc.pb.h" |
|
#include "test/core/util/port.h" |
|
#include "test/core/util/test_config.h" |
|
#include "test/cpp/end2end/test_service_impl.h" |
|
#include "test/cpp/util/byte_buffer_proto_helper.h" |
|
#include "test/cpp/util/string_ref_helper.h" |
|
|
|
#include <gtest/gtest.h> |
|
|
|
namespace grpc { |
|
namespace testing { |
|
namespace { |
|
|
|
class TestChannel : public experimental::DelegatingChannel { |
|
public: |
|
TestChannel(const std::shared_ptr<ChannelInterface>& delegate_channel) |
|
: experimental::DelegatingChannel(delegate_channel) {} |
|
// Always returns GRPC_CHANNEL_READY |
|
grpc_connectivity_state GetState(bool try_to_connect) override { |
|
return GRPC_CHANNEL_READY; |
|
} |
|
}; |
|
|
|
class DelegatingChannelTest : public ::testing::Test { |
|
protected: |
|
DelegatingChannelTest() { |
|
int port = grpc_pick_unused_port_or_die(); |
|
ServerBuilder builder; |
|
server_address_ = "localhost:" + std::to_string(port); |
|
builder.AddListeningPort(server_address_, InsecureServerCredentials()); |
|
builder.RegisterService(&service_); |
|
server_ = builder.BuildAndStart(); |
|
} |
|
|
|
~DelegatingChannelTest() { server_->Shutdown(); } |
|
|
|
std::string server_address_; |
|
TestServiceImpl service_; |
|
std::unique_ptr<Server> server_; |
|
}; |
|
|
|
TEST_F(DelegatingChannelTest, SimpleTest) { |
|
auto channel = CreateChannel(server_address_, InsecureChannelCredentials()); |
|
std::shared_ptr<TestChannel> test_channel = |
|
std::make_shared<TestChannel>(channel); |
|
// gRPC channel should be in idle state at this point but our test channel |
|
// will return ready. |
|
EXPECT_EQ(channel->GetState(false), GRPC_CHANNEL_IDLE); |
|
EXPECT_EQ(test_channel->GetState(false), GRPC_CHANNEL_READY); |
|
auto stub = grpc::testing::EchoTestService::NewStub(test_channel); |
|
ClientContext ctx; |
|
EchoRequest req; |
|
req.set_message("Hello"); |
|
EchoResponse resp; |
|
Status s = stub->Echo(&ctx, req, &resp); |
|
EXPECT_EQ(s.ok(), true); |
|
EXPECT_EQ(resp.message(), "Hello"); |
|
} |
|
|
|
} // namespace |
|
} // namespace testing |
|
} // namespace grpc |
|
|
|
int main(int argc, char** argv) { |
|
grpc::testing::TestEnvironment env(argc, argv); |
|
::testing::InitGoogleTest(&argc, argv); |
|
return RUN_ALL_TESTS(); |
|
}
|
|
|