PS Support for Migration AvailaibilitySets APIs #29085

srcharug · 2026-01-20T19:45:55Z

Description

Added Migration and Convert APIs for AvailabilitySets. Added UTs to test the same.
Migrate AvailabilitySet -https://learn.microsoft.com/en-us/cli/azure/vm/availability-set?view=azure-cli-latest#az-vm-availability-set-start-migration-to-vmss
Convert AvailabilitySet - https://learn.microsoft.com/en-us/cli/azure/vm/availability-set?view=azure-cli-latest#az-vm-availability-set-convert-to-vmss
VM migration - https://learn.microsoft.com/en-us/cli/azure/vm?view=azure-cli-latest#az-vm-migrate-to-vmss

Mandatory Checklist

Please choose the target release of Azure PowerShell. (⚠️Target release is a different concept from API readiness. Please click below links for details.)
- General release
- Public preview
- Private preview
- Engineering build
- No need for a release
Check this box to confirm: I have read the Submitting Changes section of CONTRIBUTING.md and reviewed the following information:

SHOULD update ChangeLog.md file(s) appropriately
- Update src/{{SERVICE}}/{{SERVICE}}/ChangeLog.md.
  - A snippet outlining the change(s) made in the PR should be written under the ## Upcoming Release header in the past tense.
- Should not change ChangeLog.md if no new release is required, such as fixing test case only.
SHOULD regenerate markdown help files if there is cmdlet API change. Instruction
SHOULD have proper test coverage for changes in pull request.
SHOULD NOT adjust version of module manually in pull request

This agent will create a Powershell PR based on the issue described. It will write tests, make code changes and add changelog.

Update agent name and description for Powershell PR

Agent instructions

Add Compute PR agent instructions

Co-authored-by: haagha <64601174+haagha@users.noreply.github.com>

azure-client-tools-bot-prd · 2026-01-20T19:46:00Z

Thanks for your contribution! The pull request validation has started. Please revisit this comment for updated status.

Copilot

Pull request overview

This pull request adds five new cmdlets to enable migration of Availability Sets to Flexible Virtual Machine Scale Sets (VMSS Flex) in Azure PowerShell Compute module. This feature is in public preview and requires the subscription to be enabled for the feature flag Microsoft.Compute/MigrateToVmssFlex.

Changes:

Added five new cmdlets: Test-AzAvailabilitySetMigration, Start-AzAvailabilitySetMigration, Stop-AzAvailabilitySetMigration, Convert-AzAvailabilitySet, and Move-AzVirtualMachineToVmss
Updated module manifest to export the new cmdlets
Added PowerShell and C# test scenarios
Updated ChangeLog.md with feature description

Reviewed changes

Copilot reviewed 15 out of 17 changed files in this pull request and generated 6 comments.

Show a summary per file

File	Description
Test-AzAvailabilitySetMigration.md	Help documentation for validating Availability Set migration to VMSS
Stop-AzAvailabilitySetMigration.md	Help documentation for canceling migration operations
Start-AzAvailabilitySetMigration.md	Help documentation for starting Availability Set migration
Move-AzVirtualMachineToVmss.md	Help documentation for migrating individual VMs to VMSS
Convert-AzAvailabilitySet.md	Help documentation for converting Availability Set to new VMSS
MoveAzureVMToVmssCommand.cs	C# implementation of Move-AzVirtualMachineToVmss cmdlet
ChangeLog.md	Updated with descriptions of new cmdlets
Az.Compute.psd1	Module manifest updated to export new cmdlets
TestAzureAvailabilitySetMigrationCommand.cs	C# implementation of Test-AzAvailabilitySetMigration cmdlet
StopAzureAvailabilitySetMigrationCommand.cs	C# implementation of Stop-AzAvailabilitySetMigration cmdlet
StartAzureAvailabilitySetMigrationCommand.cs	C# implementation of Start-AzAvailabilitySetMigration cmdlet
ConvertAzureAvailabilitySetCommand.cs	C# implementation of Convert-AzAvailabilitySet cmdlet
AvailabilitySetTests.ps1	PowerShell test scenarios for migration cmdlets
AvailabilitySetTests.cs	C# test runner for migration tests
computePR.agent.md	New agent documentation file for Compute PR workflows

