Microsoft终于在2008年6月26日发布了自己的又一款虚拟化产品Hyper-V。在Windows 2008中同时作为功能之一发布的Powershell则是Microsoft的新一代命令行Shell和脚本语言。有了这两样产品，对于那些做自动化测试的人员来说可谓是两个重要的好消息。这里我主要想介绍一下如何利用Hyper-V和PowerShell自动化构造测试环境。在使用Virtual Server2005的岁月里，测试人员主要是通过编写Javascript脚本访问它的COM接口来控制虚拟机。而到了Hyper-V时代，我们可以通过Hyper-V WMI(Windows Management Instrumentation)来对虚拟机做想要的操作。
试想一下我们有这样一个测试需求：
1.整个测试环境有多台测试机
2.每次在跑自动化测试用例之前都需要安装要测试的产品
3.测试产品不断更新，且每次都必须在干净的环境下安装
为了节省自动化过程的时间，在Hyper-V中，我们可以做多个虚拟机，在做一番基本的配置之后，对每个虚拟机抓一个快照，由此作为起点。下面介绍PowerShell脚本对这些虚拟机的操作，主要做两件事：
1.让虚拟机回到一个干净的状态，也就是我们之前抓得那个快照的状态。
2.启动虚拟机
先介绍两个函数，其中主要是WaitImageToState其功能是等待一台虚拟机变换到指定的状态，比如Hyper-V中虚拟机Running的状态码是2，就等到其状态变为2，当然还得有一个超时的Timeout设置：
function GetImageState
{
param
(
[string]$image = $(throw "param -image is required.")，
[string]$hostServer = $(throw "param -hostserver is required.")
)
$virtualMachines = Get-WmiObject -Class "MSVM_ComputerSystem" -Namespace "root\virtualization" -ComputerName $hostServer;
foreach ($virtualMachine in $virtualMachines)
{
if ($image -ieq $virtualMachine.ElementName)
{
return $virtualMachine.EnabledState;
}
}
throw "Cannot get the state of image [$image] on host server [$hostServer]";
}
function WaitImageToState
{
param
(
[string]$image = $(throw "param -image is required.")，
[string]$hostServer = $(throw "param -hostserver is required.")，
[int]$state = $(throw "param -$state is required.")，
[int]$timeOut = $(throw "param -$timeOut is required.")
)
do
{
$timeOut = $timeOut - 5;
sleep (5);
$currentState = GetImageState -image:$image -hostserver:$hostServer;
if ($currentState -eq $state)
{
return;
}
if ($timeOut -le 0)
{
throw "Wait for image [$image] to state [$state] time out.";
}
}while($true);
}
假设我们快照的名字叫Snapshot_StartPoint，托管机的名字叫ServerHasHyperVInstalled，下面的脚本可以恢复虚拟机到指定快照：
$snapshotName = "Snapshot_StartPoint";
$hostServer = "ServerHasHyperVInstalled";
$virtualMachines = Get-WmiObject -Class "MSVM_ComputerSystem" -Namespace "root\virtualization" -ComputerName $hostServer;
$virtualSystemService = Get-WmiObject -Class "Msvm_VirtualSystemManagementService" -Namespace "root\virtualization" -ComputerName $hostServer;
foreach ($virtualMachine in $virtualMachines)
{
if($virtualMachine.Caption -like "*Virtual Machine*")
{
# get snapshot of the virtual machine
$queryStr = "SELECT * FROM Msvm_VirtualSystemSettingData WHERE SystemName='" + $virtualMachine.Name + "' and SettingType=5";
$snapShots = Get-WmiObject -Query $queryStr -Namespace "root\virtualization" -ComputerName $hostServer;
foreach ($aSnapShot in $snapShots)
{
# revert to specified snapshot
if ($aSnapShot.ElementName -ieq $snapshotName)
{
# apply snapshot
$result = $virtualSystemService.ApplyVirtualSystemSnapShot($virtualMachine.__PATH， $aSnapShot.__PATH);
if ($result.ReturnValue -ne 0)
{
throw "Failed to apply snapshot ["+$snapshotName+"] to VM ["+$image+"]";
}
WaitImageToState -image:$image -hostserver:$hostServer -state:3 -timeOut:120;
Write-Host "Finished applying snapshot [$snapshotName] to VM [$image]";
}
}
}
}
下面的代码是启动这些虚拟机：
$hostServer = "ServerHasHyperVInstalled";
$virtualMachines = Get-WmiObject -Class "MSVM_ComputerSystem" -Namespace "root\virtualization" -ComputerName $hostServer;
foreach ($virtualMachine in $virtualMachines)
{
if($virtualMachine.Caption -like "*Virtual Machine*")
{
$result = $virtualMachine.RequestStateChange(2);
if ($result.ReturnValue -ne 4096)
{
throw "Failed to send start request to VM - " + $virtualMachine;
}
WaitImageToState -image:$virtualMachine -hostserver:$hostServer -state:$virtualMachineRun -timeOut:60;
}
}