Add Repair-AzureRmVmssServiceFabricUpdateDomain cmdlet

Author: hyonholee

src/ResourceManager/Compute/AzureRM.Compute.psd1

@@ -176,14 +176,16 @@ CmdletsToExport = 'Remove-AzureRmAvailabilitySet', 'Get-AzureRmAvailabilitySet',
                'Disable-AzureRmVmssDiskEncryption', 
                'Get-AzureRmVmssDiskEncryption', 'Get-AzureRmVmssVMDiskEncryption', 
                'Export-AzureRmLogAnalyticRequestRateByInterval', 
-               'Export-AzureRmLogAnalyticThrottledRequests'
+               'Export-AzureRmLogAnalyticThrottledRequests',
+               'Repair-AzureRmVmssServiceFabricUpdateDomain'
 
 # Variables to export from this module
 # VariablesToExport = @()
 
 # Aliases to export from this module, for best performance, do not use wildcards and do not delete the entry, use an empty array if there are no aliases to export.
 AliasesToExport = 'Get-AzureRmVmssDiskEncryptionStatus', 
-               'Get-AzureRmVmssVMDiskEncryptionStatus'
+               'Get-AzureRmVmssVMDiskEncryptionStatus',
+               'Repair-AzureRmVmssServiceFabricUD'
 
 # DSC resources to export from this module
 # DscResourcesToExport = @()

src/ResourceManager/Compute/ChangeLog.md

@@ -18,6 +18,7 @@
         - Additional information about change #1
 -->
 ## Current Release
+* `Repair-AzureRmVmssServiceFabricUpdateDomain` cmdlet is added.
 
 ## Version 4.6.0
 * `Get-AzureRmVmssDiskEncryptionStatus` supports encryption status at data disk level

src/ResourceManager/Compute/Commands.Compute.Test/Common/ComputeTestController.cs

@@ -155,6 +155,7 @@ public void RunPsTestWorkflow(
                     helper.RMStorageModule,
                     helper.GetRMModulePath("AzureRM.Compute.psd1"),
                     helper.GetRMModulePath("AzureRM.Network.psd1"),
+                    helper.GetRMModulePath("AzureRM.ServiceFabric.psd1"),
                     helper.RMStorageDataPlaneModule,
                     "AzureRM.Storage.ps1",
                     "AzureRM.Resources.ps1");

src/ResourceManager/Compute/Commands.Compute.Test/ScenarioTests/VirtualMachineScaleSetTests.cs

@@ -114,5 +114,12 @@ public void TestVirtualMachineScaleSetWriteAcceleratorUpdate()
         {
             ComputeTestController.NewInstance.RunPsTest("Test-VirtualMachineScaleSetWriteAcceleratorUpdate");
         }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestVirtualMachineScaleSetForceUDWalk()
+        {
+            ComputeTestController.NewInstance.RunPsTest("Test-VirtualMachineScaleSetForceUDWalk");
+        }
     }
 }

src/ResourceManager/Compute/Commands.Compute.Test/ScenarioTests/VirtualMachineScaleSetTests.ps1

@@ -1465,3 +1465,128 @@ function Test-VirtualMachineScaleSetWriteAcceleratorUpdate
         Clean-ResourceGroup $rgname
     }
 }
