Az.Compute - Shallow Replication for GalleryImageVersion (#27225)

Co-authored-by: PSCmdAssistant <[email protected]>
Co-authored-by: Yash <[email protected]>

Author: 3 people

src/Compute/Compute.Test/ScenarioTests/GalleryTests.cs

@@ -72,6 +72,13 @@ public void TestGalleryImageDefinitionDefaults()
             TestRunner.RunTestScript("Test-GalleryImageDefinitionDefaults");
         }
 
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void testgennewazgallery()
+        {
+            TestRunner.RunTestScript("TestGen-newazgallery");
+        }
+        
         [Fact(Skip = "Cannot test EOD as it prevents deletion")]
         [Trait(Category.AcceptanceType, Category.CheckIn)]
         public void TestGalleryBlockDeletionBeforeEndOfLife()

src/Compute/Compute.Test/ScenarioTests/GalleryTests.ps1

@@ -990,6 +990,9 @@ function Test-GalleryImageDefinitionDefaults
     }
 }
 
+<#
+.SYNOPSIS
+#>
 function TestGen-BlockDeletionBeforeEndOfLife
 {
     # Setup
@@ -1034,7 +1037,7 @@ function TestGen-BlockDeletionBeforeEndOfLife
             -BlockDeletionBeforeEndOfLife 
 
         Assert-AreEqual $imagever.SafetyProfile.BlockDeletionBeforeEndOfLife $True
-        
+
         $imagever = Update-AzGalleryImageVersion `
             -GalleryImageDefinitionName $imageDef.Name `
             -GalleryImageVersionName $imageVersionName `
@@ -1049,4 +1052,61 @@ function TestGen-BlockDeletionBeforeEndOfLife
         # Cleanup
         Remove-AzResourceGroup -Name $rgname -Force -ErrorAction SilentlyContinue;
     }
-}
+}
+
+<#
+.SYNOPSIS
+#>
+function TestGen-newazgallery
+{
+    # Setup
+    $rgname = Get-ComputeTestResourceName;
+    $galleryName = 'gallery' + $rgname;
+    $imageDefinitionName = 'imageDef' + $rgname;
+    $imageVersionName = '1.0.0';
+    $loc = "eastus2"
+    $vmname = "vmgallerytest"
+    $domainNameLabel="123"+$rgname
+
+    try
+    {
+
+        New-AzResourceGroup -Name $rgname -Location $loc -Force;
+
+        # SimpleParameterSet, no config, scenario.
+        # create credential 
+        $password = Get-PasswordForVM;
+        $user = Get-ComputeTestResourceName;
+        $securePassword = $password | ConvertTo-SecureString -AsPlainText -Force;  
+        $cred = New-Object System.Management.Automation.PSCredential ($user, $securePassword);
+
+        # Step 1: Create a new virtual machine in Azure
+        New-AzVM -ResourceGroupName $rgname -Location $loc -Name $vmname -Credential $cred -Size "Standard_D2s_v3" -Image "Win2022AzureEdition" -SecurityType "TrustedLaunch" -DomainNameLabel $domainNameLabel
+        $vm = get-azvm -ResourceGroupName $rgname -Name $vmname
+
+        # Create a gallery
+        $gallery = New-AzGallery -ResourceGroupName $rgname -Location $loc -Name $galleryName 
+
+        # Create a gallery image definition
+        $imageDef = New-AzGalleryImageDefinition -ResourceGroupName $rgname -GalleryName $galleryName -Location $loc -Name $imageDefinitionName -OsType 'Windows' -OsState 'Specialized' -Publisher 'Contoso' -Offer 'OfferName' -Sku 'SkuName' -Feature @{Name='SecurityType';Value='TrustedLaunch'}
+
+        # Test creating a gallery image version with ReplicationMode = Full
+        $imagever = New-AzGalleryImageVersion `
+            -GalleryImageDefinitionName $imageDef.Name `
+            -GalleryImageVersionName $imageVersionName `
+            -GalleryName $gallery.Name  `
+            -ResourceGroupName $rgname `
+            -Location $loc `
+            -SourceImageVMId $vm.Id.ToString() `
+            -PublishingProfileEndOfLifeDate '2030-12-01' `
+            -ReplicationMode 'Full'
+
+        Assert-AreEqual $imagever.PublishingProfile.ReplicationMode 'Full'
+
+    }
+    finally
+    {
+        # Cleanup
+        Remove-AzResourceGroup -Name $rgname -Force -ErrorAction SilentlyContinue;
+    }
+}

src/Compute/Compute.Test/SessionRecords/Microsoft.Azure.Commands.Compute.Test.ScenarioTests.GalleryTests/testgennewazgallery.json

@@ -20,6 +20,7 @@
 
 -->
 ## Upcoming Release
+* Added new parameter `-ReplicationMode` to `New-AzGalleryImageVersion` cmdlet.
 * Added new parameter `BlockDeletionBeforeEndOfLife` parameter to `New-AzGalleryImageVersion` and `Update-AzGalleryImageVersion` cmdlets.
 * Updated `New-AzVM`, `New-AzVmss`, `Update-AzVM`, and `Update-AzVmss` to pass `Standard` as an input of `-SecurityType` parameter.
 * Added breaking change message for `Get-AzVMSize`.

src/Compute/Compute.Test/SessionRecords/Microsoft.Azure.Commands.Compute.Test.ScenarioTests.GalleryTests/testgennewazgalleryimageversion.json

@@ -119,6 +119,15 @@ public override void ExecuteCmdlet()
                         galleryImageVersion.PublishingProfile.StorageAccountType = this.StorageAccountType;
                     }
 
+                    if (this.IsParameterBound(c => c.ReplicationMode))
+                    {
+                        if (galleryImageVersion.PublishingProfile == null)
+                        {
+                            galleryImageVersion.PublishingProfile = new GalleryImageVersionPublishingProfile();
+                        }
+                        galleryImageVersion.PublishingProfile.ReplicationMode = this.ReplicationMode;
+                    }
+
                     if (this.IsParameterBound(c => c.TargetRegion))
                     {
                         if (galleryImageVersion.PublishingProfile == null)
@@ -424,6 +433,13 @@ public override void ExecuteCmdlet()
             ValueFromPipelineByPropertyName = true,
             HelpMessage = "This boolean will be passed by the customers to enable their GalleryImageVersion resources from accidental deletions. If this boolean is set to true, the image deletions will be blocked before its EndOfLife date.")]
         public SwitchParameter BlockDeletionBeforeEndOfLife { get; set; }
+        
+        [Parameter(
+            Mandatory = false,
+            ValueFromPipelineByPropertyName = true,
+            HelpMessage = "To achieve parity with managed images, the Gallery service allows customers to designate an image version as being used for 'testing' by choosing ReplicationMode = Shallow. When choosing Shallow replication, the gallery image version is provisioned much quicker as a full copy of the source image is not made.")]
+        [PSArgumentCompleter("Full", "Shallow")]
+        public string ReplicationMode { get; set; }
     }
 
     [Cmdlet(VerbsData.Update, ResourceManager.Common.AzureRMConstants.AzureRMPrefix + "GalleryImageVersion", DefaultParameterSetName = "DefaultParameter", SupportsShouldProcess = true)]

src/Compute/Compute/ChangeLog.md

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