Copilot · 2026-01-20T20:01:18Z

src/Compute/Compute.Test/ScenarioTests/AvailabilitySetTests.ps1

+    Assert-AreEqual $p.StorageProfile.ImageReference.Offer $imgRef.Offer;
+    Assert-AreEqual $p.StorageProfile.ImageReference.Publisher $imgRef.PublisherName;
+    Assert-AreEqual $p.StorageProfile.ImageReference.Sku $imgRef.Skus;
+    Assert-AreEqual $p.StorageProfile.ImageReference.Version $imgRef.Version;


The variables $p and $imgRef are not defined in this function. These assertions reference undefined variables and will fail when the test runs. The assertions on lines 392-395 should either be removed if they're not needed, or the code should be updated to properly capture the VM configuration after setting the source image.

Suggested change

Assert-AreEqual $p.StorageProfile.ImageReference.Offer $imgRef.Offer;

Assert-AreEqual $p.StorageProfile.ImageReference.Publisher $imgRef.PublisherName;

Assert-AreEqual $p.StorageProfile.ImageReference.Sku $imgRef.Skus;

Assert-AreEqual $p.StorageProfile.ImageReference.Version $imgRef.Version;

Copilot · 2026-01-20T20:01:18Z

src/Compute/Compute.Test/ScenarioTests/AvailabilitySetTests.ps1

+    #$vmConfig = ($imgRef | Set-AzVMSourceImage -VM $vmConfig);
+
+    Assert-AreEqual $p.StorageProfile.ImageReference.Offer $imgRef.Offer;
+    Assert-AreEqual $p.StorageProfile.ImageReference.Publisher $imgRef.PublisherName;
+    Assert-AreEqual $p.StorageProfile.ImageReference.Sku $imgRef.Skus;
+    Assert-AreEqual $p.StorageProfile.ImageReference.Version $imgRef.Version;
+


This commented-out line and the assertion code below it (lines 392-395) suggest incomplete or leftover code from development. The commented line should either be removed or properly integrated into the test logic.

Suggested change

#$vmConfig = ($imgRef | Set-AzVMSourceImage -VM $vmConfig);

Assert-AreEqual $p.StorageProfile.ImageReference.Offer $imgRef.Offer;

Assert-AreEqual $p.StorageProfile.ImageReference.Publisher $imgRef.PublisherName;

Assert-AreEqual $p.StorageProfile.ImageReference.Sku $imgRef.Skus;

Assert-AreEqual $p.StorageProfile.ImageReference.Version $imgRef.Version;

Copilot · 2026-01-20T20:01:18Z

src/Compute/Compute.Test/ScenarioTests/AvailabilitySetTests.ps1

+       # $newVmssName = 'vmss2' + $rgname;
+       # $convertResult = Convert-AzAvailabilitySet -ResourceGroupName $rgname -Name $asetName -VirtualMachineScaleSetName $newVmssName;
+       # Assert-NotNull $convertResult;


The test for Convert-AzAvailabilitySet cmdlet is commented out (lines 289-291). This appears to be a significant piece of test coverage that should be enabled to properly validate the Convert-AzAvailabilitySet cmdlet functionality.

Copilot · 2026-01-20T20:01:19Z

src/Compute/Compute.Test/ScenarioTests/AvailabilitySetTests.ps1

