github.com/percona/percona-xtradb-cluster-operator@v1.14.0/pkg/controller/pxc/status_test.go

package pxc

import (
	"fmt"
	"testing"

	api "github.com/percona/percona-xtradb-cluster-operator/pkg/apis/pxc/v1"
	"github.com/percona/percona-xtradb-cluster-operator/pkg/pxc/app/statefulset"
	"github.com/percona/percona-xtradb-cluster-operator/version"
	"github.com/pkg/errors"

	corev1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/runtime"
	"k8s.io/client-go/kubernetes/scheme"
	"sigs.k8s.io/controller-runtime/pkg/client/fake" // nolint
)

var podStatusReady = corev1.PodStatus{
	ContainerStatuses: []corev1.ContainerStatus{
		{Ready: true},
	},
	Conditions: []corev1.PodCondition{
		{
			Type:   corev1.ContainersReady,
			Status: corev1.ConditionTrue,
		},
	},
}

func newCR(name, namespace string) *api.PerconaXtraDBCluster {
	return &api.PerconaXtraDBCluster{
		ObjectMeta: metav1.ObjectMeta{
			Name:      name,
			Namespace: namespace,
		},
		Spec: api.PerconaXtraDBClusterSpec{
			Platform:  version.PlatformKubernetes,
			CRVersion: "1.6.0",
			PXC: &api.PXCSpec{
				PodSpec: &api.PodSpec{
					Enabled: true,
					Size:    3,
				},
				Expose: api.ServiceExpose{
					Enabled: false,
					Type:    corev1.ServiceTypeClusterIP,
				},
			},
			HAProxy: &api.HAProxySpec{
				PodSpec: api.PodSpec{
					Enabled: true,
					Size:    3,
				},
				ExposePrimary: api.ServiceExpose{
					Enabled: false,
					Type:    corev1.ServiceTypeClusterIP,
				},
				ExposeReplicas: &api.ServiceExpose{
					Enabled: false,
					Type:    corev1.ServiceTypeClusterIP,
				},
			},
			ProxySQL: &api.ProxySQLSpec{
				PodSpec: api.PodSpec{
					Enabled: false,
				},
				Expose: api.ServiceExpose{
					Enabled: false,
					Type:    corev1.ServiceTypeClusterIP,
				},
			},
		},
		Status: api.PerconaXtraDBClusterStatus{},
	}
}

func newMockPod(name, namespace string, labels map[string]string, status corev1.PodStatus) *corev1.Pod {
	return &corev1.Pod{
		ObjectMeta: metav1.ObjectMeta{
			Name:      name,
			Namespace: namespace,
			Labels:    labels,
		},
		Spec:   corev1.PodSpec{},
		Status: status,
	}
}

// creates a fake client to mock API calls with the mock objects
func buildFakeClient(objs []runtime.Object) *ReconcilePerconaXtraDBCluster {
	s := scheme.Scheme

	s.AddKnownTypes(api.SchemeGroupVersion, &api.PerconaXtraDBCluster{})

	cl := fake.NewClientBuilder().
		WithScheme(s).
		WithRuntimeObjects(objs...).
		WithStatusSubresource(&api.PerconaXtraDBCluster{}).
		Build()

	return &ReconcilePerconaXtraDBCluster{client: cl, scheme: s}
}

func TestAppStatusInit(t *testing.T) {
	cr := newCR("cr-mock", "pxc")

	pxc := statefulset.NewNode(cr)
	pxcSfs := pxc.StatefulSet()

	r := buildFakeClient([]runtime.Object{cr, pxcSfs})

	status, err := r.appStatus(pxc, cr.Namespace, cr.Spec.PXC.PodSpec, cr.CompareVersionWith("1.7.0") == -1, false)
	if err != nil {
		t.Error(err)
	}

	if status.Status != api.AppStateInit {
		t.Errorf("AppStatus.Status got %#v, want %#v", status.Status, api.AppStateInit)
	}
}

