Streamlined Save-LabVm function

AutomatedLab · Sep 17, 2018 · 4642b0e · 4642b0e
1 parent 97fd9e5
commit 4642b0e
Showing 1 changed file with 7 additions and 17 deletions.
diff --git a/AutomatedLab/AutomatedLabVirtualMachines.psm1 b/AutomatedLab/AutomatedLabVirtualMachines.psm1
@@ -401,25 +401,15 @@ function Save-LabVM
             Write-ScreenInfo 'There is no machine to start' -Type Warning
             return
         }
-
-        foreach ($vm in $vms)
+
+        Write-Verbose -Message "Saving VMs '$($vms -join ',')"
+        switch ($lab.DefaultVirtualizationEngine)
         {
-            Write-Verbose "Saving VMs '$vm'"
-
-            if ($vm.HostType -eq 'HyperV')
-            {
-                Save-LWHypervVM -ComputerName $vm
-            }
-            elseif ($vm.HostType -eq 'Azure')
-            {
-                Write-Error 'Azure does not support saving machines'
-            }
-            elseif ($vm.HostType -eq 'VMWare')
-            {
-                Save-LWVMWareVM -ComputerName $vm
-            }
+            'HyperV' { Save-LWHypervVM -ComputerName $vms}
+            'VMWare' { Save-LWVMWareVM -ComputerName $vms}
+            'Azure'  { Write-Warning -Message "Skipping Azure VMs '$($vms -join ',')' as suspending the VMs is not supported on Azure."}
         }
-
+                
         Write-LogFunctionExit
     }
 }