solve flaky test

Author: hugehoo

xds/internal/resolver/helpers_test.go

@@ -109,16 +109,31 @@ func buildResolverForTarget(t *testing.T, target resolver.Target, bootstrapConte
 		}
 	}
 
-	stateCh := make(chan resolver.State, 1)
+	stateCh := make(chan resolver.State, 10)
 	updateStateF := func(s resolver.State) error {
-		stateCh <- s
+		select {
+		case stateCh <- s:
+		default:
+			// If channel is full, drop the oldest update to prevent blocking
+			select {
+			case <-stateCh:
+				stateCh <- s
+			default:
+			}
+		}
 		return nil
 	}
-	errCh := make(chan error, 1)
+	errCh := make(chan error, 10)
 	reportErrorF := func(err error) {
 		select {
 		case errCh <- err:
 		default:
+			// If channel is full, drop the oldest error to prevent blocking
+			select {
+			case <-errCh:
+				errCh <- err
+			default:
+			}
 		}
 	}
 	tcc := &testutils.ResolverClientConn{Logger: t, UpdateStateF: updateStateF, ReportErrorF: reportErrorF}
@@ -128,7 +143,20 @@ func buildResolverForTarget(t *testing.T, target resolver.Target, bootstrapConte
 	if err != nil {
 		t.Fatalf("Failed to build xDS resolver for target %q: %v", target, err)
 	}
-	t.Cleanup(r.Close)
+	t.Cleanup(func() {
+		// Drain channels before closing to prevent blocking
+		go func() {
+			for range stateCh {
+			}
+		}()
+		go func() {
+			for range errCh {
+			}
+		}()
+		r.Close()
+		close(stateCh)
+		close(errCh)
+	})
 	return stateCh, errCh, r
 }