Add envtests and kuttl tests.

controllers/k8ssandra/cassandra_metrics_agent_test.go

@@ -0,0 +1,107 @@
+package k8ssandra
+
+import (
+	"context"
+	"testing"
+
+	cassdcapi "github.com/k8ssandra/cass-operator/apis/cassandra/v1beta1"
+	api "github.com/k8ssandra/k8ssandra-operator/apis/k8ssandra/v1alpha1"
+	"github.com/k8ssandra/k8ssandra-operator/pkg/cassandra"
+	"github.com/k8ssandra/k8ssandra-operator/test/framework"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	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/pointer"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+)
+
+// createSingleDcCluster verifies that the CassandraDatacenter is created and that the
+// expected status updates happen on the K8ssandraCluster.
+func createSingleDcClusterWithMetricsAgent(t *testing.T, ctx context.Context, f *framework.Framework, namespace string) {
+	require := require.New(t)
+
+	kc := &api.K8ssandraCluster{
+		ObjectMeta: metav1.ObjectMeta{
+			Namespace: namespace,
+			Name:      "test",
+		},
+		Spec: api.K8ssandraClusterSpec{
+			Cassandra: &api.CassandraClusterTemplate{
+				Datacenters: []api.CassandraDatacenterTemplate{
+					{
+						Meta: api.EmbeddedObjectMeta{
+							Name: "dc1",
+						},
+						K8sContext: f.DataPlaneContexts[0],
+						Size:       1,
+						DatacenterOptions: api.DatacenterOptions{
+							ServerVersion: "4.0.4",
+							StorageConfig: &cassdcapi.StorageConfig{
+								CassandraDataVolumeClaimSpec: &corev1.PersistentVolumeClaimSpec{
+									StorageClassName: &defaultStorageClass,
+								},
+							},
+							PodSecurityContext: &corev1.PodSecurityContext{
+								RunAsUser: pointer.Int64(999),
+							},
+							ManagementApiAuth: &cassdcapi.ManagementApiAuthConfig{
+								Insecure: &cassdcapi.ManagementApiAuthInsecureConfig{},
+							},
+						},
+					},
+				},
+			},
+		},
+	}
+
+	err := f.Client.Create(ctx, kc)
+	require.NoError(err, "failed to create K8ssandraCluster")
+
+	verifyFinalizerAdded(ctx, t, f, client.ObjectKey{Namespace: kc.Namespace, Name: kc.Name})
+
+	verifySuperuserSecretCreated(ctx, t, f, kc)
+
+	verifyReplicatedSecretReconciled(ctx, t, f, kc)
+
+	verifySystemReplicationAnnotationSet(ctx, t, f, kc)
+
+	t.Log("check that the datacenter was created")
+	dcKey := framework.ClusterKey{NamespacedName: types.NamespacedName{Namespace: namespace, Name: "dc1"}, K8sContext: f.DataPlaneContexts[0]}
+	require.Eventually(f.DatacenterExists(ctx, dcKey), timeout, interval)
+
+	// Check that we have the right volumes and volume mounts.
+	sts := &appsv1.StatefulSet{}
+	if err := f.Client.Get(ctx, types.NamespacedName{Name: "test-dc1-default-sts", Namespace: namespace}, sts); err != nil {
+		assert.Fail(t, "could not find sts")
+	}
+	_, found := cassandra.FindVolume(&sts.Spec.Template, "metrics-agent-config")
+	if !found {
+		assert.Fail(t, "could not find expected metrics-agent-config volume")
+	}
+	cassContainerIdx, _ := cassandra.FindContainer(&sts.Spec.Template, "cassandra")
+	volMount := cassandra.FindVolumeMount(&sts.Spec.Template.Spec.Containers[cassContainerIdx], "metrics-agent-config")
+	if volMount == nil {
+		assert.Fail(t, "could not find expected metrics-agent-config volumeMount")
+	}
+
+	// check that we have the right ConfigMap
+	agentCmKey := types.NamespacedName{Name: "test" + "metrics-agent-config", Namespace: namespace}
+	agentCm := corev1.ConfigMap{}
+	if err := f.Client.Get(ctx, agentCmKey, &agentCm); err != nil {
+		assert.Fail(t, "could not find expected metrics-agent-config configmap")
+	}
+
+	// Test cluster deletion
+	t.Log("deleting K8ssandraCluster")
+	err = f.DeleteK8ssandraCluster(ctx, client.ObjectKey{Namespace: namespace, Name: kc.Name}, timeout, interval)
+	require.NoError(err, "failed to delete K8ssandraCluster")
+	f.AssertObjectDoesNotExist(ctx, t, dcKey, &cassdcapi.CassandraDatacenter{}, timeout, interval)
+	f.AssertObjectDoesNotExist(ctx, t,
+		framework.ClusterKey{K8sContext: f.DataPlaneContexts[1], NamespacedName: agentCmKey},
+		&corev1.ConfigMap{},
+		timeout,
+		interval)
+}

controllers/k8ssandra/k8ssandracluster_controller_test.go

@@ -115,6 +115,7 @@ func TestK8ssandraCluster(t *testing.T) {
 	t.Run("CreateMultiDcDseCluster", testEnv.ControllerTest(ctx, createMultiDcDseCluster))
 	t.Run("PerNodeConfiguration", testEnv.ControllerTest(ctx, perNodeConfiguration))
 	t.Run("CreateSingleDcClusterWithVector", testEnv.ControllerTest(ctx, createSingleDcClusterWithVector))
+	t.Run("createSingleDcClusterWithMetricsAgent", testEnv.ControllerTest(ctx, createSingleDcClusterWithMetricsAgent))
 }
 
 // createSingleDcCluster verifies that the CassandraDatacenter is created and that the

test/kuttl/test-servicemonitors/04-assert.yaml

@@ -0,0 +1,6 @@
+# Test for presence of expected config for cassandra metrics agent
+apiVersion: v1
+kind: ConfigMap
+metadata:
+  name: test-metrics-agent-config
+  namespace: k8ssandra-operator