fix error handling in DeletePVCs

To return an error when we can't get a prime PVC rather than continue to
the next PVC.

Also refactored the code a bit to reduce cognitive complexity.

Signed-off-by: Arik Hadas <ahadas@redhat.com>

pkg/controller/plan/kubevirt.go

@@ -937,11 +937,11 @@ func (r *KubeVirt) DeletePVCs(vm *plan.VMStatus) (err error) {
 	for _, pvc := range pvcs {
 		primePVC := core.PersistentVolumeClaim{}
 		err = r.Destination.Client.Get(context.TODO(), client.ObjectKey{Namespace: r.Plan.Spec.TargetNamespace, Name: fmt.Sprintf("prime-%s", string(pvc.UID))}, &primePVC)
+		if err != nil && k8serr.IsNotFound(err) {
+			err = nil
+		}
 		if err != nil {
-			if k8serr.IsNotFound(err) {
-				err = nil
-			}
-			continue
+			return
 		}
 		// Best effort deletion
 		err = r.DeleteObject(&primePVC, vm, "Deleted prime PVC.", "pvc")

pkg/controller/plan/migration.go

@@ -424,7 +424,9 @@ func (r *Migration) Cancel() (err error) {
 }
 
 func (r *Migration) cleanUpPopulatorPVCs(vm *plan.VMStatus) (err error) {
-	err = r.kubevirt.DeletePVCs(vm)
+	if r.builder.SupportsVolumePopulators() {
+		err = r.kubevirt.DeletePVCs(vm)
+	}
 	return
 }
 
@@ -435,11 +437,9 @@ func (r *Migration) cleanup(vm *plan.VMStatus) (err error) {
 		if err != nil {
 			return
 		}
-		if r.builder.SupportsVolumePopulators() {
-			err = r.cleanUpPopulatorPVCs(vm)
-			if err != nil {
-				return
-			}
+		err = r.cleanUpPopulatorPVCs(vm)
+		if err != nil {
+			return
 		}
 	}
 	err = r.deleteImporterPods(vm)