[Compute] Redeploy Dedicated Hosts

src/Compute/Compute.Test/ScenarioTests/DedicatedHostTests.cs

@@ -51,5 +51,12 @@ public void TestDedicatedHostUpdateAndSize()
         {
             TestRunner.RunTestScript("Test-DedicatedHostUpdateAndSize");
         }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void testgenupdateazhost()
+        {
+            TestRunner.RunTestScript("TestGen-updateazhost");
+        }
     }
 }

src/Compute/Compute.Test/ScenarioTests/DedicatedHostTests.ps1

@@ -1,4 +1,4 @@
-# ----------------------------------------------------------------------------------
+# ----------------------------------------------------------------------------------
 #
 # Copyright Microsoft Corporation
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -333,4 +333,45 @@ function Test-DedicatedHostUpdateAndSize
         # Cleanup
         Clean-ResourceGroup $rgname
     }
-}
+}
+
+<#
+.SYNOPSIS
+Test Update-AzHost with Redeploy parameter.
+#>
+function TestGen-updateazhost
+{
+    # Setup
+    $rgname = Get-ComputeTestResourceName
+
+    try
+    {
+        # Common
+        [string]$loc = Get-Location "Microsoft.Resources" "resourceGroups" "East US 2 EUAP";
+        $loc = $loc.Replace(' ', '');
+
+        New-AzResourceGroup -Name $rgname -Location $loc -Force;
+
+        $hostGroupName = $rgname + 'hostgroup';
+        New-AzHostGroup -ResourceGroupName $rgname -Name $hostGroupName -Location $loc -PlatformFaultDomain 1 -Zone "2" -Tag @{key1 = "val1"};
+
+        $hostName = $rgname + 'host';
+        New-AzHost -ResourceGroupName $rgname -HostGroupName $hostGroupName -Name $hostName -Location $loc -Sku "Dadsv5-Type1" -Tag @{key1 = "val2"};
+
+        # Test Update-AzHost with Redeploy parameter
+        $updateHost = Update-AzHost -ResourceGroupName $rgname -HostGroupName $hostGroupName -Name $hostName -Redeploy $true;
+        Assert-AreEqual $true $updateHost.Redeploy;
+
+        # Verify the host is updated
+        $dedicatedHost = Get-AzHost -ResourceGroupName $rgname -HostGroupName $hostGroupName -Name $hostName;
+        Assert-AreEqual $hostName $dedicatedHost.Name;
+        Assert-AreEqual $rgname $dedicatedHost.ResourceGroupName;
+        Assert-AreEqual $loc $dedicatedHost.Location;
+
+    }
+    finally
+    {
+        # Cleanup
+        Clean-ResourceGroup $rgname
+    }
+}

src/Compute/Compute/ChangeLog.md

@@ -20,6 +20,7 @@
 
 -->
 ## Upcoming Release
+* Added `-Redeploy` parameter as an optional boolean for `Update-AzHost` cmdlet to enhance host management capabilities.
 * Made `-PublicIpSku` parameter Standard by default in `New-AzVM`
 
 ## Version 8.5.0

src/Compute/Compute/Generated/DedicatedHost/DedicatedHostUpdateMethod.cs

@@ -78,11 +78,19 @@ public override void ExecuteCmdlet()
                         parameters.Sku = new Sku(this.Sku, null, null);
                     }
 
+                    if (this.IsParameterBound(c => c.Redeploy))
+                    {
+                        var redeployResult = DedicatedHostsClient.Redeploy(resourceGroupName, hostGroupName, Name);
+                        WriteObject(redeployResult);
+                    }
+                    else
+                    {
+                        var result = DedicatedHostsClient.Update(resourceGroupName, hostGroupName, Name, parameters);
 
-                    var result = DedicatedHostsClient.Update(resourceGroupName, hostGroupName, Name, parameters);
-                    var psObject = new PSHost();
-                    ComputeAutomationAutoMapperProfile.Mapper.Map<DedicatedHost, PSHost>(result, psObject);
-                    WriteObject(psObject);
+                        var psObject = new PSHost();
+                        ComputeAutomationAutoMapperProfile.Mapper.Map<DedicatedHost, PSHost>(result, psObject);
+                        WriteObject(psObject);
+                    }
                 }
             });
         }
@@ -123,6 +131,11 @@ public override void ExecuteCmdlet()
             Mandatory = false)]
         public DedicatedHostLicenseTypes LicenseType { get; set; }
 
+        [Parameter(
+            Mandatory = false,
+            HelpMessage = "Redeploy the dedicated host. The operation will complete successfully once the dedicated host has migrated to a new node and is running. To determine the health of VMs deployed on the dedicated host after the redeploy check the Resource Health Center in the Azure Portal. Please refer to https://docs.microsoft.com/azure/service-health/resource-health-overview for more details.")]
+        public bool Redeploy { get; set; }
+
         [Parameter(
             ParameterSetName = "ResourceIdParameter",
             Position = 0,
@@ -131,4 +144,4 @@ public override void ExecuteCmdlet()
         public string ResourceId { get; set; }
 
     }
-}
+}

src/Compute/Compute/Generated/Models/PSHost.cs

@@ -54,6 +54,5 @@ public string ResourceGroupName
         public string Type { get; set; }
         public string Location { get; set; }
         public IDictionary<string, string> Tags { get; set; }
-
     }
-}
+}