DockerCLI/cli/command/stack/kubernetes/services_test.go

package kubernetes

import (
	"testing"

	"github.com/docker/docker/api/types/filters"
	"github.com/docker/docker/api/types/swarm"
	"gotest.tools/assert"
	"gotest.tools/assert/cmp"
)

func TestServiceFiltersLabelSelectorGen(t *testing.T) {
	cases := []struct {
		name                  string
		stackName             string
		filters               filters.Args
		expectedSelectorParts []string
	}{
		{
			name:      "no-filter",
			stackName: "test",
			filters:   filters.NewArgs(),
			expectedSelectorParts: []string{
				"com.docker.stack.namespace=test",
			},
		},
		{
			name:      "label present filter",
			stackName: "test",
			filters: filters.NewArgs(
				filters.KeyValuePair{Key: "label", Value: "label-is-present"},
			),
			expectedSelectorParts: []string{
				"com.docker.stack.namespace=test",
				"label-is-present",
			},
		},
		{
			name:      "single value label filter",
			stackName: "test",
			filters: filters.NewArgs(
				filters.KeyValuePair{Key: "label", Value: "label1=test"},
			),
			expectedSelectorParts: []string{
				"com.docker.stack.namespace=test",
				"label1=test",
			},
		},
		{
			name:      "multi value label filter",
			stackName: "test",
			filters: filters.NewArgs(
				filters.KeyValuePair{Key: "label", Value: "label1=test"},
				filters.KeyValuePair{Key: "label", Value: "label1=test2"},
			),
			expectedSelectorParts: []string{
				"com.docker.stack.namespace=test",
				"label1=test",
				"label1=test2",
			},
		},
		{
			name:      "2 different labels filter",
			stackName: "test",
			filters: filters.NewArgs(
				filters.KeyValuePair{Key: "label", Value: "label1=test"},
				filters.KeyValuePair{Key: "label", Value: "label2=test2"},
			),
			expectedSelectorParts: []string{
				"com.docker.stack.namespace=test",
				"label1=test",
				"label2=test2",
			},
		},
	}

	for _, c := range cases {
		t.Run(c.name, func(t *testing.T) {
			result := generateLabelSelector(c.filters, c.stackName)
			for _, toFind := range c.expectedSelectorParts {
				assert.Assert(t, cmp.Contains(result, toFind))
			}
		})
	}
}
func TestServiceFiltersServiceByName(t *testing.T) {
	cases := []struct {
		name             string
		filters          []string
		services         []swarm.Service
		expectedServices []swarm.Service
	}{
		{
			name:             "no filter",
			filters:          []string{},
			services:         makeServices("s1", "s2"),
			expectedServices: makeServices("s1", "s2"),
		},
		{
			name:             "single-name filter",
			filters:          []string{"s1"},
			services:         makeServices("s1", "s2"),
			expectedServices: makeServices("s1"),
		},
		{
			name:             "filter by prefix",
			filters:          []string{"prefix"},
			services:         makeServices("prefix-s1", "prefix-s2", "s2"),
			expectedServices: makeServices("prefix-s1", "prefix-s2"),
		},
		{
			name:             "multi-name filter",
			filters:          []string{"s1", "s2"},
			services:         makeServices("s1", "s2", "s3"),
			expectedServices: makeServices("s1", "s2"),
		},
		{
			name:             "stack name prefix is valid",
			filters:          []string{"stack_s1"},
			services:         makeServices("s1", "s11", "s2"),
			expectedServices: makeServices("s1", "s11"),
		},
	}
	for _, c := range cases {
		t.Run(c.name, func(t *testing.T) {
			result := filterServicesByName(c.services, c.filters, "stack")
			assert.DeepEqual(t, c.expectedServices, result)
		})
	}
}

func makeServices(names ...string) []swarm.Service {
	result := make([]swarm.Service, len(names))
	for i, n := range names {
		result[i] = swarm.Service{Spec: swarm.ServiceSpec{Annotations: swarm.Annotations{Name: "stack_" + n}}}
	}
	return result
}