pickfirst: cleanup tests (#5263)

grpc · Mar 22, 2022 · 96bdede · 96bdede
1 parent e49486d
commit 96bdede
Show file tree

Hide file tree

Showing 3 changed files with 268 additions and 349 deletions.
diff --git a/balancer_switching_test.go b/balancer_switching_test.go
@@ -32,6 +32,7 @@ import (
 	"google.golang.org/grpc/resolver"
 	"google.golang.org/grpc/resolver/manual"
 	"google.golang.org/grpc/serviceconfig"
+	"google.golang.org/grpc/status"
 )
 
 var _ balancer.Builder = &magicalLB{}
@@ -79,6 +80,13 @@ func startServers(t *testing.T, numServers int, maxStreams uint32) ([]*server, f
 	}
 }
 
+func errorDesc(err error) string {
+	if s, ok := status.FromError(err); ok {
+		return s.Message()
+	}
+	return err.Error()
+}
+
 func checkPickFirst(cc *ClientConn, servers []*server) error {
 	var (
 		req   = "port"

diff --git a/pickfirst_test.go b/pickfirst_test.go