Method TestBalancerExitIdle_All

internal/balancergroup/balancergroup_test.go:762–821

(t *testing.T)

}

func (s) TestBalancerExitIdle_All(t *testing.T) {
	balancer1 := t.Name() + "-1"
	balancer2 := t.Name() + "-2"

	testID1, testID2 := testBalancerIDs[0], testBalancerIDs[1]

	exitIdleCh1, exitIdleCh2 := make(chan struct{}, 1), make(chan struct{}, 1)

	stub.Register(balancer1, stub.BalancerFuncs{
		ExitIdle: func(_ *stub.BalancerData) {
			exitIdleCh1 <- struct{}{}
		},
	})

	stub.Register(balancer2, stub.BalancerFuncs{
		ExitIdle: func(_ *stub.BalancerData) {
			exitIdleCh2 <- struct{}{}
		},
	})

	cc := testutils.NewBalancerClientConn(t)
	bg := balancergroup.New(balancergroup.Options{
		CC:              cc,
		BuildOpts:       balancer.BuildOptions{},
		StateAggregator: nil,
		Logger:          nil,
	})
	defer bg.Close()

	bg.Add(testID1, balancer.Get(balancer1))
	bg.Add(testID2, balancer.Get(balancer2))

	bg.ExitIdle()

	errCh := make(chan error, 2)

	go func() {
		select {
		case <-exitIdleCh1:
			errCh <- nil
		case <-time.After(defaultTestTimeout):
			errCh <- fmt.Errorf("timeout waiting for ExitIdle on balancer1")
		}
	}()

	go func() {
		select {
		case <-exitIdleCh2:
			errCh <- nil
		case <-time.After(defaultTestTimeout):
			errCh <- fmt.Errorf("timeout waiting for ExitIdle on balancer2")
		}
	}()

	for i := 0; i < 2; i++ {
		if err := <-errCh; err != nil {
			t.Fatal(err)
		}