+Service Extension Tests

Author: huangpf

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

@@ -231,6 +231,9 @@
     <None Include="SessionRecords\Microsoft.WindowsAzure.Commands.ScenarioTest.ServiceManagementTests\RunNewAzureComputeParameterObjectTests.json">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>
+    <None Include="SessionRecords\Microsoft.WindowsAzure.Commands.ScenarioTest.ServiceManagementTests\RunServiceExtensionSetCmdletTests.json">
+      <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+    </None>
     <None Include="SessionRecords\Microsoft.WindowsAzure.Commands.ScenarioTest.ServiceManagementTests\RunServiceManagementCloudExceptionTests.json">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>

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

@@ -449,3 +449,56 @@ function Run-AutoGeneratedServiceExtensionCmdletTests
         Cleanup-CloudService $svcName;
     }
 }
+
+# Run Service Extension Set Cmdlet Tests
+function Run-ServiceExtensionSetCmdletTests
+{
+    # Setup
+    $location = Get-DefaultLocation;
+    $imgName = Get-DefaultImage $location;
+
+    $storageName = 'pstest' + (getAssetName);
+    New-AzureStorageAccount -StorageAccountName $storageName -Location $location;
+
+    # Associate the new storage account with the current subscription
+    Set-CurrentStorageAccountName $storageName;
+
+    $svcName = 'pstest' + (Get-CloudServiceName);
+    $userName = "pstestuser";
+    $password = "p@ssw0rd";
+    $sPassword = ConvertTo-SecureString $password -AsPlainText -Force;
+    $credential = New-Object System.Management.Automation.PSCredential ($userName, $sPassword);
+
+    # Test
+    New-AzureService -ServiceName $svcName -Location $location;
+
+    try
+    {
+        # New-AzureDeployment (in Azure.psd1)
+        $cspkg = '.\Resources\ServiceManagement\Files\OneWebOneWorker.cspkg';
+        $cscfg = '.\Resources\ServiceManagement\Files\OneWebOneWorker.cscfg';
+
+        # Staging 1st
+        $st = New-AzureDeployment -ServiceName $svcName -Package $cspkg -Configuration $cscfg -Label $svcName -Slot Staging;
+        $st = Set-AzureServiceRemoteDesktopExtension -ServiceName $svcName -Slot Staging -Credential $credential;
+        $ex = Get-AzureServiceExtension -ServiceName $svcName -Slot Staging;
+        $st = Move-AzureDeployment -ServiceName $svcName;
+        $ex = Get-AzureServiceExtension -ServiceName $svcName -Slot Production;
+
+        # Staging 2nd
+        $st = New-AzureDeployment -ServiceName $svcName -Package $cspkg -Configuration $cscfg -Label $svcName -Slot Staging;
+        $st = Set-AzureServiceRemoteDesktopExtension -ServiceName $svcName -Slot Staging -Credential $credential;
+        $ex = Get-AzureServiceExtension -ServiceName $svcName -Slot Staging;
+        $st = Move-AzureDeployment -ServiceName $svcName;
+        $ex = Get-AzureServiceExtension -ServiceName $svcName -Slot Production;
+
+        # Set Extensions
+        $st = Set-AzureServiceRemoteDesktopExtension -ServiceName $svcName -Slot Production -Credential $credential;
+        $st = Set-AzureServiceRemoteDesktopExtension -ServiceName $svcName -Slot Staging -Credential $credential;
+    }
+    finally
+    {
+        # Cleanup
+        Cleanup-CloudService $svcName;
+    }
+}

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

@@ -107,5 +107,14 @@ public void RunAutoGeneratedServiceExtensionCmdletTests()
         {
             this.RunPowerShellTest("Run-AutoGeneratedServiceExtensionCmdletTests");
         }
+
+        [Fact]
+        [Trait(Category.Service, Category.ServiceManagement)]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        [Trait(Category.AcceptanceType, Category.BVT)]
+        public void RunServiceExtensionSetCmdletTests()
+        {
+            this.RunPowerShellTest("Run-ServiceExtensionSetCmdletTests");
+        }
     }
 }