resolver_test.go

package consul_test

import (
	"testing"

	"github.com/golang/mock/gomock"
	"github.com/hashicorp/consul/api"
	"google.golang.org/grpc/resolver"

	consul "github.com/mbobakov/grpc-consul-resolver"
	"github.com/mbobakov/grpc-consul-resolver/internal/mocks"
)

//go:generate mockgen -package mocks -destination internal/mocks/resolverClientConn.go  google.golang.org/grpc/resolver ClientConn

func TestResolver_ChangesHandler(t *testing.T) {
	tests := []struct {
		name      string
		changes   interface{}
		updates   int
		addresses []resolver.Address
	}{
		{
			"empty changes",
			nil,
			0,
			nil,
		},
		{
			"unexpected type",
			[]string{"a", "b", "c"},
			0,
			nil,
		},
		{
			"empty data",
			[]*api.ServiceEntry{},
			1,
			nil,
		},
		{
			"service addresses",
			[]*api.ServiceEntry{
				{
					Service: &api.AgentService{
						Address: "A",
					},
				},
				{
					Service: &api.AgentService{
						Address: "B",
						Port:    123,
					},
				},
			},
			1,
			[]resolver.Address{
				{Addr: "A"},
				{Addr: "B:123"},
			},
		},
		{
			"node addresses",
			[]*api.ServiceEntry{
				{
					Node: &api.Node{
						Address: "A",
					},
				},
				{
					Node: &api.Node{
						Address: "B",
					},
				},
			},
			1,
			[]resolver.Address{
				{Addr: "A"},
				{Addr: "B"},
			},
		},
		{
			"mix",
			[]*api.ServiceEntry{
				{
					Service: &api.AgentService{
						Address: "A",
						Port:    123,
					},
					Node: &api.Node{
						Address: "B",
					},
				},
				{
					Service: &api.AgentService{
						Port: 123,
					},
					Node: &api.Node{
						Address: "D",
					},
				},
			},
			1,
			[]resolver.Address{
				{Addr: "A:123"},
				{Addr: "D:123"},
			},
		},
	}
	ctrl := gomock.NewController(t)
	defer ctrl.Finish()
	t.Parallel()
	for _, test := range tests {
		test := test
		t.Run(test.name, func(t *testing.T) {
			t.Parallel()

			mock := mocks.NewMockClientConn(ctrl)
			mock.EXPECT().UpdateState(resolver.State{Addresses: test.addresses}).Times(test.updates)

			r := consul.NewResolver(mock, nil)
			r.ChangesHandler(1, test.changes)
		})
	}
}