func TestPXCAppStatusReady(t *testing.T) {
	cr := newCR("cr-mock", "pxc")

	pxc := statefulset.NewNode(cr)
	pxcSfs := pxc.StatefulSet()

	objs := []runtime.Object{cr, pxcSfs}

	for i := 0; i < int(cr.Spec.PXC.Size); i++ {
		objs = append(objs, newMockPod(fmt.Sprintf("pxc-mock-%d", i), cr.Namespace, pxc.Labels(), podStatusReady))
	}

	r := buildFakeClient(objs)

	status, err := r.appStatus(pxc, cr.Namespace, cr.Spec.PXC.PodSpec, cr.CompareVersionWith("1.7.0") == -1, false)
	if err != nil {
		t.Error(err)
	}

	if status.Status != api.AppStateReady {
		t.Errorf("AppStatus.Status got %#v, want %#v", status.Status, api.AppStateReady)
	}

	if status.Ready != cr.Spec.PXC.Size {
		t.Errorf("AppStatus.Ready got %#v, want %#v", status.Ready, cr.Spec.PXC.Size)
	}
}

func TestHAProxyAppStatusReady(t *testing.T) {
	cr := newCR("cr-mock", "pxc")

	haproxy := statefulset.NewHAProxy(cr)
	haproxySfs := haproxy.StatefulSet()

	objs := []runtime.Object{cr, haproxySfs}

	for i := 0; i < int(cr.Spec.HAProxy.Size); i++ {
		objs = append(objs, newMockPod(fmt.Sprintf("haproxy-mock-%d", i), cr.Namespace, haproxy.Labels(), podStatusReady))
	}

	r := buildFakeClient(objs)

	status, err := r.appStatus(haproxy, cr.Namespace, cr.Spec.PXC.PodSpec, cr.CompareVersionWith("1.7.0") == -1, false)
	if err != nil {
		t.Error(err)
	}

	if status.Status != api.AppStateReady {
		t.Errorf("AppStatus.Status got %#v, want %#v", status.Status, api.AppStateReady)
	}

	if status.Ready != cr.Spec.HAProxy.Size {
		t.Errorf("AppStatus.Ready got %#v, want %#v", status.Ready, cr.Spec.HAProxy.Size)
	}
}

func TestUpdateStatusReady(t *testing.T) {
	cr := newCR("cr-mock", "pxc")

	pxc := statefulset.NewNode(cr)
	pxcSfs := pxc.StatefulSet()
	haproxy := statefulset.NewHAProxy(cr)
	haproxySfs := haproxy.StatefulSet()

	objs := []runtime.Object{cr, pxcSfs, haproxySfs}

	for i := 0; i < int(cr.Spec.PXC.Size); i++ {
		objs = append(objs, newMockPod(fmt.Sprintf("pxc-mock-%d", i), cr.Namespace, pxc.Labels(), podStatusReady))
	}

	for i := 0; i < int(cr.Spec.HAProxy.Size); i++ {
		objs = append(objs, newMockPod(fmt.Sprintf("haproxy-mock-%d", i), cr.Namespace, haproxy.Labels(), podStatusReady))
	}

	r := buildFakeClient(objs)

	if err := r.updateStatus(cr, false, nil); err != nil {
		t.Error(err)
	}

	if cr.Status.Status != api.AppStateReady {
		t.Errorf("cr.Status.Status got %#v, want %#v", cr.Status.Status, api.AppStateReady)
	}
}

func TestUpdateStatusError(t *testing.T) {
	cr := newCR("cr-mock", "pxc")

	pxc := statefulset.NewNode(cr)
	pxcSfs := pxc.StatefulSet()

	haproxy := statefulset.NewHAProxy(cr)
	haproxySfs := haproxy.StatefulSet()

	r := buildFakeClient([]runtime.Object{cr, pxcSfs, haproxySfs})

	if err := r.updateStatus(cr, false, errors.New("mock error")); err != nil {
		t.Error(err)
	}

	if cr.Status.Status != api.AppStateError {
		t.Errorf("cr.Status.Status got %#v, want %#v", cr.Status.Status, api.AppStateError)
	}
}

func TestAppHostNoLoadBalancer(t *testing.T) {
	cr := newCR("cr-mock", "pxc")

	pxc := statefulset.NewNode(cr)
	pxcSfs := pxc.StatefulSet()

	haproxy := statefulset.NewHAProxy(cr)
	haproxySfs := haproxy.StatefulSet()

	r := buildFakeClient([]runtime.Object{cr, pxcSfs, haproxySfs})

	host, err := r.appHost(cr, haproxy, &cr.Spec.HAProxy.PodSpec, &cr.Spec.HAProxy.ExposePrimary)
	if err != nil {
		t.Error(err)
	}

	want := haproxy.Service() + "." + cr.Namespace
	if host != want {
		t.Errorf("host got %#v, want %#v", host, want)
	}
}