+function Test-AvailabilitySetMigration
+{
+    param ($loc)
+    # Setup
+    $rgname = Get-ComputeTestResourceName
+
+    try
+    {
+        # Common
+        if ($loc -eq $null)
+        {
+            $loc = Get-ComputeVMLocation;
+        }
+        New-AzResourceGroup -Name $rgname -Location $loc -Force;
+
+        # Create Availability Set
+        $asetName = 'aset' + $rgname;
+        New-AzAvailabilitySet -ResourceGroupName $rgname -Name $asetName -Location $loc -Sku 'Aligned' -PlatformFaultDomainCount 2 -PlatformUpdateDomainCount 7;
+        $aset = Get-AzAvailabilitySet -ResourceGroupName $rgname -Name $asetName;
+        Assert-NotNull $aset;
+
+        # Create a VM in the Availability Set
+        $vmname = 'vm' + $rgname;
+        $vm = New-TestVmInAvailabilitySet -ResourceGroupName $rgname -Location $loc -AvailabilitySetId $aset.Id -VmName $vmname;
+        $a = $vm | Out-String;
+        Write-Verbose("Get-AzVM output:");
+        Write-Verbose($a);
+        Assert-NotNull $a
+
+        # Create a Flexible VMSS for migration target
+        $vmssName = 'vmss' + $rgname;
+        $vmssConfig = New-AzVmssConfig -Location $loc -OrchestrationMode 'Flexible' -PlatformFaultDomainCount 2 -SinglePlacementGroup $false;
+        $vmss = New-AzVmss -ResourceGroupName $rgname -VMScaleSetName $vmssName -VirtualMachineScaleSet $vmssConfig;
+        Assert-NotNull $vmss;
+        $vmssId = $vmss.Id;
+
+        # Test Validate Migration cmdlet
+        $validateResult = Test-AzAvailabilitySetMigration -ResourceGroupName $rgname -Name $asetName -VirtualMachineScaleSetFlexibleId $vmssId;
+        Assert-NotNull $validateResult;
+
+        # Test StartMigration cmdlet
+        $migrationResult = Start-AzAvailabilitySetMigration -ResourceGroupName $rgname -Name $asetName -VirtualMachineScaleSetFlexibleId $vmssId;
+        Assert-NotNull $migrationResult;
+
+        # Migrate VM to VMSS Flex
+        $migratedVM = Move-AzVirtualMachineToVmss -Id $vm.Id
+        Assert-NotNull $migratedVM;
+
+        # Test Convert cmdlet (creates a new VMSS)
+       # $newVmssName = 'vmss2' + $rgname;
+       # $convertResult = Convert-AzAvailabilitySet -ResourceGroupName $rgname -Name $asetName -VirtualMachineScaleSetName $newVmssName;
+       # Assert-NotNull $convertResult;
+
+        Write-Host "Availability Set Migration cmdlets test completed successfully";
+    }
+    finally
+    {
+        # Cleanup
+        Clean-ResourceGroup $rgname
+    }
+}


The test function Test-AvailabilitySetMigration does not include test coverage for the Stop-AzAvailabilitySetMigration cmdlet. This cmdlet is part of the migration workflow and should be tested to ensure it can properly cancel migration operations.

Copilot · 2026-01-20T20:01:19Z

src/Compute/Compute.Test/ScenarioTests/AvailabilitySetTests.ps1

+<#
+.SYNOPSIS
+Test Availability Set Convert to VMSS Flex
+Note: This test requires the subscription to be enabled for the feature flag Microsoft.Compute/ConvertToVmssFlex


The comment mentions the feature flag "Microsoft.Compute/ConvertToVmssFlex", but based on the PR description and other test comments, the feature flag should be "Microsoft.Compute/MigrateToVmssFlex". This inconsistency could cause confusion about what feature flag is required for testing.

Suggested change

Note: This test requires the subscription to be enabled for the feature flag Microsoft.Compute/ConvertToVmssFlex

Note: This test requires the subscription to be enabled for the feature flag Microsoft.Compute/MigrateToVmssFlex

Copilot · 2026-01-20T20:01:19Z

src/Compute/Compute.Test/ScenarioTests/AvailabilitySetTests.ps1

+        $convertResult = Convert-AzAvailabilitySet -ResourceGroupName $rgname -Name $asetName -VirtualMachineScaleSetName $newVmssName;
+        Assert-NotNull $convertResult;
+
+        Write-Host "Availability Set Migration cmdlets test completed successfully";


