Generated SDK modules and recorded tests for UltraSSDCapability prope…

…rty in Dedicated Hosts for API version 2022-03-01 Compute #27949 (#27984) * generated * updated code Co-authored-by: Theodore Chang <thchan@microsoft.com>
Azure · May 10, 2022 · 4229cd6 · 4229cd6
1 parent 0f06c1c
commit 4229cd6
Show file tree

Hide file tree

Showing 7 changed files with 769 additions and 224 deletions.
diff --git a/sdk/compute/Microsoft.Azure.Management.Compute/src/Generated/Models/DedicatedHostGroup.cs b/sdk/compute/Microsoft.Azure.Management.Compute/src/Generated/Models/DedicatedHostGroup.cs
diff --git a/...gement.Compute/src/Generated/Models/DedicatedHostGroupPropertiesAdditionalCapabilities.cs b/...gement.Compute/src/Generated/Models/DedicatedHostGroupPropertiesAdditionalCapabilities.cs
diff --git a/...mpute/Microsoft.Azure.Management.Compute/src/Generated/Models/DedicatedHostGroupUpdate.cs b/...mpute/Microsoft.Azure.Management.Compute/src/Generated/Models/DedicatedHostGroupUpdate.cs
diff --git a/sdk/compute/Microsoft.Azure.Management.Compute/src/Microsoft.Azure.Management.Compute.csproj b/sdk/compute/Microsoft.Azure.Management.Compute/src/Microsoft.Azure.Management.Compute.csproj
@@ -9,7 +9,7 @@
       Provides developers with libraries for the updated compute platform under Azure Resource manager to deploy virtual machine, virtual machine extensions and availability set management capabilities. Launch, restart, scale, capture and manage VMs, VM Extensions and more. Note: This client library is for Virtual Machines under Azure Resource Manager.
       Development of this library has shifted focus to the Azure Unified SDK. The future development will be focused on "Azure.ResourceManager.Compute" (https://www.nuget.org/packages/Azure.ResourceManager.Compute/). Please see the package changelog for more information.
     </Description>
-    <Version>54.1.0-preview.1</Version>
+    <Version>54.0.0</Version>
     <AssemblyName>Microsoft.Azure.Management.Compute</AssemblyName>
     <PackageTags>management;virtual machine;compute;</PackageTags>
     <PackageReleaseNotes>

diff --git a/sdk/compute/Microsoft.Azure.Management.Compute/tests/ScenarioTests/DedicatedHostTests.cs b/sdk/compute/Microsoft.Azure.Management.Compute/tests/ScenarioTests/DedicatedHostTests.cs
@@ -30,6 +30,7 @@ public void TestDedicatedHostOperations()
                 string baseRGName = ComputeManagementTestUtilities.GenerateName(TestPrefix);
                 string rgName = baseRGName + "DH";
                 string dhgName = "DHG-1";
+                string dhgWithUltraSSDName = "DHG-UltraSSD-1";
                 string dhName = "DH-1";
 
                 try
@@ -64,7 +65,7 @@ public void TestDedicatedHostOperations()
                     var createdDH = CreateDedicatedHost(rgName, dhgName, dhName, "ESv3-Type1");
                     var returnedDH = m_CrpClient.DedicatedHosts.Get(rgName, dhgName, dhName);
                     ValidateDedicatedHost(createdDH, returnedDH);
-
+                    
                     //List DedicatedHosts
                     var listDHsResponse = m_CrpClient.DedicatedHosts.ListByHostGroup(rgName, dhgName);
                     Assert.Single(listDHsResponse);
@@ -74,6 +75,10 @@ public void TestDedicatedHostOperations()
                     m_CrpClient.DedicatedHosts.Delete(rgName, dhgName, dhName);
                     m_CrpClient.DedicatedHostGroups.Delete(rgName, dhgName);
 
+                    // Create a dedicated host group with ultraSSDCapabilty set to true, then get the dedicated host group and validate that they match
+                    createdDHG = CreateDedicatedHostGroup(rgName, dhgWithUltraSSDName, ultraSSDCapability: true);
+                    returnedDHG = m_CrpClient.DedicatedHostGroups.Get(rgName, dhgWithUltraSSDName);
+                    ValidateDedicatedHostGroup(createdDHG, returnedDHG);
                 }
                 finally
                 {
@@ -192,6 +197,12 @@ private void ValidateDedicatedHostGroup(DedicatedHostGroup expectedDHG, Dedicate
                 Assert.Equal(expectedDHG.Location, actualDHG.Location);
                 Assert.Equal(expectedDHG.Name, actualDHG.Name);
                 Assert.Equal(expectedDHG.SupportAutomaticPlacement, actualDHG.SupportAutomaticPlacement);
+
+                if(expectedDHG.AdditionalCapabilities != null)
+                {
+                    Assert.NotNull(actualDHG.AdditionalCapabilities);
+                    Assert.Equal(expectedDHG.AdditionalCapabilities.UltraSSDEnabled, actualDHG.AdditionalCapabilities.UltraSSDEnabled);
+                }
             }
         }
 

diff --git a/sdk/compute/Microsoft.Azure.Management.Compute/tests/ScenarioTests/VMTestBase.cs b/sdk/compute/Microsoft.Azure.Management.Compute/tests/ScenarioTests/VMTestBase.cs
@@ -1027,7 +1027,7 @@ protected VirtualMachine CreateDefaultVMInput(string rgName, string storageAccou
             return vm;
         }
 
-        protected DedicatedHostGroup CreateDedicatedHostGroup( string rgName, string dedicatedHostGroupName, int? availabilityZone = 1)
+        protected DedicatedHostGroup CreateDedicatedHostGroup(string rgName, string dedicatedHostGroupName, int? availabilityZone = 1, bool ultraSSDCapability = false)
         {
             m_ResourcesClient.ResourceGroups.CreateOrUpdate(
                    rgName,
@@ -1042,7 +1042,11 @@ protected DedicatedHostGroup CreateDedicatedHostGroup( string rgName, string ded
                 Location = m_location,
                 Zones = availabilityZone == null ? null : new List<string> { availabilityZone.ToString() },
                 PlatformFaultDomainCount = 1,
-                SupportAutomaticPlacement = true
+                SupportAutomaticPlacement = true,
+                AdditionalCapabilities = new DedicatedHostGroupPropertiesAdditionalCapabilities
+                {
+                    UltraSSDEnabled = ultraSSDCapability
+                }
             };
             return m_CrpClient.DedicatedHostGroups.CreateOrUpdate(rgName, dedicatedHostGroupName, dedicatedHostGroup);
         }