fix: Flaky test ClientSendsAGoAway

grpc · May 10, 2024 · 63ba062 · 63ba062
1 parent 59954c8
commit 63ba062
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/test/goaway_test.go b/test/goaway_test.go
@@ -800,6 +800,7 @@ func (s) TestClientSendsAGoAway(t *testing.T) {
  for {
  f, err := ct.fr.ReadFrame()
  if err != nil {
+ t.Logf("error reading frame: %v", err)
  return
  }
  switch fr := f.(type) {
@@ -808,6 +809,7 @@ func (s) TestClientSendsAGoAway(t *testing.T) {
  if fr.ErrCode == http2.ErrCodeNo {
  t.Logf("GoAway received from client")
  close(goAwayReceived)
+ return
  }
  default:
  t.Errorf("server tester received unexpected frame type %T", f)
@@ -816,6 +818,7 @@ func (s) TestClientSendsAGoAway(t *testing.T) {
  }
  }
  }()
+ cc.WaitForStateChange(ctx, connectivity.Connecting)
  cc.Close()
  defer ct.conn.Close()
  select {