Add script-based test for SM

Author: huangpf

src/Common/Commands.ScenarioTest/Commands.ScenarioTest.csproj

@@ -203,6 +203,9 @@
     <None Include="SessionRecords\Microsoft.WindowsAzure.Commands.ScenarioTest.AutomationTests\TestAutomationSuspendAndResumeJob.json">
       <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
     </None>
+    <None Include="SessionRecords\Microsoft.WindowsAzure.Commands.ScenarioTest.AzureVMTests\TestGetAzureLocation.json">
+      <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+    </None>
     <None Include="SessionRecords\Microsoft.WindowsAzure.Commands.ScenarioTest.AzureVMTests\TestGetAzureVM.json">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>
@@ -287,6 +290,7 @@
     <Compile Include="Resources\ResourceLocator.cs" />
     <Compile Include="Scheduler\SchedulerTests.cs" />
     <Compile Include="ServiceBusTests\ServiceBusAuthorizationRuleTests.cs" />
+    <Compile Include="ServiceManagement\GetLocationTests.cs" />
     <Compile Include="ServiceManagement\GetVMTests.cs" />
     <Compile Include="StorageTests\StorageContainerTest.cs" />
     <Compile Include="StorageTests\StorageContextTest.cs" />

src/Common/Commands.ScenarioTest/Resources/ServiceManagement/AzureVMTests.ps1

@@ -12,36 +12,52 @@
 # limitations under the License.
 # ----------------------------------------------------------------------------------
 
-########################################################################### Create-AzureVM Scenario Tests ###########################################################################
-
 <#
 .SYNOPSIS
 Tests Create-AzureVM with valid information.
 #>
 function Test-GetAzureVM
 {
-	# Setup
+    # Setup
 
-	$location = Get-DefaultLocation
-	$imgName = Get-DefaultImage $location
+    $location = Get-DefaultLocation
+    $imgName = Get-DefaultImage $location
 
 
-	$storageName = getAssetName
-	New-AzureStorageAccount -StorageAccountName $storageName -Location $location
+    $storageName = getAssetName
+    New-AzureStorageAccount -StorageAccountName $storageName -Location $location
 
-	Set-CurrentStorageAccountName $storageName
+    Set-CurrentStorageAccountName $storageName
 
-	$vmName = "vm1"
-	$svcName = Get-CloudServiceName
+    $vmName = "vm1"
+    $svcName = Get-CloudServiceName
 
-	# Test
-	New-AzureService -ServiceName $svcName -Location $location
-	New-AzureQuickVM -Windows -ImageName $imgName -Name $vmName -ServiceName $svcName -AdminUsername "pstestuser" -Password "p@ssw0rd"
+    # Test
+    New-AzureService -ServiceName $svcName -Location $location
+    New-AzureQuickVM -Windows -ImageName $imgName -Name $vmName -ServiceName $svcName -AdminUsername "pstestuser" -Password "p@ssw0rd"
 
-	Get-AzureVM -ServiceName $svcName -Name $vmName
+    Get-AzureVM -ServiceName $svcName -Name $vmName
 
 
-	# Cleanup
-	Cleanup-CloudService $svcName
+    # Cleanup
+    Cleanup-CloudService $svcName
 }
 
+
+<#
+.SYNOPSIS
+Test Get-AzureLocation
+#>
+function Test-GetAzureLocation
+{
+    $locations = Get-AzureLocation;
+
+    foreach ($loc in $locations)
+    {
+        $svcName = getAssetName;
+        $st = New-AzureService -ServiceName $svcName -Location $loc.Name;
+        
+        # Cleanup
+        Cleanup-CloudService $svcName
+    }
+}

src/Common/Commands.ScenarioTest/Resources/ServiceManagement/Common.ps1

@@ -20,9 +20,9 @@ Sets the default storage account
 function Set-CurrentStorageAccountName
 {
     param([string] $storageAccountName)
-	$currentSubscription = Get-AzureSubscription -Current
+    $currentSubscription = Get-AzureSubscription -Current
 
-	Set-AzureSubscription -SubscriptionId $currentSubscription.SubscriptionId -CurrentStorageAccountName $storageAccountName
+    Set-AzureSubscription -SubscriptionId $currentSubscription.SubscriptionId -CurrentStorageAccountName $storageAccountName
 }
 
 <#
