diff options
author | 2016-02-03 07:38:36 -0800 | |
---|---|---|
committer | 2016-02-03 07:41:31 -0800 | |
commit | 9bc6fa0a40edb9014922e579346a6fce888303d2 (patch) | |
tree | 37a529410b8961b38974d41141eec7a933abb18d /test/cpp/end2end/test_service_impl.h | |
parent | a0a8eaab0e86148f40fb629a36c0529eea9e1b35 (diff) | |
parent | 6b4ec07ec9028e6c4727a3f1b83a166087d44f11 (diff) |
Merge branch 'master' into server_try_cancel_api
Diffstat (limited to 'test/cpp/end2end/test_service_impl.h')
-rw-r--r-- | test/cpp/end2end/test_service_impl.h | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/test/cpp/end2end/test_service_impl.h b/test/cpp/end2end/test_service_impl.h index 2c35b5614c..1ab6ced9e0 100644 --- a/test/cpp/end2end/test_service_impl.h +++ b/test/cpp/end2end/test_service_impl.h @@ -44,7 +44,16 @@ namespace grpc { namespace testing { +const int kNumResponseStreamsMsgs = 3; const char* const kServerCancelAfterReads = "cancel_after_reads"; +const char* const kServerTryCancelRequest = "server_try_cancel"; + +typedef enum { + DO_NOT_CANCEL = 0, + CANCEL_BEFORE_PROCESSING, + CANCEL_DURING_PROCESSING, + CANCEL_AFTER_PROCESSING +} ServerTryCancelRequestPhase; class TestServiceImpl : public ::grpc::testing::EchoTestService::Service { public: @@ -74,6 +83,14 @@ class TestServiceImpl : public ::grpc::testing::EchoTestService::Service { } private: + int GetIntValueFromMetadata( + const char* key, + const std::multimap<grpc::string_ref, grpc::string_ref>& metadata, + int default_value); + + void ServerTryCancel(ServerContext* context); + + private: bool signal_client_; std::mutex mu_; std::unique_ptr<grpc::string> host_; |