test: validate CA status ConfigMap

Author: dkoshkin

test/e2e/clusterautoscaler_helpers.go

@@ -14,12 +14,14 @@ import (
 	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
 	appsv1 "k8s.io/api/apps/v1"
+	corev1 "k8s.io/api/core/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/types"
 	"k8s.io/utils/ptr"
 	clusterv1 "sigs.k8s.io/cluster-api/api/v1beta1"
 	addonsv1 "sigs.k8s.io/cluster-api/exp/addons/api/v1beta1"
 	"sigs.k8s.io/cluster-api/test/framework"
+	"sigs.k8s.io/yaml"
 
 	"github.com/nutanix-cloud-native/cluster-api-runtime-extensions-nutanix/api/v1alpha1"
 	"github.com/nutanix-cloud-native/cluster-api-runtime-extensions-nutanix/common/pkg/capi/utils"
@@ -130,4 +132,24 @@ func WaitForClusterAutoscalerToBeReadyForWorkloadCluster(
 			),
 		),
 	)
+
+	statusConfigMap := &corev1.ConfigMap{
+		ObjectMeta: metav1.ObjectMeta{
+			Namespace: "kube-system",
+			Name:      "cluster-autoscaler-status",
+		},
+	}
+
+	WaitForConfigMapData(ctx, WaitForConfigMapDataInput{
+		Getter:    workloadClusterClient,
+		ConfigMap: statusConfigMap,
+		DataValidator: func(data map[string]string) bool {
+			type clusterAutoscalerStatus struct {
+				AutoScalerStatus string `json:"autoscalerStatus,omitempty" yaml:"autoscalerStatus,omitempty"`
+			}
+			status := &clusterAutoscalerStatus{}
+			err = yaml.Unmarshal([]byte(data["status"]), status)
+			return err == nil && status.AutoScalerStatus == "Running"
+		},
+	}, input.DeploymentIntervals...)
 }

test/e2e/configmap_helpers.go

@@ -0,0 +1,62 @@
+//go:build e2e
+
+// Copyright 2024 Nutanix. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+package e2e
+
+import (
+	"context"
+	"fmt"
+	"strings"
+	"time"
+
+	. "github.com/onsi/gomega"
+	corev1 "k8s.io/api/core/v1"
+	"k8s.io/klog/v2"
+	capie2e "sigs.k8s.io/cluster-api/test/e2e"
+	"sigs.k8s.io/cluster-api/test/framework"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+)
+
+type WaitForConfigMapDataInput struct {
+	Getter        framework.Getter
+	ConfigMap     *corev1.ConfigMap
+	DataValidator func(data map[string]string) bool
+}
+
+func WaitForConfigMapData(
+	ctx context.Context,
+	input WaitForConfigMapDataInput,
+	intervals ...interface{},
+) {
+	start := time.Now()
+	key := client.ObjectKeyFromObject(input.ConfigMap)
+	capie2e.Byf("waiting for ConfigMap %s to have expected data", key)
+	Log("starting to wait for ConfigMap to have expected data")
+	Eventually(func() bool {
+		if err := input.Getter.Get(ctx, key, input.ConfigMap); err == nil {
+			return input.DataValidator(input.ConfigMap.Data)
+		}
+		return false
+	}, intervals...).Should(BeTrue(), func() string {
+		return DescribeIncorrectConfigMapData(input, input.ConfigMap)
+	})
+	Logf("ConfigMap %s has expected data, took %v", key, time.Since(start))
+}
+
+// DescribeIncorrectConfigMapData returns detailed output to help debug a ConfigMap data validation failure in e2e.
+func DescribeIncorrectConfigMapData(
+	input WaitForConfigMapDataInput,
+	configMap *corev1.ConfigMap,
+) string {
+	b := strings.Builder{}
+	b.WriteString(fmt.Sprintf("ConfigMap %s failed to get expected data:\n",
+		klog.KObj(input.ConfigMap)))
+	if configMap == nil {
+		b.WriteString("\nConfigMap: nil\n")
+	} else {
+		b.WriteString(fmt.Sprintf("\nConfigMap:\n%s\n", framework.PrettyPrint(configMap)))
+	}
+	return b.String()
+}