@@ -31,7 +31,7 @@ Gets the default location
 #>
 function Get-DefaultLocation
 {
-	return (Get-AzureLocation)[0].Name
+    return (Get-AzureLocation)[0].Name
 }
 
 <#
@@ -41,7 +41,7 @@ Gets the default image
 function Get-DefaultImage
 {
     param([string] $loc)
-	return (Get-AzureVMImage | where {$_.OS -eq "Windows"} | where {$_.Location.Contains($loc)})[0].ImageName
+    return (Get-AzureVMImage | where {$_.OS -eq "Windows"} | where {$_.Location.Contains($loc)})[0].ImageName
 }
 
 
@@ -51,7 +51,7 @@ Gets valid and available cloud service name.
 #>
 function Get-CloudServiceName
 {
-	return getAssetName
+    return getAssetName
 }
 
 <#
@@ -61,14 +61,14 @@ Cleanup cloud service
 function Cleanup-CloudService
 {
     param([string] $name)
-	try
-	{
-	     Remove-AzureService -ServiceName $name -Force
-	}
-	catch
-	{
-	     Write-Warning "Cannot Remove the Cloud Service"
-	}
+    try
+    {
+         Remove-AzureService -ServiceName $name -Force
+    }
+    catch
+    {
+         Write-Warning "Cannot Remove the Cloud Service"
+    }
 }
 
 <#
@@ -78,13 +78,13 @@ Cleanup storage
 function Cleanup-Storage
 {
     param([string] $name)
-	Remove-AzureStorageAccount -StorageAccountName $name
-	try
-	{
-	     Remove-AzureStorageAccount -StorageAccountName $name
-	}
-	catch
-	{
-	     Write-Warning "Cannot Remove the Storage Account"
-	}
+    Remove-AzureStorageAccount -StorageAccountName $name
+    try
+    {
+         Remove-AzureStorageAccount -StorageAccountName $name
+    }
+    catch
+    {
+         Write-Warning "Cannot Remove the Storage Account"
+    }
 }

src/Common/Commands.ScenarioTest/ServiceManagement/GetLocationTests.cs

@@ -0,0 +1,31 @@
+// ----------------------------------------------------------------------------------
+//
+// Copyright Microsoft Corporation
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+// http://www.apache.org/licenses/LICENSE-2.0
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+// ----------------------------------------------------------------------------------
+
+using Microsoft.WindowsAzure.Commands.ScenarioTest.Common;
+using Xunit;
+
+namespace Microsoft.WindowsAzure.Commands.ScenarioTest
+{
+    public partial class AzureVMTests
+    {
+        [Fact]
+        [Trait(Category.Service, Category.ServiceManagement)]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        [Trait(Category.AcceptanceType, Category.BVT)]
+        public void TestGetAzureLocation()
+        {
+            this.RunPowerShellTest("Test-GetAzureLocation");
+        }
+    }
+}

src/Common/Commands.ScenarioTest/ServiceManagement/GetVMTests.cs

@@ -21,7 +21,7 @@
 
 namespace Microsoft.WindowsAzure.Commands.ScenarioTest
 {
-    public class AzureVMTests
+    public partial class AzureVMTests
     {
         private EnvironmentSetupHelper helper = new EnvironmentSetupHelper();
 
@@ -52,9 +52,10 @@ protected void RunPowerShellTest(params string[] scripts)
 
                 List<string> modules = Directory.GetFiles("Resources\\ServiceManagement", "*.ps1").ToList();
                 modules.Add("Common.ps1");
+                modules.Add(@"..\..\..\..\Package\Debug\ServiceManagement\Azure\Azure.psd1");
 
                 helper.SetupEnvironment(AzureModule.AzureServiceManagement);
-                helper.SetupModules(AzureModule.AzureServiceManagement, modules.ToArray());
+                helper.SetupModules(modules.ToArray());
 
                 helper.RunPowerShellTest(scripts);
             }