fix unit tests

kubernetes-sigs · Nov 3, 2023 · ba851e4 · ba851e4
1 parent 4b6c4c2
commit ba851e4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/controllers/azurejson_machinepool_controller_test.go b/controllers/azurejson_machinepool_controller_test.go
@@ -137,7 +137,7 @@ func TestAzureJSONPoolReconciler(t *testing.T) {
 				azureMachinePool,
 			},
 			fail: true,
-			err:  "azureclusters.infrastructure.cluster.x-k8s.io \"my-azure-cluster\" not found",
+			err:  "failed to create cluster scope for cluster /my-cluster: azureclusters.infrastructure.cluster.x-k8s.io \"my-azure-cluster\" not found",
 		},
 		"infra ref is nil": {
 			objects: []runtime.Object{
@@ -173,7 +173,8 @@ func TestAzureJSONPoolReconciler(t *testing.T) {
 				machinePool,
 				azureMachinePool,
 			},
-			fail: false,
+			fail: true,
+			err:  "failed to create cluster scope for cluster /my-cluster: unsupported infrastructure type \"FooCluster\", should be AzureCluster or AzureManagedCluster",
 		},
 	}
 

diff --git a/exp/controllers/azuremachinepoolmachine_controller_test.go b/exp/controllers/azuremachinepoolmachine_controller_test.go
@@ -144,8 +144,9 @@ func getAReadyMachinePoolMachineCluster() (*clusterv1.Cluster, *infrav1.AzureClu
 		},
 		Spec: clusterv1.ClusterSpec{
 			InfrastructureRef: &corev1.ObjectReference{
-				Name: azCluster.Name,
-				Kind: "AzureCluster",
+				Name:      azCluster.Name,
+				Namespace: "default",
+				Kind:      "AzureCluster",
 			},
 		},
 		Status: clusterv1.ClusterStatus{