address review comments

kubernetes-sigs · Jan 24, 2025 · 2f617f5 · 2f617f5
1 parent 5e33401
commit 2f617f5
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/controlplane/kubeadm/internal/control_plane_test.go b/controlplane/kubeadm/internal/control_plane_test.go
@@ -184,14 +184,12 @@ func TestControlPlane(t *testing.T) {
 		g.Expect(fd).To(BeNil())
 	})
 
-	t.Run("ControlPlane returns infra error", func(t *testing.T) {
+	t.Run("ControlPlane returns error when getting infra resources", func(t *testing.T) {
 		g := NewWithT(t)
 		cluster := clusterv1.Cluster{
 			Status: clusterv1.ClusterStatus{
 				FailureDomains: clusterv1.FailureDomains{
-					"one":   failureDomain(true),
-					"two":   failureDomain(true),
-					"three": failureDomain(true),
+					"one": failureDomain(true),
 				},
 			},
 		}
@@ -213,7 +211,7 @@ func TestControlPlane(t *testing.T) {
 		g.Expect(err).To(HaveOccurred())
 	})
 
-	t.Run("When infra and bootstrap config is exists", func(t *testing.T) {
+	t.Run("When infra and bootstrap config exists", func(t *testing.T) {
 		g := NewWithT(t)
 		ns, err := env.CreateNamespace(ctx, "test-machine-watches")
 		kcp := &controlplanev1.KubeadmControlPlane{
@@ -583,8 +581,9 @@ func TestMachineWithDeleteAnnotation(t *testing.T) {
 		}
 
 		g := NewWithT(t)
-		annotedMachines := c.MachineWithDeleteAnnotation(machines)
-		g.Expect(annotedMachines).NotTo(BeNil())
+		annotatedMachines := c.MachineWithDeleteAnnotation(machines)
+		g.Expect(annotatedMachines).NotTo(BeNil())
+		g.Expect(annotatedMachines.Len()).To(BeEquivalentTo(2))
 	})
 }