The success message says "Availability Set Migration cmdlets test completed successfully" but this is the Test-AvailabilitySetConvert function, which tests the Convert cmdlet specifically. The message should be "Availability Set Convert cmdlet test completed successfully" to accurately reflect what was tested.

Suggested change

Write-Host "Availability Set Migration cmdlets test completed successfully";

Write-Host "Availability Set Convert cmdlet test completed successfully";

isra-fel · 2026-01-20T20:08:02Z

/azp run

azure-pipelines · 2026-01-20T20:08:19Z

Azure Pipelines successfully started running 3 pipeline(s).

src/Compute/Compute.Test/ScenarioTests/AvailabilitySetTests.cs

vidai-msft · 2026-01-22T13:03:17Z

src/Compute/Compute/ChangeLog.md

 * Updated Azure.Core from 1.45.0 to 1.47.3
 * Added `-EnableAutomaticUpgrade` and `-TreatFailureAsDeploymentFailure` parameters (Bool) to `New-AzVmGalleryApplication` and `New-AzVmssGalleryApplication` cmdlets.
 * Added `-EnableAutomaticUpgrade` and `-TreatFailureAsDeploymentFailure` parameters (Switch) to `Add-AzVmGalleryApplication` and `Add-AzVmssGalleryApplication` cmdlets.
+* Added new cmdlets for Availability Set to Virtual Machine Scale Set (VMSS) Flex migration (Public Preview)


Please sync your local branch with the latest main. The position of the change log entries is incorrect.

github-actions · 2026-01-22T13:07:04Z

This PR was labeled "needs-revision" because it has unresolved review comments or CI failures.
Please resolve all open review comments and make sure all CI checks are green. Refer to our guide to troubleshoot common CI failures.

Copilot

Pull request overview

Copilot reviewed 14 out of 16 changed files in this pull request and generated 5 comments.

Copilot · 2026-01-23T00:33:52Z

src/Compute/Compute.Test/ScenarioTests/AvailabilitySetTests.ps1

+    $user = "User$($VmName.Substring(0,4))";
+    $password = $PLACEHOLDER;
+    $securePassword = ConvertTo-SecureString $password -AsPlainText -Force;
+    $cred = New-Object System.Management.Automation.PSCredential ($user, $securePassword);
+    $vmConfig = Set-AzVMOperatingSystem -VM $vmConfig -Windows -ComputerName $VmName -Credential $cred;
+
+    $vmConfig = Set-AzVMSourceImage -VM $vmConfig -publisherName "MicrosoftWindowsServer" -offer "WindowsServer" -skus "2022-datacenter-g2" -version "latest";  
+    #$vmConfig = ($imgRef | Set-AzVMSourceImage -VM $vmConfig);
+
+    Assert-AreEqual $p.StorageProfile.ImageReference.Offer $imgRef.Offer;
+    Assert-AreEqual $p.StorageProfile.ImageReference.Publisher $imgRef.PublisherName;
+    Assert-AreEqual $p.StorageProfile.ImageReference.Sku $imgRef.Skus;
+    Assert-AreEqual $p.StorageProfile.ImageReference.Version $imgRef.Version;


$p and $imgRef are referenced but never defined in this function (and $imgRef is explicitly commented out), which will cause the scenario tests to fail at runtime. Remove these assertions or rewrite them to assert against $vmConfig.StorageProfile.ImageReference (or define $imgRef and set $p appropriately). Also ensure $PLACEHOLDER is defined/initialized in-scope for the test run (or replaced with the standard test credential source used in this repo).

Copilot · 2026-01-23T00:33:53Z

src/Compute/Compute.Test/ScenarioTests/AvailabilitySetTests.cs

+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestAvailabilitySetMigration()
+        {
+            TestRunner.RunTestScript("Test-AvailabilitySetMigration 'eastus2euap'");
+        }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestAvailabilitySetConvert()
+        {
+            TestRunner.RunTestScript("Test-AvailabilitySetConvert 'eastus2euap'");
+        }


