[better_conn_close_error] test 2

grpc · May 25, 2021 · f3e0c6b · f3e0c6b
1 parent 6bb168a
commit f3e0c6b
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/internal/transport/transport_test.go b/internal/transport/transport_test.go
@@ -780,8 +780,12 @@ func (s) TestGracefulClose(t *testing.T) {
 		go func() {
 			defer wg.Done()
 			str, err := ct.NewStream(ctx, &CallHdr{})
-			if err == nil {
-				t.Errorf("_.NewStream(_, _) = _, %v, want _, non-nil", err)
+			if err != nil {
+				select {
+				case <-ct.ctxDone:
+				default:
+					t.Errorf("_.NewStream(_, _) = _, %v, while transport is not closed", err)
+				}
 				return
 			}
 			ct.Write(str, nil, nil, &Options{Last: true})