Update machinepool_helpers.go

Author: Cecile Robert-Michon

test/framework/machinepool_helpers.go

@@ -134,13 +134,10 @@ func UpgradeMachinePoolAndWait(ctx context.Context, input UpgradeMachinePoolAndW
 		patchHelper, err := patch.NewHelper(mp, mgmtClient)
 		Expect(err).ToNot(HaveOccurred())
 
+		oldVersion := mp.Spec.Template.Spec.Version
 		mp.Spec.Template.Spec.Version = &input.UpgradeVersion
 		Expect(patchHelper.Patch(ctx, mp)).To(Succeed())
-	}
 
-	for i := range input.MachinePools {
-		mp := input.MachinePools[i]
-		oldVersion := mp.Spec.Template.Spec.Version
 		log.Logf("Waiting for Kubernetes versions of machines in MachinePool %s/%s to be upgraded from %s to %s",
 			mp.Namespace, mp.Name, *oldVersion, input.UpgradeVersion)
 		WaitForMachinePoolInstancesToBeUpgraded(ctx, WaitForMachinePoolInstancesToBeUpgradedInput{
@@ -216,7 +213,7 @@ func WaitForMachinePoolInstancesToBeUpgraded(ctx context.Context, input WaitForM
 			return 0, err
 		}
 		versions := getMachinePoolInstanceVersions(ctx, GetMachinesPoolInstancesInput{
-			Getter:      input.Getter,
+			Getter:      input.WorkloadClusterGetter,
 			Namespace:   input.Cluster.Namespace,
 			MachinePool: input.MachinePool,
 		})