commit b761f4c9400b831906c89aa88430928bcf477227
author fayang <fayang@google.com> Wed May 06 11:58:49 2020 -0700
committer Copybara-Service <copybara-worker@google.com> Wed May 06 11:59:25 2020 -0700
tree 1b981ed5eec6dbd4ba48c91dc637dd0518aa849c
parent c31106bdd2b981551ec568937ca550641ce12756

Use WaitUntil in EndToEndTest.SimpleStopSendingTest to fix the flakiness.

PiperOrigin-RevId: 310198911
Change-Id: I67a0131c145897e2ddc8a28b8fda579804107d53

quic/core/http/end_to_end_test.cc

diff --git a/quic/core/http/end_to_end_test.cc b/quic/core/http/end_to_end_test.cc
index ccbe3af..7490e28 100644
--- a/quic/core/http/end_to_end_test.cc
+++ b/quic/core/http/end_to_end_test.cc
@@ -4016,7 +4016,8 @@
   // Ensure the stream has been write closed upon receiving STOP_SENDING.
   EXPECT_EQ(stream_id, client_stream->id());
   EXPECT_TRUE(client_stream->write_side_closed());
-  EXPECT_LT(0u, observer.num_stop_sending_frames());
+  client_->WaitUntil(
+      -1, [&observer]() { return observer.num_stop_sending_frames() > 0; });
   if (!observer.stream_write_side_closed_before_receiving_stop_sending()) {
     EXPECT_EQ(kStopSendingTestCode,
               static_cast<uint16_t>(client_stream->stream_error()));