commit 357523509a16fec1994098b955783bec2917e2d2
author Craig Tiller <ctiller@google.com> Thu May 11 11:00:12 2017 -0700
committer GitHub <noreply@github.com> Thu May 11 11:00:12 2017 -0700
tree e2d5dc513a70dbb68aa928019a7a2efd07b65dc9
parent 63cde40c6dc031dcb358b7058fe3805284a6ad16
parent a1d00facbcfcd3a47d5617fd65f6ff5842c15649

Merge pull request #11074 from ctiller/pump

Finish pumped streams: should alleviate infinite hang

test/cpp/microbenchmarks/bm_fullstack_streaming_pump.cc

diff --git a/test/cpp/microbenchmarks/bm_fullstack_streaming_pump.cc b/test/cpp/microbenchmarks/bm_fullstack_streaming_pump.cc
index 47705d3..01ff391 100644
--- a/test/cpp/microbenchmarks/bm_fullstack_streaming_pump.cc
+++ b/test/cpp/microbenchmarks/bm_fullstack_streaming_pump.cc
@@ -105,6 +105,17 @@
       GPR_ASSERT(need_tags & (1 << i));
       need_tags &= ~(1 << i);
     }
+    response_rw.Finish(Status::OK, tag(0));
+    Status final_status;
+    request_rw->Finish(&final_status, tag(1));
+    need_tags = (1 << 0) | (1 << 1);
+    while (need_tags) {
+      GPR_ASSERT(fixture->cq()->Next(&t, &ok));
+      int i = (int)(intptr_t)t;
+      GPR_ASSERT(need_tags & (1 << i));
+      need_tags &= ~(1 << i);
+    }
+    GPR_ASSERT(final_status.ok());
   }
   fixture->Finish(state);
   fixture.reset();