From 72e76f6a8673ad1af50c4e123f4b6feaad3ec62f Mon Sep 17 00:00:00 2001
From: "Mark D. Roth" <roth@google.com>
Date: Mon, 22 Aug 2022 14:42:10 -0700
Subject: [PATCH] client_lb_e2e_test: fix flake in
 PickFirstTest.CheckStateBeforeStartWatch (#30698)

---
 test/cpp/end2end/client_lb_end2end_test.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/test/cpp/end2end/client_lb_end2end_test.cc b/test/cpp/end2end/client_lb_end2end_test.cc
index c0a8aafddcc..4ee6f697552 100644
--- a/test/cpp/end2end/client_lb_end2end_test.cc
+++ b/test/cpp/end2end/client_lb_end2end_test.cc
@@ -1255,6 +1255,7 @@ TEST_F(PickFirstTest, CheckStateBeforeStartWatch) {
   WaitForServer(DEBUG_LOCATION, stub_1, 0);
   gpr_log(GPR_INFO, "****** CHANNEL 1 CONNECTED *******");
   servers_[0]->Shutdown();
+  EXPECT_TRUE(WaitForChannelNotReady(channel_1.get()));
   // Channel 1 will receive a re-resolution containing the same server. It will
   // create a new subchannel and hold a ref to it.
   StartServers(1, ports);