func TestAppHostLoadBalancerNoSvc(t *testing.T) {
	cr := newCR("cr-mock", "pxc")
	cr.Spec.CRVersion = "1.14.0"

	pxc := statefulset.NewNode(cr)
	pxcSfs := pxc.StatefulSet()

	haproxy := statefulset.NewHAProxy(cr)
	haproxySfs := haproxy.StatefulSet()
	cr.Spec.HAProxy.ExposePrimary.Type = corev1.ServiceTypeLoadBalancer

	r := buildFakeClient([]runtime.Object{cr, pxcSfs, haproxySfs})

	_, err := r.appHost(cr, haproxy, &cr.Spec.HAProxy.PodSpec, &cr.Spec.HAProxy.ExposePrimary)
	if err == nil {
		t.Error("want err, got nil")
	}
}

func TestAppHostLoadBalancerOnlyIP(t *testing.T) {
	cr := newCR("cr-mock", "pxc")
	cr.Spec.CRVersion = "1.14.0"

	pxc := statefulset.NewNode(cr)
	pxcSfs := pxc.StatefulSet()

	haproxy := statefulset.NewHAProxy(cr)
	haproxySfs := haproxy.StatefulSet()
	cr.Spec.HAProxy.ExposePrimary.Type = corev1.ServiceTypeLoadBalancer
	ip := "99.99.99.99"
	haproxySvc := &corev1.Service{
		ObjectMeta: metav1.ObjectMeta{
			Name:      haproxy.Service(),
			Namespace: cr.Namespace,
		},
		Status: corev1.ServiceStatus{
			LoadBalancer: corev1.LoadBalancerStatus{
				Ingress: []corev1.LoadBalancerIngress{{IP: ip}},
			},
		},
	}

	r := buildFakeClient([]runtime.Object{cr, pxcSfs, haproxySfs, haproxySvc})

	host, err := r.appHost(cr, haproxy, &cr.Spec.HAProxy.PodSpec, &cr.Spec.HAProxy.ExposePrimary)
	if err != nil {
		t.Error(err)
	}

	if host != ip {
		t.Errorf("host got %#v, want %#v", host, ip)
	}
}

func TestAppHostLoadBalancerWithHostname(t *testing.T) {
	cr := newCR("cr-mock", "pxc")
	cr.Spec.CRVersion = "1.14.0"

	pxc := statefulset.NewNode(cr)
	pxcSfs := pxc.StatefulSet()

	haproxy := statefulset.NewHAProxy(cr)
	haproxySfs := haproxy.StatefulSet()
	cr.Spec.HAProxy.ExposePrimary.Type = corev1.ServiceTypeLoadBalancer
	wantHost := "cr-mock.haproxy.test"
	haproxySvc := &corev1.Service{
		ObjectMeta: metav1.ObjectMeta{
			Name:      haproxy.Service(),
			Namespace: cr.Namespace,
		},
		Status: corev1.ServiceStatus{
			LoadBalancer: corev1.LoadBalancerStatus{
				Ingress: []corev1.LoadBalancerIngress{{IP: "99.99.99.99", Hostname: wantHost}},
			},
		},
	}

	r := buildFakeClient([]runtime.Object{cr, pxcSfs, haproxySfs, haproxySvc})

	gotHost, err := r.appHost(cr, haproxy, &cr.Spec.HAProxy.PodSpec, &cr.Spec.HAProxy.ExposePrimary)
	if err != nil {
		t.Error(err)
	}

	if gotHost != wantHost {
		t.Errorf("host got %#v, want %#v", gotHost, wantHost)
	}
}

