Method TestResolverAddressesToEndpointsUsingNewAddresses

resolver_test.go:150–197

Test ensures one Endpoint is created for each entry in resolver.State.Addresses automatically. The test calls the deprecated NewAddresses API to send a list of addresses to the channel.

(t *testing.T)

// resolver.State.Addresses automatically. The test calls the deprecated
// NewAddresses API to send a list of addresses to the channel.
func (s) TestResolverAddressesToEndpointsUsingNewAddresses(t *testing.T) {
	ctx, cancel := context.WithTimeout(context.Background(), defaultTestTimeout)
	defer cancel()

	const scheme = "testresolveraddressestoendpoints"
	r := manual.NewBuilderWithScheme(scheme)

	stateCh := make(chan balancer.ClientConnState, 1)
	bf := stub.BalancerFuncs{
		UpdateClientConnState: func(_ *stub.BalancerData, ccs balancer.ClientConnState) error {
			stateCh <- ccs
			return nil
		},
	}
	balancerName := "stub-balancer-" + scheme
	stub.Register(balancerName, bf)

	a1 := attributes.New("x", "y")
	a2 := attributes.New("a", "b")
	addrs := []resolver.Address{
		{Addr: "addr1", BalancerAttributes: a1},
		{Addr: "addr2", BalancerAttributes: a2},
	}

	cc, err := NewClient(r.Scheme()+":///",
		WithTransportCredentials(insecure.NewCredentials()),
		WithResolvers(r),
		WithDefaultServiceConfig(fmt.Sprintf(`{"loadBalancingConfig": [{"%s":{}}]}`, balancerName)))
	if err != nil {
		t.Fatalf("NewClient() failed: %v", err)
	}
	cc.Connect()
	defer cc.Close()
	r.CC().NewAddress(addrs)

	select {
	case got := <-stateCh:
		want := []resolver.Endpoint{
			{Addresses: []resolver.Address{{Addr: "addr1"}}, Attributes: a1},
			{Addresses: []resolver.Address{{Addr: "addr2"}}, Attributes: a2},
		}
		if diff := cmp.Diff(got.ResolverState.Endpoints, want); diff != "" {
			t.Errorf("Did not receive expected endpoints.  Diff (-got +want):\n%v", diff)
		}
	case <-ctx.Done():
		t.Fatalf("timed out waiting for endpoints")
	}
}