These tests are marked as Category.CheckIn but the scripts explicitly require a subscription feature flag to be enabled; this is likely to make check-in CI flaky/fail in environments where the flag isn’t enabled. Consider categorizing them as LiveOnly (or the repo’s equivalent non-check-in category), and/or add a deterministic skip path in the script when the required feature flag isn’t available.

Copilot · 2026-01-23T00:33:53Z

src/Compute/Compute/VirtualMachine/Action/MoveAzureVMToVmssCommand.cs

+        {
+            base.ExecuteCmdlet();
+
+            if (ParameterSetName.Equals(IdParameterSet) && !string.IsNullOrEmpty(Id))


Use an explicit StringComparison when comparing ParameterSetName (e.g., ordinal/ignore-case) to avoid culture-sensitive comparisons and make the intent clear.

Suggested change

if (ParameterSetName.Equals(IdParameterSet) && !string.IsNullOrEmpty(Id))

if (ParameterSetName.Equals(IdParameterSet, StringComparison.OrdinalIgnoreCase) && !string.IsNullOrEmpty(Id))

Copilot · 2026-01-23T00:33:53Z

src/Compute/Compute/AvailabilitySets/StartAzureAvailabilitySetMigrationCommand.cs

+            Mandatory = true,
+            Position = 2,
+            ValueFromPipelineByPropertyName = true,
+            HelpMessage = "The resource ID of the Flexible Virtual Machine Scale Set to migrate to.")]


Consider adding a ResourceIdCompleter for VirtualMachineScaleSetFlexibleId (e.g., Microsoft.Compute/virtualMachineScaleSets) to match other ID-typed parameters in this module and improve interactive usability.

Suggested change

HelpMessage = "The resource ID of the Flexible Virtual Machine Scale Set to migrate to.")]

HelpMessage = "The resource ID of the Flexible Virtual Machine Scale Set to migrate to.")]

[ResourceIdCompleter("Microsoft.Compute/virtualMachineScaleSets")]

Copilot · 2026-01-23T00:33:54Z

src/Compute/Compute/AvailabilitySets/TestAzureAvailabilitySetMigrationCommand.cs

+            Mandatory = true,
+            Position = 2,
+            ValueFromPipelineByPropertyName = true,
+            HelpMessage = "The resource ID of the Flexible Virtual Machine Scale Set to validate migration to.")]


Consider adding a ResourceIdCompleter for VirtualMachineScaleSetFlexibleId (e.g., Microsoft.Compute/virtualMachineScaleSets) for consistency with other cmdlets that accept resource IDs and to reduce input errors.

Suggested change

HelpMessage = "The resource ID of the Flexible Virtual Machine Scale Set to validate migration to.")]

HelpMessage = "The resource ID of the Flexible Virtual Machine Scale Set to validate migration to.")]

[ResourceIdCompleter("Microsoft.Compute/virtualMachineScaleSets")]

isra-fel · 2026-01-23T00:38:25Z

/azp run

azure-pipelines · 2026-01-23T00:38:40Z

Azure Pipelines successfully started running 3 pipeline(s).

haagha and others added 13 commits November 12, 2025 15:30

Update agent name and description for Powershell PR

56bf91e

This agent will create a Powershell PR based on the issue described. It will write tests, make code changes and add changelog.

Merge pull request Azure#210 from PSCmdAssistant/haagha-patch-1

6916d82

Update agent name and description for Powershell PR

Merge branch 'Azure:main' into main

8b7b761

Merge branch 'Azure:main' into main

21a28ca

Add agent instructions

c866aa3

update name

0815f74

Merge pull request Azure#219 from PSCmdAssistant/agent-instructions

2595424

Agent instructions

Fix agent instruction formatting

c5f2be4

Merge pull request Azure#220 from PSCmdAssistant/agent-instructions

095010d

Add Compute PR agent instructions

Initial plan

e96e305

Add Availability Set to VMSS migration cmdlets

5656aef

Co-authored-by: haagha <64601174+haagha@users.noreply.github.com>

Add commands to get identified

a2df751

Add UTs

106253c

Copilot AI review requested due to automatic review settings January 20, 2026 19:45

