Function testStateTransitionSingleAddress

test/clientconn_state_transition_test.go:176–233

(t *testing.T, wantStates []connectivity.State, server func(net.Listener) net.Conn)

}

func testStateTransitionSingleAddress(t *testing.T, wantStates []connectivity.State, server func(net.Listener) net.Conn) {
	pl := testutils.NewPipeListener()
	defer pl.Close()

	// Launch the server.
	var conn net.Conn
	var connMu sync.Mutex
	go func() {
		connMu.Lock()
		conn = server(pl)
		connMu.Unlock()
	}()

	dopts := []grpc.DialOption{
		grpc.WithTransportCredentials(insecure.NewCredentials()),
		grpc.WithDialer(pl.Dialer()),
		grpc.WithConnectParams(grpc.ConnectParams{
			Backoff:           backoff.Config{},
			MinConnectTimeout: 100 * time.Millisecond,
		}),
	}
	cc, err := grpc.NewClient("passthrough:///", dopts...)
	if err != nil {
		t.Fatal(err)
	}
	defer cc.Close()

	// Ensure that the client is in IDLE before connecting.
	ctx, cancel := context.WithTimeout(context.Background(), defaultTestTimeout)
	defer cancel()
	testutils.AwaitState(ctx, t, cc, connectivity.Idle)

	// Subscribe to state updates.
	stateCh := make(chan connectivity.State, 1)
	s := &funcConnectivityStateSubscriber{
		onMsg: func(s connectivity.State) {
			select {
			case stateCh <- s:
			case <-ctx.Done():
			}
		},
	}
	internal.SubscribeToConnectivityStateChanges.(func(cc *grpc.ClientConn, s grpcsync.Subscriber) func())(cc, s)

	cc.Connect()
	for _, wantState := range wantStates {
		waitForState(ctx, t, stateCh, wantState)
	}

	connMu.Lock()
	defer connMu.Unlock()
	if conn != nil {
		err = conn.Close()
		if err != nil {
			t.Fatal(err)
		}
	}
}