func TestClusterStatus(t *testing.T) {
	tests := map[string]struct {
		status api.PerconaXtraDBClusterStatus
		want   api.AppState
	}{
		"Unknown": {
			status: api.PerconaXtraDBClusterStatus{},
			want:   api.AppStateInit,
		},
		"PXC init": {
			status: api.PerconaXtraDBClusterStatus{PXC: api.AppStatus{ComponentStatus: api.ComponentStatus{Status: api.AppStateInit}}},
			want:   api.AppStateInit,
		},
		"PXC ready without host": {
			status: api.PerconaXtraDBClusterStatus{PXC: api.AppStatus{ComponentStatus: api.ComponentStatus{Status: api.AppStateReady}}},
			want:   api.AppStateInit,
		},
		"PXC ready with host": {
			status: api.PerconaXtraDBClusterStatus{
				PXC: api.AppStatus{
					ComponentStatus: api.ComponentStatus{
						Status: api.AppStateReady,
					},
				},
				Host: "localhost",
			},
			want: api.AppStateReady,
		},
		"PXC stopping": {
			status: api.PerconaXtraDBClusterStatus{PXC: api.AppStatus{ComponentStatus: api.ComponentStatus{Status: api.AppStateStopping}}},
			want:   api.AppStateStopping,
		},
		"PXC paused, HAProxy stopping": {
			status: api.PerconaXtraDBClusterStatus{
				PXC:     api.AppStatus{ComponentStatus: api.ComponentStatus{Status: api.AppStatePaused}},
				HAProxy: api.AppStatus{ComponentStatus: api.ComponentStatus{Status: api.AppStateStopping}},
			},
			want: api.AppStateStopping,
		},
		"PXC paused, ProxySQL stopping": {
			status: api.PerconaXtraDBClusterStatus{
				PXC:      api.AppStatus{ComponentStatus: api.ComponentStatus{Status: api.AppStatePaused}},
				ProxySQL: api.AppStatus{ComponentStatus: api.ComponentStatus{Status: api.AppStateStopping}},
			},
			want: api.AppStateStopping,
		},
		"PXC paused, HAProxy paused": {
			status: api.PerconaXtraDBClusterStatus{
				PXC:     api.AppStatus{ComponentStatus: api.ComponentStatus{Status: api.AppStatePaused}},
				HAProxy: api.AppStatus{ComponentStatus: api.ComponentStatus{Status: api.AppStatePaused}},
			},
			want: api.AppStatePaused,
		},
		"HAProxy init": {
			status: api.PerconaXtraDBClusterStatus{HAProxy: api.AppStatus{ComponentStatus: api.ComponentStatus{Status: api.AppStateInit}}},
			want:   api.AppStateInit,
		},
		"HAProxy ready without host": {
			status: api.PerconaXtraDBClusterStatus{
				PXC:     api.AppStatus{ComponentStatus: api.ComponentStatus{Status: api.AppStateReady}},
				HAProxy: api.AppStatus{ComponentStatus: api.ComponentStatus{Status: api.AppStateReady}},
			},
			want: api.AppStateInit,
		},
		"HAProxy ready with host": {
			status: api.PerconaXtraDBClusterStatus{
				PXC:     api.AppStatus{ComponentStatus: api.ComponentStatus{Status: api.AppStateReady}},
				HAProxy: api.AppStatus{ComponentStatus: api.ComponentStatus{Status: api.AppStateReady}},
				Host:    "localhost",
			},
			want: api.AppStateReady,
		},
		"ProxySQL init": {
			status: api.PerconaXtraDBClusterStatus{ProxySQL: api.AppStatus{ComponentStatus: api.ComponentStatus{Status: api.AppStateInit}}},
			want:   api.AppStateInit,
		},
		"ProxySQL ready without host": {
			status: api.PerconaXtraDBClusterStatus{
				PXC:      api.AppStatus{ComponentStatus: api.ComponentStatus{Status: api.AppStateReady}},
				ProxySQL: api.AppStatus{ComponentStatus: api.ComponentStatus{Status: api.AppStateReady}},
			},
			want: api.AppStateInit,
		},
		"ProxySQL ready with host": {
			status: api.PerconaXtraDBClusterStatus{
				PXC:      api.AppStatus{ComponentStatus: api.ComponentStatus{Status: api.AppStateReady}},
				ProxySQL: api.AppStatus{ComponentStatus: api.ComponentStatus{Status: api.AppStateReady}},
				Host:     "localhost",
			},
			want: api.AppStateReady,
		},
	}

	for name, test := range tests {
		t.Run(name, func(tt *testing.T) {
			got := test.status.ClusterStatus(false, false)

			if got != test.want {
				tt.Errorf("AppState got %#v, want %#v", got, test.want)
			}
		})
	}
}