srcharug requested review from DanielMicrosoft, NoriZC, Pan-Qi, VeryEarly, YanaXu, audreyttt, dolauli, haagha, isra-fel, notyashhh, ookoka, vidai-msft and wyunchi-ms as code owners January 20, 2026 19:45

Copilot started reviewing on behalf of srcharug January 20, 2026 19:53 View session

Copilot AI reviewed Jan 20, 2026

View reviewed changes

vidai-msft requested changes Jan 22, 2026

View reviewed changes

vidai-msft added the needs-revision label Jan 22, 2026

haagha previously approved these changes Jan 22, 2026

View reviewed changes

resolve comments

52ff159

srcharug dismissed haagha’s stale review via 52ff159 January 23, 2026 00:20

Merge remote-tracking branch 'origin/main' into migrationPS

11f7dfe

Copilot AI review requested due to automatic review settings January 23, 2026 00:22

Copilot started reviewing on behalf of srcharug January 23, 2026 00:24 View session

Copilot AI reviewed Jan 23, 2026

View reviewed changes

srcharug changed the title ~~Migration ps~~ PS Support for Migration AvailaibilitySets APIs Jan 23, 2026

	Note: This test requires the subscription to be enabled for the feature flag Microsoft.Compute/ConvertToVmssFlex
	Note: This test requires the subscription to be enabled for the feature flag Microsoft.Compute/MigrateToVmssFlex

	Write-Host "Availability Set Migration cmdlets test completed successfully";
	Write-Host "Availability Set Convert cmdlet test completed successfully";

	if (ParameterSetName.Equals(IdParameterSet) && !string.IsNullOrEmpty(Id))
	if (ParameterSetName.Equals(IdParameterSet, StringComparison.OrdinalIgnoreCase) && !string.IsNullOrEmpty(Id))

	HelpMessage = "The resource ID of the Flexible Virtual Machine Scale Set to migrate to.")]
	HelpMessage = "The resource ID of the Flexible Virtual Machine Scale Set to migrate to.")]
	[ResourceIdCompleter("Microsoft.Compute/virtualMachineScaleSets")]

PS Support for Migration AvailaibilitySets APIs #29085

Are you sure you want to change the base?

PS Support for Migration AvailaibilitySets APIs #29085

Conversation

srcharug commented Jan 20, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Description

Mandatory Checklist

Uh oh!

azure-client-tools-bot-prd bot commented Jan 20, 2026

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Reviewed changes

Uh oh!

Copilot AI Jan 20, 2026

Choose a reason for hiding this comment

Uh oh!

Copilot AI Jan 20, 2026

Choose a reason for hiding this comment

Uh oh!

Copilot AI Jan 20, 2026

Choose a reason for hiding this comment

Uh oh!

Copilot AI Jan 20, 2026

Choose a reason for hiding this comment

Uh oh!

Copilot AI Jan 20, 2026

Choose a reason for hiding this comment

Uh oh!

Copilot AI Jan 20, 2026

Choose a reason for hiding this comment

Uh oh!

isra-fel commented Jan 20, 2026

Uh oh!

azure-pipelines bot commented Jan 20, 2026

Uh oh!

Uh oh!

Uh oh!

vidai-msft Jan 22, 2026

Choose a reason for hiding this comment

Uh oh!

github-actions bot commented Jan 22, 2026

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Uh oh!

Copilot AI Jan 23, 2026

Choose a reason for hiding this comment

Uh oh!

Copilot AI Jan 23, 2026

Choose a reason for hiding this comment

Uh oh!

Copilot AI Jan 23, 2026

Choose a reason for hiding this comment

Uh oh!

Copilot AI Jan 23, 2026

Choose a reason for hiding this comment

Uh oh!

Copilot AI Jan 23, 2026

Choose a reason for hiding this comment

Uh oh!

isra-fel commented Jan 23, 2026

Uh oh!

azure-pipelines bot commented Jan 23, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

6 participants

srcharug commented Jan 20, 2026 •

edited

Loading