added test for DiskRestorePoint and AcceleratedNetwork (#25591)

* added test for DiskRestorePoint and AcceleratedNetwork

* added recorded tests

Co-authored-by: REDMOND\rytruong <[email protected]>

Author: rytruong

sdk/compute/Microsoft.Azure.Management.Compute/tests/DiskRPTests/DiskRPScenarioTests.cs

@@ -74,6 +74,12 @@ public void Snapshot_CRUD_WithSecurityType()
             Snapshot_CRUD_WithSecurityType_Execute("Snapshot_CRUD_WithSecurityType", location: "eastus2");
         }
 
+        [Fact]
+        public void Snapshot_CRUD_WithAcceleratedNetwork()
+        {
+            Snapshot_CRUD_WithAcceleratedNetwork_Execute(DiskCreateOption.Empty, "Snapshot_CRUD_WithAcceleratedNetwork", diskSizeGB: 5, location: "eastus2");
+        }
+
         [Fact]
         public void Disk_List_EmptyDisk()
         {

sdk/compute/Microsoft.Azure.Management.Compute/tests/DiskRPTests/DiskRPTestsBase.cs

@@ -392,6 +392,50 @@ protected void Snapshot_CRUD_WithSecurityType_Execute(string methodName, string
             }
         }
 
+        protected void Snapshot_CRUD_WithAcceleratedNetwork_Execute(string diskCreateOption, string methodName, int? diskSizeGB = null, string location = null)
+        {
+            using (MockContext context = MockContext.Start(this.GetType(), methodName))
+            {
+                EnsureClientsInitialized(context);
+                DiskRPLocation = location ?? DiskRPLocation;
+
+                var rgName = TestUtilities.GenerateName(TestPrefix);
+                var diskName = TestUtilities.GenerateName(DiskNamePrefix);
+                var snapshotName = TestUtilities.GenerateName(DiskNamePrefix);
+
+                Disk disk = GenerateDefaultDisk(diskCreateOption, rgName, diskSizeGB);
+                disk.SupportedCapabilities = new SupportedCapabilities { AcceleratedNetwork = true };
+
+                try
+                {
+                    m_ResourcesClient.ResourceGroups.CreateOrUpdate(rgName, new ResourceGroup { Location = DiskRPLocation });
+
+                    //put disk
+                    m_CrpClient.Disks.CreateOrUpdate(rgName, diskName, disk);
+                    Disk diskOut = m_CrpClient.Disks.Get(rgName, diskName);
+
+                    Validate(disk, diskOut, disk.Location);
+
+                    // Generate snapshot using disk info
+                    Snapshot snapshot = GenerateDefaultSnapshot(diskOut.Id, location: location);
+
+                    // **********
+                    // TEST
+                    // **********
+
+                    Snapshot snapshotOut = m_CrpClient.Snapshots.CreateOrUpdate(rgName, snapshotName, snapshot);
+                    Validate(snapshot, snapshotOut);
+                    Assert.NotNull(snapshotOut.SupportedCapabilities.AcceleratedNetwork);
+                    Assert.Equal(disk.SupportedCapabilities.AcceleratedNetwork, snapshotOut.SupportedCapabilities.AcceleratedNetwork);
+                    OperateSnapshot(snapshot, rgName, snapshotName);
+                }
+                finally
+                {
+                    m_ResourcesClient.ResourceGroups.Delete(rgName);
+                }
+            }
+        }
+
         protected void DiskEncryptionSet_CRUD_Execute(string methodName, string encryptionType, string location = null)
         {
             using (MockContext context = MockContext.Start(this.GetType(), methodName))
@@ -1474,13 +1518,11 @@ protected void Disk_CRUD_WithPurchasePlan_Execute(string diskCreateOption, strin
                     m_ResourcesClient.ResourceGroups.Delete(rgName);
                 }
             }
-
         }
 
-
         #endregion
 
-        #region Generation
+            #region Generation
         public static readonly GrantAccessData AccessDataDefault = new GrantAccessData { Access = AccessLevel.Read, DurationInSeconds = 1000 };
 
         protected Disk GenerateDefaultDisk(string diskCreateOption, string rgName, int? diskSizeGB = null, IList<string> zones = null, string location = null)

sdk/compute/Microsoft.Azure.Management.Compute/tests/ScenarioTests/RestorePointTests.cs

@@ -93,8 +93,9 @@ public void CreateRpcAndRestorePoints()
                     VerifyRestorePointDetails(createdRP, rpName, osDisk, 1,
                         excludeDiskId: dataDiskId, vmSize: vmSize);
                     RestorePoint getRP = GetRP(rgName, rpcName, rpName);
-                    VerifyRestorePointDetails(createdRP, rpName, osDisk, 1,
+                    VerifyRestorePointDetails(getRP, rpName, osDisk, 1,
                         excludeDiskId: dataDiskId, vmSize: vmSize);
+                    VerifyDiskRestorePoint(rgName, rpcName, rpName);
 
                     // get RPC without $expand=restorePoints
                     RestorePointCollection returnedRpc = GetRpc(rgName, rpcName);
@@ -110,6 +111,7 @@ public void CreateRpcAndRestorePoints()
                     RestorePoint rpInRpc = returnedRpc.RestorePoints[0];
                     VerifyRestorePointDetails(rpInRpc, rpName, osDisk, 1,
                         excludeDiskId: dataDiskId, vmSize: vmSize);
+                    VerifyDiskRestorePoint(rgName, rpcName, rpName);
 
                     // delete the restore point
                     DeleteRP(rgName, rpcName, rpName);
@@ -271,5 +273,32 @@ private RestorePointCollection CreateRpc(string sourceVMId, string rpcName,
                 inputRpc);
             return restorePointCollection;
         }
+
+        // Verify disk restore points.
+        private void VerifyDiskRestorePoint(string rgName, string rpcName, string rpName)
+        {
+            IPage<DiskRestorePoint> listDiskRestorePoint = m_CrpClient.DiskRestorePoint.ListByRestorePoint(rgName, rpcName, rpName);
+            GrantAccessData accessData = new GrantAccessData { Access = AccessLevel.Read, DurationInSeconds = 1000 };
+            foreach (DiskRestorePoint drp in listDiskRestorePoint)
+            {
+                var getDrp = m_CrpClient.DiskRestorePoint.Get(rgName, rpcName, rpName, drp.Name);
+                ValidateDiskRestorePoint(getDrp, drp.Name);
+
+                AccessUri accessUri = m_CrpClient.DiskRestorePoint.GrantAccess(rgName, rpcName, rpName, getDrp.Name, accessData);
+                Assert.NotNull(accessUri.AccessSAS);
+
+                getDrp = m_CrpClient.DiskRestorePoint.Get(rgName, rpcName, rpName, drp.Name);
+                ValidateDiskRestorePoint(getDrp, drp.Name);
+
+                m_CrpClient.DiskRestorePoint.RevokeAccess(rgName, rpcName, rpName, drp.Name);
+            }
+        }
+
+        private void ValidateDiskRestorePoint(DiskRestorePoint drp, string rpName)
+        {
+            Assert.NotNull(drp);
+            Assert.NotNull(drp.Id);
+            Assert.Equal(rpName, drp.Name);
+        }
     }
 }