Yash Tibrewal
6 years ago
parent
4ed889382a
commit
94e7edad99
14 changed files with 368 additions and 0 deletions
@ -0,0 +1,87 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2019 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef GRPCPP_IMPL_CODEGEN_DELEGATING_CHANNEL_H |
||||||
|
#define GRPCPP_IMPL_CODEGEN_DELEGATING_CHANNEL_H |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
namespace internal { |
||||||
|
|
||||||
|
class DelegatingChannel : public ::grpc::ChannelInterface { |
||||||
|
public: |
||||||
|
virtual ~DelegatingChannel() {} |
||||||
|
|
||||||
|
DelegatingChannel(std::shared_ptr<::grpc::ChannelInterface> delegate_channel) |
||||||
|
: delegate_channel_(delegate_channel) {} |
||||||
|
|
||||||
|
grpc_connectivity_state GetState(bool try_to_connect) override { |
||||||
|
delegate_channel()->GetState(try_to_connect); |
||||||
|
} |
||||||
|
|
||||||
|
std::shared_ptr<::grpc::ChannelInterface> delegate_channel() { |
||||||
|
return delegate_channel_; |
||||||
|
} |
||||||
|
|
||||||
|
private: |
||||||
|
internal::Call CreateCall(const internal::RpcMethod& method, |
||||||
|
ClientContext* context, |
||||||
|
::grpc_impl::CompletionQueue* cq) override { |
||||||
|
return delegate_channel()->CreateCall(method, context, cq); |
||||||
|
} |
||||||
|
|
||||||
|
void PerformOpsOnCall(internal::CallOpSetInterface* ops, |
||||||
|
internal::Call* call) override { |
||||||
|
delegate_channel()->PerformOpsOnCall(ops, call); |
||||||
|
} |
||||||
|
|
||||||
|
void* RegisterMethod(const char* method) override { |
||||||
|
return delegate_channel()->RegisterMethod(method); |
||||||
|
} |
||||||
|
|
||||||
|
void NotifyOnStateChangeImpl(grpc_connectivity_state last_observed, |
||||||
|
gpr_timespec deadline, |
||||||
|
::grpc_impl::CompletionQueue* cq, |
||||||
|
void* tag) override { |
||||||
|
delegate_channel()->NotifyOnStateChangeImpl(last_observed, deadline, cq, |
||||||
|
tag); |
||||||
|
} |
||||||
|
|
||||||
|
bool WaitForStateChangeImpl(grpc_connectivity_state last_observed, |
||||||
|
gpr_timespec deadline) override { |
||||||
|
return delegate_channel()->WaitForStateChangeImpl(last_observed, deadline); |
||||||
|
} |
||||||
|
|
||||||
|
internal::Call CreateCallInternal(const internal::RpcMethod& method, |
||||||
|
ClientContext* context, |
||||||
|
::grpc_impl::CompletionQueue* cq, |
||||||
|
size_t interceptor_pos) override { |
||||||
|
return delegate_channel()->CreateCallInternal(method, context, cq, |
||||||
|
interceptor_pos); |
||||||
|
} |
||||||
|
|
||||||
|
::grpc_impl::CompletionQueue* CallbackCQ() override { |
||||||
|
return delegate_channel()->CallbackCQ(); |
||||||
|
} |
||||||
|
|
||||||
|
std::shared_ptr<::grpc::ChannelInterface> delegate_channel_; |
||||||
|
}; |
||||||
|
|
||||||
|
} // namespace internal
|
||||||
|
} // namespace grpc
|
||||||
|
|
||||||
|
#endif // GRPCPP_IMPL_CODEGEN_DELEGATING_CHANNEL_H
|
||||||
@ -0,0 +1,100 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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 internal::DelegatingChannel { |
||||||
|
public: |
||||||
|
TestChannel(std::shared_ptr<ChannelInterface> delegate_channel) |
||||||
|
: internal::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(); |
||||||
|
} |
||||||
Loading…
Reference in new issue