open-consul/command/services/register/config_test.go

package register

import (
	"testing"

	"github.com/hashicorp/consul/agent/config"
	"github.com/hashicorp/consul/api"
	"github.com/stretchr/testify/require"
)

func TestConfigToAgentService(t *testing.T) {
	cases := []struct {
		Name   string
		Input  *config.ServiceDefinition
		Output *api.AgentServiceRegistration
	}{
		{
			"Basic service with port",
			&config.ServiceDefinition{
				Name: strPtr("web"),
				Tags: []string{"leader"},
				Port: intPtr(1234),
			},
			&api.AgentServiceRegistration{
				Name: "web",
				Tags: []string{"leader"},
				Port: 1234,
			},
		},
		{
			"Service with a check",
			&config.ServiceDefinition{
				Name: strPtr("web"),
				Check: &config.CheckDefinition{
					Name: strPtr("ping"),
				},
			},
			&api.AgentServiceRegistration{
				Name: "web",
				Check: &api.AgentServiceCheck{
					Name: "ping",
				},
			},
		},
		{
			"Service with checks",
			&config.ServiceDefinition{
				Name: strPtr("web"),
				Checks: []config.CheckDefinition{
					config.CheckDefinition{
						Name: strPtr("ping"),
					},
					config.CheckDefinition{
						Name: strPtr("pong"),
					},
				},
			},
			&api.AgentServiceRegistration{
				Name: "web",
				Checks: api.AgentServiceChecks{
					&api.AgentServiceCheck{
						Name: "ping",
					},
					&api.AgentServiceCheck{
						Name: "pong",
					},
				},
			},
		},
	}

	for _, tc := range cases {
		t.Run(tc.Name, func(t *testing.T) {
			require := require.New(t)
			actual, err := configToAgentService(tc.Input)
			require.NoError(err)
			require.Equal(tc.Output, actual)
		})
	}
}

func intPtr(v int) *int       { return &v }
func strPtr(v string) *string { return &v }