selector_test.go

// Copyright 2022 PingCAP, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// See the License for the specific language governing permissions and
// limitations under the License.

package volumes

import (
	"testing"

	. "github.com/onsi/gomega"

	"github.com/pingcap/tidb-operator/pkg/apis/pingcap/v1alpha1"
)

func TestNewSelector(t *testing.T) {
	cases := []struct {
		desc     string
		instance string
		mt       v1alpha1.MemberType

		expected       string
		expectedHasErr bool
	}{
		{
			desc:     "selector for pd",
			instance: "aaa",
			mt:       v1alpha1.PDMemberType,

			expected: "app.kubernetes.io/component=pd,app.kubernetes.io/instance=aaa,app.kubernetes.io/managed-by=tidb-operator,app.kubernetes.io/name=tidb-cluster",
		},
		{
			desc:     "selector for tso",
			instance: "aaa",
			mt:       v1alpha1.PDMSTSOMemberType,

			expected: "app.kubernetes.io/component=tso,app.kubernetes.io/instance=aaa,app.kubernetes.io/managed-by=tidb-operator,app.kubernetes.io/name=tidb-cluster",
		},
		{
			desc:     "selector for scheduling",
			instance: "aaa",
			mt:       v1alpha1.PDMSSchedulingMemberType,

			expected: "app.kubernetes.io/component=scheduling,app.kubernetes.io/instance=aaa,app.kubernetes.io/managed-by=tidb-operator,app.kubernetes.io/name=tidb-cluster",
		},
		{
			desc:     "selector for router",
			instance: "aaa",
			mt:       v1alpha1.PDMSRouterMerberType,

			expected: "app.kubernetes.io/component=router,app.kubernetes.io/instance=aaa,app.kubernetes.io/managed-by=tidb-operator,app.kubernetes.io/name=tidb-cluster",
		},
		{
			desc:     "selector for tidb",
			instance: "aaa",
			mt:       v1alpha1.TiDBMemberType,

			expected: "app.kubernetes.io/component=tidb,app.kubernetes.io/instance=aaa,app.kubernetes.io/managed-by=tidb-operator,app.kubernetes.io/name=tidb-cluster",
		},
		{
			desc:     "selector for tikv",
			instance: "aaa",
			mt:       v1alpha1.TiKVMemberType,

			expected: "app.kubernetes.io/component=tikv,app.kubernetes.io/instance=aaa,app.kubernetes.io/managed-by=tidb-operator,app.kubernetes.io/name=tidb-cluster",
		},
		{
			desc:     "selector for tiflash",
			instance: "aaa",
			mt:       v1alpha1.TiFlashMemberType,

			expected: "app.kubernetes.io/component=tiflash,app.kubernetes.io/instance=aaa,app.kubernetes.io/managed-by=tidb-operator,app.kubernetes.io/name=tidb-cluster",
		},
		{
			desc:     "selector for ticdc",
			instance: "aaa",
			mt:       v1alpha1.TiCDCMemberType,

			expected: "app.kubernetes.io/component=ticdc,app.kubernetes.io/instance=aaa,app.kubernetes.io/managed-by=tidb-operator,app.kubernetes.io/name=tidb-cluster",
		},
		{
			desc:     "selector for pump",
			instance: "aaa",
			mt:       v1alpha1.PumpMemberType,

			expected: "app.kubernetes.io/component=pump,app.kubernetes.io/instance=aaa,app.kubernetes.io/managed-by=tidb-operator,app.kubernetes.io/name=tidb-cluster",
		},
	}

	sf := MustNewSelectorFactory()

	g := NewGomegaWithT(t)
	for _, c := range cases {
		s, err := sf.NewSelector(c.instance, c.mt)
		if c.expectedHasErr {
			g.Expect(err).Should(HaveOccurred())
		} else {
			g.Expect(err).Should(Succeed(), c.desc)
		}
		g.Expect(s.String()).Should(Equal(c.expected), c.desc)
	}
}