postgres-operator/pkg/cluster/util_test.go

package cluster

import (
	"context"
	"testing"

	"github.com/stretchr/testify/assert"
	acidv1 "github.com/zalando/postgres-operator/pkg/apis/acid.zalan.do/v1"
	"github.com/zalando/postgres-operator/pkg/util"
	"github.com/zalando/postgres-operator/pkg/util/config"
	"github.com/zalando/postgres-operator/pkg/util/k8sutil"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/client-go/kubernetes/fake"
)

func newFakeK8sAnnotationsClient() (k8sutil.KubernetesClient, *fake.Clientset) {
	clientSet := fake.NewSimpleClientset()

	return k8sutil.KubernetesClient{
		DeploymentsGetter:          clientSet.AppsV1(),
		EndpointsGetter:            clientSet.CoreV1(),
		PodsGetter:                 clientSet.CoreV1(),
		PodDisruptionBudgetsGetter: clientSet.PolicyV1beta1(),
		SecretsGetter:              clientSet.CoreV1(),
		ServicesGetter:             clientSet.CoreV1(),
		StatefulSetsGetter:         clientSet.AppsV1(),
	}, clientSet
}

func TestInheritedAnnotations(t *testing.T) {
	testName := "test inheriting annotations from manifest"
	client, _ := newFakeK8sAnnotationsClient()
	clusterName := "acid-test-cluster"
	namespace := "default"

	annotationKey := "acid"
	pg := acidv1.Postgresql{
		ObjectMeta: metav1.ObjectMeta{
			Name: clusterName,
			Annotations: map[string]string{
				"owned-by": annotationKey,
			},
		},
		Spec: acidv1.PostgresSpec{
			EnableReplicaConnectionPooler: boolToPointer(true),
		},
	}

	var cluster = New(
		Config{
			OpConfig: config.Config{
				Resources: config.Resources{
					ClusterLabels:        map[string]string{"application": "spilo"},
					ClusterNameLabel:     "cluster-name",
					InheritedAnnotations: []string{"owned-by"},
				},
			},
		}, client, pg, logger, eventRecorder)

	cluster.Name = clusterName
	cluster.Namespace = namespace

	// test annotationsSet function
	inheritedAnnotations := cluster.annotationsSet(map[string]string{})

	listOptions := metav1.ListOptions{
		LabelSelector: cluster.labelsSet(false).String(),
	}

	// check pooler deployment annotations
	deployList, err := cluster.KubeClient.Deployments(namespace).List(context.TODO(), listOptions)
	assert.NoError(t, err)
	for _, deploy := range deployList.Items {
		if !(util.MapContains(deploy.ObjectMeta.Annotations, inheritedAnnotations)) {
			t.Errorf("%s: Deployment %v not inherited annotations %#v, got %#v", testName, deploy.ObjectMeta.Name, inheritedAnnotations, deploy.ObjectMeta.Annotations)
		}
	}

	// check statefulset annotations
	stsList, err := cluster.KubeClient.StatefulSets(namespace).List(context.TODO(), listOptions)
	assert.NoError(t, err)
	for _, sts := range stsList.Items {
		if !(util.MapContains(sts.ObjectMeta.Annotations, inheritedAnnotations)) {
			t.Errorf("%s: StatefulSet %v not inherited annotations %#v, got %#v", testName, sts.ObjectMeta.Name, inheritedAnnotations, sts.ObjectMeta.Annotations)
		}
	}

	// check pod annotations
	podList, err := cluster.KubeClient.Pods(namespace).List(context.TODO(), listOptions)
	assert.NoError(t, err)
	for _, pod := range podList.Items {
		if !(util.MapContains(pod.ObjectMeta.Annotations, inheritedAnnotations)) {
			t.Errorf("%s: Pod %v not inherited annotations %#v, got %#v", testName, pod.ObjectMeta.Name, inheritedAnnotations, pod.ObjectMeta.Annotations)
		}
	}

	// check service annotations
	svcList, err := cluster.KubeClient.Services(namespace).List(context.TODO(), listOptions)
	assert.NoError(t, err)
	for _, svc := range svcList.Items {
		if !(util.MapContains(svc.ObjectMeta.Annotations, inheritedAnnotations)) {
			t.Errorf("%s: Service %v not inherited annotations %#v, got %#v", testName, svc.ObjectMeta.Name, inheritedAnnotations, svc.ObjectMeta.Annotations)
		}
	}

	// check endpoint annotations
	epList, err := cluster.KubeClient.Endpoints(namespace).List(context.TODO(), listOptions)
	assert.NoError(t, err)
	for _, ep := range epList.Items {
		if !(util.MapContains(ep.ObjectMeta.Annotations, inheritedAnnotations)) {
			t.Errorf("%s: Endpoint %v not inherited annotations %#v, got %#v", testName, ep.ObjectMeta.Name, inheritedAnnotations, ep.ObjectMeta.Annotations)
		}
	}

	// check pod disruption budget annotations
	pdbList, err := cluster.KubeClient.PodDisruptionBudgets(namespace).List(context.TODO(), listOptions)
	assert.NoError(t, err)
	for _, pdb := range pdbList.Items {
		if !(util.MapContains(pdb.ObjectMeta.Annotations, inheritedAnnotations)) {
			t.Errorf("%s: Pod Disruption Budget %v not inherited annotations %#v, got %#v", testName, pdb.ObjectMeta.Name, inheritedAnnotations, pdb.ObjectMeta.Annotations)
		}
	}

	// check secret annotations
	secretList, err := cluster.KubeClient.Secrets(namespace).List(context.TODO(), listOptions)
	assert.NoError(t, err)
	for _, secret := range secretList.Items {
		if !(util.MapContains(secret.ObjectMeta.Annotations, inheritedAnnotations)) {
			t.Errorf("%s: Secret %v not inherited annotations %#v, got %#v", testName, secret.ObjectMeta.Name, inheritedAnnotations, secret.ObjectMeta.Annotations)
		}
	}
}