+
+<#
+.SYNOPSIS
+Test Virtual Machine Scale Set Force Service Fabric UD Walk
+#>
+function Test-VirtualMachineScaleSetForceUDWalk
+{
+    # Setup
+    $rgname = "vmsssfudtest"
+
+    try
+    {
+        # Common
+        $loc = 'CentralUS';
+        $adminUsername = 'Foo12';
+        $adminPassword = $PLACEHOLDER;
+        New-AzureRMResourceGroup -Name $rgname -Location $loc -Force;
+        $testMode = Get-ComputeTestMode;
+
+        if ($testMode -eq [Microsoft.Azure.Test.HttpRecorder.HttpRecorderMode]::Playback)
+        {
+            $sf_endpoint = "https://${loc}.servicefabric.azure.com/runtime/clusters/828bd729-f938-41de-8fba-24139669e2d2";
+            $sf_nodename = "nt1vm";
+            $sf_durability = "Bronze";
+        }
+        else
+        {
+            ###
+            ### For record test, first run the following commands.
+            ###
+            $password = ConvertTo-SecureString $adminPassword -AsPlainText -Force;
+            New-AzureRmServiceFabricCluster -ResourceGroupName $rgname -Location $loc -VmPassword $password;
+            $sf_cluster = Get-AzureRmServiceFabricCluster -ResourceGroupName $rgname -Name $rgname;
+            $sf_endpoint = $sf_cluster.ClusterEndpoint;
+            $sf_nodename = $sf_cluster.NodeTypes[0].Name;
+            $sf_durability = $sf_cluster.NodeTypes[0].DurabilityLevel;
+        }
+
+        # NRP
+        $subnet = New-AzureRMVirtualNetworkSubnetConfig -Name ('subnet' + $rgname) -AddressPrefix "10.0.0.0/24";
+        $vnet = New-AzureRMVirtualNetwork -Force -Name ('vnet' + $rgname) -ResourceGroupName $rgname -Location $loc -AddressPrefix "10.0.0.0/16" -Subnet $subnet;
+        $vnet = Get-AzureRMVirtualNetwork -Name ('vnet' + $rgname) -ResourceGroupName $rgname;
+        $subnetId = $vnet.Subnets[0].Id;
+
+        # New VMSS Parameters
+        $vmssName = 'vmss' + $rgname;
+        $imgRef = Get-DefaultCRPImage -loc $loc;
+
+        $extname = 'udwalktest';
+        $publisher = 'Microsoft.Azure.ServiceFabric';
+        $exttype = 'ServiceFabricNode';
+        $extver = '1.0';
+        $settings = @{"clusterEndpoint"=${sf_endpoint};"nodeTypeRef"=${sf_nodename};"durabilityLevel"=${sf_durability};"enableParallelJobs"="true";"dataPath"="D:\\\\SvcFab"};
+
+        $ipCfg = New-AzureRmVmssIPConfig -Name 'test' -SubnetId $subnetId;
+        $vmss = New-AzureRmVmssConfig -Location $loc -SkuCapacity 2 -SkuName 'Standard_A0' -UpgradePolicyMode 'Automatic' `
+            | Add-AzureRmVmssNetworkInterfaceConfiguration -Name 'test' -Primary $true -IPConfiguration $ipCfg `
+            | Set-AzureRmVmssOSProfile -ComputerNamePrefix 'test' -AdminUsername $adminUsername -AdminPassword $adminPassword `
+            | Set-AzureRmVmssStorageProfile -OsDiskCreateOption 'FromImage' -OsDiskCaching 'None' `
+            -ImageReferenceOffer $imgRef.Offer -ImageReferenceSku $imgRef.Skus -ImageReferenceVersion $imgRef.Version `
+            -ImageReferencePublisher $imgRef.PublisherName `
+            | Add-AzureRmVmssExtension -Name $extname -Publisher $publisher -Type $exttype -TypeHandlerVersion $extver -Setting $settings -AutoUpgradeMinorVersion $true;
+
+        $result = New-AzureRmVmss -ResourceGroupName $rgname -Name $vmssName -VirtualMachineScaleSet $vmss;
+
+        Assert-AreEqual $LOC $RESULT.LOCATION;
+        Assert-AreEqual 2 $RESULT.SKU.CAPACITY;
+        Assert-AreEqual 'STANDARD_A0' $RESULT.SKU.NAME;
+        Assert-AreEqual 'Automatic' $result.UpgradePolicy.Mode;
+
+        # Validate Network Profile
+        Assert-AreEqual 'test' $result.VirtualMachineProfile.NetworkProfile.NetworkInterfaceConfigurations[0].Name;
+        Assert-AreEqual $true $result.VirtualMachineProfile.NetworkProfile.NetworkInterfaceConfigurations[0].Primary;
+        Assert-AreEqual $subnetId `
+           $result.VirtualMachineProfile.NetworkProfile.NetworkInterfaceConfigurations[0].IpConfigurations[0].Subnet.Id;
+
+        # Validate OS Profile
+        Assert-AreEqual 'test' $result.VirtualMachineProfile.OsProfile.ComputerNamePrefix;
+        Assert-AreEqual $adminUsername $result.VirtualMachineProfile.OsProfile.AdminUsername;
+        Assert-Null $result.VirtualMachineProfile.OsProfile.AdminPassword;
+
+        # Validate Storage Profile
+        Assert-AreEqual $imgRef.Offer $result.VirtualMachineProfile.StorageProfile.ImageReference.Offer;
+        Assert-AreEqual $imgRef.Skus $result.VirtualMachineProfile.StorageProfile.ImageReference.Sku;
+        Assert-AreEqual $imgRef.Version $result.VirtualMachineProfile.StorageProfile.ImageReference.Version;
+        Assert-AreEqual $imgRef.PublisherName $result.VirtualMachineProfile.StorageProfile.ImageReference.Publisher;
+
+        # Validate Extension Profile
+        Assert-AreEqual $extname $result.VirtualMachineProfile.ExtensionProfile.Extensions[0].Name;
+        Assert-AreEqual $publisher $result.VirtualMachineProfile.ExtensionProfile.Extensions[0].Publisher;
+        Assert-AreEqual $exttype $result.VirtualMachineProfile.ExtensionProfile.Extensions[0].Type;
+        Assert-AreEqual $extver $result.VirtualMachineProfile.ExtensionProfile.Extensions[0].TypeHandlerVersion;
+        Assert-AreEqual $true $result.VirtualMachineProfile.ExtensionProfile.Extensions[0].AutoUpgradeMinorVersion;
+
+        # Verify the result of VMSS
+        $vmss = Get-AzureRmVmss -ResourceGroupName $rgname -VMScaleSetName $vmssName;
+        Assert-AreEqual $null $vmss.Zones;
+        Assert-AreEqual 0 $vmss.Tags.Count;
+        Assert-AreEqual 2 $vmss.Sku.Capacity;
+        Assert-AreEqual $false $vmss.VirtualMachineProfile.StorageProfile.OsDisk.WriteAcceleratorEnabled;
+
+        $vmss = Get-AzureRmVmss -ResourceGroupName $rgname -Name $vmssName;
+        $job = Update-AzureRmVmss -ResourceGroupName $rgname -Name $vmssName -VirtualMachineScaleSet $vmss -SkuName 'Standard_A1' -AsJob;
+
+        Wait-Seconds 300;
+        $vmss_view = Get-AzureRmVmss -ResourceGroupName $rgname -Name $vmssName -InstanceView;
+        $output = $vmss_view | Out-String
+
+        $result = Repair-AzureRmVmssServiceFabricUpdateDomain -ResourceGroupName $rgname -VMScaleSetName $vmssName -PlatformUpdateDomain 0;
+        Assert-True {$result.WalkPerformed};
+        Assert-AreEqual 1 $result.NextPlatformUpdateDomain;
+
+        $result = $vmss | Repair-AzureRmVmssServiceFabricUD -PlatformUpdateDomain 1;
+        Assert-True {$result.WalkPerformed};
+        Assert-AreEqual 2 $result.NextPlatformUpdateDomain;
+
+        $result = Repair-AzureRmVmssServiceFabricUD -ResourceId $vmss.Id -PlatformUpdateDomain 2;
+        Assert-True {$result.WalkPerformed};
+    }
+    finally
+    {
+        # Cleanup
+        Clean-ResourceGroup $rgname
+    }
+}