e2e test to check labels on Machines created by RKE2ControlPlane

Author: dharmit

test/e2e/data/infrastructure/cluster-template-docker-updated.yaml

@@ -117,11 +117,15 @@ spec:
     kubeAPIServer:
       extraArgs:
       - --anonymous-auth=true
-  infrastructureRef:
-    apiVersion: infrastructure.cluster.x-k8s.io/v1beta1
-    kind: DockerMachineTemplate
-    name:  "${CLUSTER_NAME}-control-plane"
-  nodeDrainTimeout: 30s
+  machineTemplate:
+    metadata:
+      labels:
+        key: value
+    infrastructureRef:
+      apiVersion: infrastructure.cluster.x-k8s.io/v1beta1
+      kind: DockerMachineTemplate
+      name:  "${CLUSTER_NAME}-control-plane"
+    nodeDrainTimeout: 30s
 ---
 apiVersion: infrastructure.cluster.x-k8s.io/v1beta1
 kind: DockerMachineTemplate

test/e2e/data/infrastructure/cluster-template-docker.yaml

@@ -118,11 +118,15 @@ spec:
     kubeAPIServer:
       extraArgs:
       - --anonymous-auth=true
-  infrastructureRef:
-    apiVersion: infrastructure.cluster.x-k8s.io/v1beta1
-    kind: DockerMachineTemplate
-    name:  "${CLUSTER_NAME}-control-plane"
-  nodeDrainTimeout: 30s
+  machineTemplate:
+    metadata:
+      labels:
+        key: value
+    infrastructureRef:
+      apiVersion: infrastructure.cluster.x-k8s.io/v1beta1
+      kind: DockerMachineTemplate
+      name: "${CLUSTER_NAME}-control-plane"
+    nodeDrainTimeout: 30s
 ---
 apiVersion: infrastructure.cluster.x-k8s.io/v1beta1
 kind: DockerMachineTemplate

test/e2e/helpers.go

@@ -369,6 +369,20 @@ func WaitForRKE2ControlPlaneMachinesToExist(ctx context.Context, input WaitForRK
 		}
 		return count, nil
 	}, intervals...).Should(Equal(int(*input.ControlPlane.Spec.Replicas)), "Timed out waiting for %d control plane machines to exist", int(*input.ControlPlane.Spec.Replicas))
+
+	controlPlaneMachineTemplateLabels := input.ControlPlane.Spec.MachineTemplate.ObjectMeta.Labels
+
+	// check if machines owned by RKE2ControlPlane have the labels provided in .spec.machineTemplate.metadata.labels
+	machineList := &clusterv1.MachineList{}
+	err := input.Lister.List(ctx, machineList, inClustersNamespaceListOption, matchClusterListOption)
+	Expect(err).To(BeNil(), "failed to list the machines")
+
+	for _, machine := range machineList.Items {
+		machineLabels := machine.ObjectMeta.Labels
+		for k := range controlPlaneMachineTemplateLabels {
+			Expect(machineLabels[k]).To(Equal(controlPlaneMachineTemplateLabels[k]))
+		}
+	}
 }
 
 // WaitForControlPlaneToBeReadyInput is the input for WaitForControlPlaneToBeReady.