Update test and fix parameter issue

Author: audreyttt

src/Compute/Compute.Test/ScenarioTests/DiskRPTests.ps1

@@ -1959,23 +1959,29 @@ Test SupportedSecurityOption Parameter during creation of managed disk using Cre
 function Test-SupportedSecurityOption 
 {
 	$rgname = Get-ComputeTestResourceName;
-	$loc = "eastus2euap";
+	$loc = Get-ComputeVMLocation;
 
     try{
     	New-AzResourceGroup -Name $rgname -Location $loc -Force;
-        $sourceUri = "https://teststorage.blob.core.windows.net/vhds/test.vhd"
-        $subId = Get-SubscriptionIdFromResourceGroup $rgname;
-        $storageAccountId = "/subscriptions/$subId/resourceGroups/$rgname/providers/Microsoft.Storage/storageAccounts/teststorage"
-
-        $diskConfig = New-AzDiskConfig -Location $loc -AccountType 'Premium_LRS' -CreateOption Import -SourceUri $sourceUri -StorageAccountId $storageAccountId -SupportedSecurityOption "TrustedLaunchSupported"
+        # Get the current Azure context (subscription, tenant, account)
+        $context = Get-AzContext
+        Write-Debug "Current Subscription: $($context.Subscription.Name) ($($context.Subscription.Id))"
+        Write-Debug "Current Tenant: $($context.Tenant.Id)"
+        Write-Debug "Current Account: $($context.Account.Id)"
+        Write-Debug "Current region: $loc"
+
+        $diskConfig = New-AzDiskConfig -Location $loc -SkuName 'PremiumV2_LRS' -DiskSizeGB 2 -CreateOption Empty -SupportedSecurityOption 'TrustedLaunchSupported';
 		$diskname = "disk" + $rgname;
 		New-AzDisk -ResourceGroupName $rgname -DiskName $diskname -Disk $diskConfig;
         $disk = Get-AzDisk -ResourceGroupName $rgname -DiskName $diskname;
-        Assert-AreEqual $disk.SupportedSecurityOption "TrustedLaunchSupported";
+        
+        # Check the SupportedCapabilities object
+        Assert-NotNull $disk.SupportedCapabilities;
+        Assert-AreEqual "TrustedLaunchSupported" $disk.SupportedCapabilities.SupportedSecurityOption;
 
         $updateconfig = New-AzDiskUpdateConfig -SupportedSecurityOption "TrustedLaunchAndConfidentialVMSupported";
         $disk = Update-AzDisk -ResourceGroupName $rgname -DiskName $diskname -DiskUpdate $updateconfig;
-        Assert-AreEqual $disk.SupportedSecurityOption "TrustedLaunchAndConfidentialVMSupported";
+        Assert-AreEqual $disk.SupportedCapabilities.SupportedSecurityOption; "TrustedLaunchAndConfidentialVMSupported";
     }
 
     finally

src/Compute/Compute/Generated/Disk/Config/NewAzureRmDiskConfigCommand.cs

@@ -498,6 +498,15 @@ private void Run()
                 vSupportedCapabilities.Architecture = this.Architecture;
             }
 
+            if (this.IsParameterBound(c => c.SupportedSecurityOption))
+            {
+                if (vSupportedCapabilities == null)
+                {
+                    vSupportedCapabilities = new SupportedCapabilities();
+                }
+                vSupportedCapabilities.SupportedSecurityOption = this.SupportedSecurityOption;
+            }
+
             var vDisk = new PSDisk
             {
                 Zones = this.IsParameterBound(c => c.Zone) ? this.Zone : null,
@@ -525,8 +534,7 @@ private void Run()
                 SupportedCapabilities = vSupportedCapabilities,
                 PublicNetworkAccess = this.IsParameterBound(c => c.PublicNetworkAccess) ? PublicNetworkAccess : null,
                 DataAccessAuthMode = this.IsParameterBound(c => c.DataAccessAuthMode) ? DataAccessAuthMode : null,
-                OptimizedForFrequentAttach = this.IsParameterBound(c => c.OptimizedForFrequentAttach) ? OptimizedForFrequentAttach : null,
-                SupportedSecurityOption = this.IsParameterBound(c => c.SupportedSecurityOption) ? SupportedSecurityOption : null
+                OptimizedForFrequentAttach = this.IsParameterBound(c => c.OptimizedForFrequentAttach) ? OptimizedForFrequentAttach : null
             };
 
             WriteObject(vDisk);

src/Compute/Compute/Generated/Disk/Config/NewAzureRmDiskUpdateConfigCommand.cs

@@ -179,6 +179,12 @@ public partial class NewAzureRmDiskUpdateConfigCommand : Microsoft.Azure.Command
         [PSArgumentCompleter("X64", "Arm64")]
         public string Architecture { get; set; }
 
+        [Parameter(
+            Mandatory = false,
+            ValueFromPipelineByPropertyName = true,
+            HelpMessage = "Refers to the security capability of the disk supported to create a Trusted launch or Confidential VM.")]
+        [PSArgumentCompleter("TrustedLaunchSupported", "TrustedLaunchAndConfidentialVMSupported")]
+        public string SupportedSecurityOption { get; set; }
 
         protected override void ProcessRecord()
         {
@@ -301,6 +307,15 @@ private void Run()
                 vSupportedCapabilities.Architecture = this.Architecture;
             }
 
+            if (this.IsParameterBound(c => c.SupportedSecurityOption))
+            {
+                if (vSupportedCapabilities == null)
+                {
+                    vSupportedCapabilities = new SupportedCapabilities();
+                }
+                vSupportedCapabilities.SupportedSecurityOption = this.SupportedSecurityOption;
+            }
+
             var vDiskUpdate = new PSDiskUpdate
             {
                 OsType = this.IsParameterBound(c => c.OsType) ? this.OsType : (OperatingSystemTypes?)null,

src/Compute/Compute/Generated/Models/PSDisk.cs

@@ -81,6 +81,5 @@ public string ResourceGroupName
         public bool? OptimizedForFrequentAttach { get; set; }
         public string SecurityDataUri { get; set; }
         public string SecurityMetadataUri { get; set; }
-        public string SupportedSecurityOption { get; set; }
     }
 }