mirror of https://github.com/grpc/grpc.git
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