rls: fix flaky test introduced by #6514 (#6535)

easwars · web-flow · commit e2741524dd1b · 2023-08-10T09:03:37.000-07:00
diff --git a/balancer/rls/balancer_test.go b/balancer/rls/balancer_test.go
@@ -919,7 +919,7 @@ func (s) TestUpdateStatePauses(t *testing.T) {
 	}
 	stub.Register(childPolicyName, stub.BalancerFuncs{
 		Init: func(bd *stub.BalancerData) {
-			bd.Data = balancer.Get(grpc.PickFirstBalancerName).Build(ccWrapper, bd.BuildOptions)
+			bd.Data = balancer.Get(grpc.PickFirstBalancerName).Build(bd.ClientConn, bd.BuildOptions)
 		},
 		ParseConfig: func(sc json.RawMessage) (serviceconfig.LoadBalancingConfig, error) {
 			cfg := &childPolicyConfig{}

Original file line number	Diff line number	Diff line change
`@@ -919,7 +919,7 @@ func (s) TestUpdateStatePauses(t *testing.T) {`
`919`	`919`	`}`
`920`	`920`	`stub.Register(childPolicyName, stub.BalancerFuncs{`
`921`	`921`	`Init: func(bd *stub.BalancerData) {`
`922`		`- bd.Data = balancer.Get(grpc.PickFirstBalancerName).Build(ccWrapper, bd.BuildOptions)`
	`922`	`+ bd.Data = balancer.Get(grpc.PickFirstBalancerName).Build(bd.ClientConn, bd.BuildOptions)`
`923`	`923`	`},`
`924`	`924`	`ParseConfig: func(sc json.RawMessage) (serviceconfig.LoadBalancingConfig, error) {`
`925`	`925`	`cfg := &childPolicyConfig{}`