add a test to ensure that exactly one control plane machine initializes if there are multiple control plane machines defined

Author: SataQiu

controllers/kubeadmconfig_controller_test.go

@@ -220,7 +220,7 @@ func TestKubeadmConfigReconciler_Reconcile_RequeueJoiningNodesIfControlPlaneNotI
 	workerMachine := newWorkerMachine(cluster)
 	workerJoinConfig := newWorkerJoinKubeadmConfig(workerMachine)
 
-	controlPaneMachine := newControlPlaneMachine(cluster)
+	controlPaneMachine := newControlPlaneMachine(cluster, "control-plane-machine")
 	controlPaneJoinConfig := newControlPlaneJoinKubeadmConfig(controlPaneMachine, "control-plane-join-cfg")
 
 	testcases := []struct {
@@ -286,7 +286,7 @@ func TestKubeadmConfigReconciler_Reconcile_GenerateCloudConfigData(t *testing.T)
 	cluster := newCluster("cluster")
 	cluster.Status.InfrastructureReady = true
 
-	controlPlaneMachine := newControlPlaneMachine(cluster)
+	controlPlaneMachine := newControlPlaneMachine(cluster, "control-plane-machine")
 	controlPlaneInitConfig := newControlPlaneInitKubeadmConfig(controlPlaneMachine, "control-plane-init-cfg")
 
 	objects := []runtime.Object{
@@ -384,7 +384,7 @@ func TestKubeadmConfigReconciler_Reconcile_ErrorIfJoiningControlPlaneHasInvalidC
 	cluster.Status.ControlPlaneInitialized = true
 	cluster.Status.APIEndpoints = []clusterv1.APIEndpoint{{Host: "100.105.150.1", Port: 6443}}
 
-	controlPaneMachine := newControlPlaneMachine(cluster)
+	controlPaneMachine := newControlPlaneMachine(cluster, "control-plane-machine")
 	controlPaneJoinConfig := newControlPlaneJoinKubeadmConfig(controlPaneMachine, "control-plane-join-cfg")
 	controlPaneJoinConfig.Spec.JoinConfiguration.ControlPlane = nil // Makes controlPaneJoinConfig invalid for a control plane machine
 
@@ -463,7 +463,7 @@ func TestReconcileIfJoinNodesAndControlPlaneIsReady(t *testing.T) {
 	workerMachine := newWorkerMachine(cluster)
 	workerJoinConfig := newWorkerJoinKubeadmConfig(workerMachine)
 
-	controlPaneMachine := newControlPlaneMachine(cluster)
+	controlPaneMachine := newControlPlaneMachine(cluster, "control-plane-machine")
 	controlPaneJoinConfig := newControlPlaneJoinKubeadmConfig(controlPaneMachine, "control-plane-join-cfg")
 
 	objects := []runtime.Object{
@@ -970,6 +970,67 @@ func TestKubeadmConfigReconciler_ClusterToKubeadmConfigs(t *testing.T) {
 	}
 }
 
+// Exactly one control plane machine initializes if there are multiple control plane machines defined
+func TestKubeadmConfigReconciler_Reconcile_ExactlyOneControlPlaneMachineInitializes(t *testing.T) {
+	cluster := newCluster("cluster")
+	cluster.Status.InfrastructureReady = true
+
+	controlPlaneInitMachineFirst := newControlPlaneMachine(cluster, "control-plane-init-machine-first")
+	controlPlaneInitConfigFirst := newControlPlaneInitKubeadmConfig(controlPlaneInitMachineFirst, "control-plane-init-cfg-first")
+
+	controlPlaneInitMachineSecond := newControlPlaneMachine(cluster, "control-plane-init-machine-second")
+	controlPlaneInitConfigSecond := newControlPlaneInitKubeadmConfig(controlPlaneInitMachineSecond, "control-plane-init-cfg-second")
+
+	objects := []runtime.Object{
+		cluster,
+		controlPlaneInitMachineFirst,
+		controlPlaneInitConfigFirst,
+		controlPlaneInitMachineSecond,
+		controlPlaneInitConfigSecond,
+	}
+	myclient := fake.NewFakeClientWithScheme(setupScheme(), objects...)
+	k := &KubeadmConfigReconciler{
+		Log:                  log.Log,
+		Client:               myclient,
+		SecretsClientFactory: newFakeSecretFactory(),
+		KubeadmInitLock:      &myInitLocker{},
+	}
+
+	request := ctrl.Request{
+		NamespacedName: types.NamespacedName{
+			Namespace: "default",
+			Name:      "control-plane-init-cfg-first",
+		},
+	}
+	result, err := k.Reconcile(request)
+	if err != nil {
+		t.Fatal(fmt.Sprintf("Failed to reconcile:\n %+v", err))
+	}
+	if result.Requeue == true {
+		t.Fatal("did not expected to requeue")
+	}
+	if result.RequeueAfter != time.Duration(0) {
+		t.Fatal("did not expected to requeue after")
+	}
+
+	request = ctrl.Request{
+		NamespacedName: types.NamespacedName{
+			Namespace: "default",
+			Name:      "control-plane-init-cfg-second",
+		},
+	}
+	result, err = k.Reconcile(request)
+	if err != nil {
+		t.Fatal(fmt.Sprintf("Failed to reconcile:\n %+v", err))
+	}
+	if result.Requeue == true {
+		t.Fatal("did not expected to requeue")
+	}
+	if result.RequeueAfter != 30*time.Second {
+		t.Fatal("expected to requeue after 30s")
+	}
+}
+
 // test utils
 
 // newCluster return a CAPI cluster object
@@ -1018,8 +1079,8 @@ func newWorkerMachine(cluster *clusterv1.Cluster) *clusterv1.Machine {
 	return newMachine(cluster, "worker-machine") // machine by default is a worker node (not the bootstrapNode)
 }
 
-func newControlPlaneMachine(cluster *clusterv1.Cluster) *clusterv1.Machine {
-	m := newMachine(cluster, "control-plane-machine")
+func newControlPlaneMachine(cluster *clusterv1.Cluster, name string) *clusterv1.Machine {
+	m := newMachine(cluster, name)
 	m.Labels[clusterv1.MachineControlPlaneLabelName] = "true"
 	return m
 }
@@ -1106,9 +1167,21 @@ func (f FakeSecretFactory) NewSecretsClient(client client.Client, cluster *clust
 	return f.client, nil
 }
 
-type myInitLocker struct{}
+type myInitLocker struct {
+	locked bool
+}
 
 func (m *myInitLocker) Lock(_ context.Context, _ *clusterv1.Cluster, _ *clusterv1.Machine) bool {
+	if !m.locked {
+		m.locked = true
+		return true
+	}
+	return false
+}
+
+func (m *myInitLocker) Unlock(_ context.Context, _ *clusterv1.Cluster) bool {
+	if m.locked {
+		m.locked = false
+	}
 	return true
 }
-func (m *myInitLocker) Unlock(_ context.Context, _ *clusterv1.Cluster) bool { return true }