|
|
|
|
@ -60,6 +60,8 @@ class ClientInterceptorsEnd2endTest : public ::testing::Test { |
|
|
|
|
std::unique_ptr<Server> server_; |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
/* This interceptor does nothing. Just keeps a global count on the number of
|
|
|
|
|
* times it was invoked. */ |
|
|
|
|
class DummyInterceptor : public experimental::Interceptor { |
|
|
|
|
public: |
|
|
|
|
DummyInterceptor(experimental::ClientRpcInfo* info) {} |
|
|
|
|
@ -91,6 +93,7 @@ class DummyInterceptorFactory |
|
|
|
|
} |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
/* Hijacks Echo RPC and fills in the expected values */ |
|
|
|
|
class HijackingInterceptor : public experimental::Interceptor { |
|
|
|
|
public: |
|
|
|
|
HijackingInterceptor(experimental::ClientRpcInfo* info) { |
|
|
|
|
@ -195,6 +198,111 @@ class HijackingInterceptorFactory |
|
|
|
|
} |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
class HijackingInterceptorMakesAnotherCall : public experimental::Interceptor { |
|
|
|
|
public: |
|
|
|
|
HijackingInterceptorMakesAnotherCall(experimental::ClientRpcInfo* info) { |
|
|
|
|
info_ = info; |
|
|
|
|
// Make sure it is the right method
|
|
|
|
|
EXPECT_EQ(strcmp("/grpc.testing.EchoTestService/Echo", info->method()), 0); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
virtual void Intercept(experimental::InterceptorBatchMethods* methods) { |
|
|
|
|
gpr_log(GPR_ERROR, "ran this"); |
|
|
|
|
bool hijack = false; |
|
|
|
|
if (methods->QueryInterceptionHookPoint( |
|
|
|
|
experimental::InterceptionHookPoints::PRE_SEND_INITIAL_METADATA)) { |
|
|
|
|
auto* map = methods->GetSendInitialMetadata(); |
|
|
|
|
// Check that we can see the test metadata
|
|
|
|
|
ASSERT_EQ(map->size(), 1); |
|
|
|
|
auto iterator = map->begin(); |
|
|
|
|
EXPECT_EQ("testkey", iterator->first); |
|
|
|
|
EXPECT_EQ("testvalue", iterator->second); |
|
|
|
|
hijack = true; |
|
|
|
|
// Make a copy of the map
|
|
|
|
|
metadata_map_ = *map; |
|
|
|
|
} |
|
|
|
|
if (methods->QueryInterceptionHookPoint( |
|
|
|
|
experimental::InterceptionHookPoints::PRE_SEND_MESSAGE)) { |
|
|
|
|
EchoRequest req; |
|
|
|
|
auto* buffer = methods->GetSendMessage(); |
|
|
|
|
auto copied_buffer = *buffer; |
|
|
|
|
SerializationTraits<EchoRequest>::Deserialize(&copied_buffer, &req); |
|
|
|
|
EXPECT_EQ(req.message(), "Hello"); |
|
|
|
|
auto stub = grpc::testing::EchoTestService::NewStub( |
|
|
|
|
std::shared_ptr<Channel>(info_->channel())); |
|
|
|
|
ClientContext ctx; |
|
|
|
|
EchoResponse resp; |
|
|
|
|
Status s = stub->Echo(&ctx, req, &resp); |
|
|
|
|
EXPECT_EQ(s.ok(), true); |
|
|
|
|
EXPECT_EQ(resp.message(), "Hello"); |
|
|
|
|
} |
|
|
|
|
if (methods->QueryInterceptionHookPoint( |
|
|
|
|
experimental::InterceptionHookPoints::PRE_SEND_CLOSE)) { |
|
|
|
|
// Got nothing to do here for now
|
|
|
|
|
} |
|
|
|
|
if (methods->QueryInterceptionHookPoint( |
|
|
|
|
experimental::InterceptionHookPoints::POST_RECV_INITIAL_METADATA)) { |
|
|
|
|
auto* map = methods->GetRecvInitialMetadata(); |
|
|
|
|
// Got nothing better to do here for now
|
|
|
|
|
EXPECT_EQ(map->size(), 0); |
|
|
|
|
} |
|
|
|
|
if (methods->QueryInterceptionHookPoint( |
|
|
|
|
experimental::InterceptionHookPoints::POST_RECV_MESSAGE)) { |
|
|
|
|
EchoResponse* resp = |
|
|
|
|
static_cast<EchoResponse*>(methods->GetRecvMessage()); |
|
|
|
|
// Check that we got the hijacked message, and re-insert the expected
|
|
|
|
|
// message
|
|
|
|
|
EXPECT_EQ(resp->message(), "Hello1"); |
|
|
|
|
resp->set_message("Hello"); |
|
|
|
|
} |
|
|
|
|
if (methods->QueryInterceptionHookPoint( |
|
|
|
|
experimental::InterceptionHookPoints::POST_RECV_STATUS)) { |
|
|
|
|
auto* map = methods->GetRecvTrailingMetadata(); |
|
|
|
|
bool found = false; |
|
|
|
|
// Check that we received the metadata as an echo
|
|
|
|
|
for (const auto& pair : *map) { |
|
|
|
|
found = pair.first.starts_with("testkey") && |
|
|
|
|
pair.second.starts_with("testvalue"); |
|
|
|
|
if (found) break; |
|
|
|
|
} |
|
|
|
|
EXPECT_EQ(found, true); |
|
|
|
|
auto* status = methods->GetRecvStatus(); |
|
|
|
|
EXPECT_EQ(status->ok(), true); |
|
|
|
|
} |
|
|
|
|
if (methods->QueryInterceptionHookPoint( |
|
|
|
|
experimental::InterceptionHookPoints::PRE_RECV_INITIAL_METADATA)) { |
|
|
|
|
auto* map = methods->GetRecvInitialMetadata(); |
|
|
|
|
// Got nothing better to do here at the moment
|
|
|
|
|
EXPECT_EQ(map->size(), 0); |
|
|
|
|
} |
|
|
|
|
if (methods->QueryInterceptionHookPoint( |
|
|
|
|
experimental::InterceptionHookPoints::PRE_RECV_MESSAGE)) { |
|
|
|
|
// Insert a different message than expected
|
|
|
|
|
EchoResponse* resp = |
|
|
|
|
static_cast<EchoResponse*>(methods->GetRecvMessage()); |
|
|
|
|
resp->set_message("Hello1"); |
|
|
|
|
} |
|
|
|
|
if (methods->QueryInterceptionHookPoint( |
|
|
|
|
experimental::InterceptionHookPoints::PRE_RECV_STATUS)) { |
|
|
|
|
auto* map = methods->GetRecvTrailingMetadata(); |
|
|
|
|
// insert the metadata that we want
|
|
|
|
|
EXPECT_EQ(map->size(), 0); |
|
|
|
|
map->insert(std::make_pair("testkey", "testvalue")); |
|
|
|
|
auto* status = methods->GetRecvStatus(); |
|
|
|
|
*status = Status(StatusCode::OK, ""); |
|
|
|
|
} |
|
|
|
|
if (hijack) { |
|
|
|
|
methods->Hijack(); |
|
|
|
|
} else { |
|
|
|
|
methods->Proceed(); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
private: |
|
|
|
|
experimental::ClientRpcInfo* info_; |
|
|
|
|
std::multimap<grpc::string, grpc::string> metadata_map_; |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
class LoggingInterceptor : public experimental::Interceptor { |
|
|
|
|
public: |
|
|
|
|
LoggingInterceptor(experimental::ClientRpcInfo* info) { |
|
|
|
|
@ -268,6 +376,19 @@ class LoggingInterceptorFactory |
|
|
|
|
} |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
void MakeCall(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"); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
TEST_F(ClientInterceptorsEnd2endTest, ClientInterceptorLoggingTest) { |
|
|
|
|
ChannelArguments args; |
|
|
|
|
DummyInterceptor::Reset(); |
|
|
|
|
@ -284,16 +405,7 @@ TEST_F(ClientInterceptorsEnd2endTest, ClientInterceptorLoggingTest) { |
|
|
|
|
} |
|
|
|
|
auto channel = experimental::CreateCustomChannelWithInterceptors( |
|
|
|
|
server_address_, InsecureChannelCredentials(), args, std::move(creators)); |
|
|
|
|
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"); |
|
|
|
|
MakeCall(channel); |
|
|
|
|
// Make sure all 20 dummy interceptors were run
|
|
|
|
|
EXPECT_EQ(DummyInterceptor::GetNumTimesRun(), 20); |
|
|
|
|
} |
|
|
|
|
|
Yash Tibrewal
7 years ago