Extract common SKU prefixes

Author: jaredmoo

src/Sql/Sql.Test/UnitTests/AzureSqlDatabaseUnitTests.cs

@@ -55,6 +55,9 @@ public void GetDatabaseSkuName()
             Assert.Equal(
                 "BC",
                 AzureSqlDatabaseAdapter.GetDatabaseSkuName("BusinessCritical"));
+            Assert.Equal(
+                "HS",
+                AzureSqlDatabaseAdapter.GetDatabaseSkuName("Hyperscale"));
         }
     }
 }

src/Sql/Sql/Common/SqlSkuUtils.cs

@@ -0,0 +1,53 @@
+// ----------------------------------------------------------------------------------
+//
+// Copyright Microsoft Corporation
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+// http://www.apache.org/licenses/LICENSE-2.0
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+// ----------------------------------------------------------------------------------
+
+namespace Microsoft.Azure.Commands.Sql.Common
+{
+    /// <summary>
+    /// Utility methods to help with SKU's.
+    /// </summary>
+    internal static class SqlSkuUtils
+    {
+        /// <summary>
+        /// Gets sku name based on edition name for vcore editions.
+        /// Returns null for unknown editions.
+        /// 
+        ///    Edition              | SkuName
+        ///    GeneralPurpose       | GP
+        ///    BusinessCritical     | BC
+        ///    Hyperscale           | HS
+        /// </summary>
+        /// <param name="tier">Azure Sql database edition</param>
+        /// <returns>The sku name</returns>
+        public static string GetVcoreSkuPrefix(string tier)
+        {
+            if (string.IsNullOrWhiteSpace(tier))
+            {
+                return null;
+            }
+
+            switch (tier.ToLowerInvariant())
+            {
+                case "generalpurpose":
+                    return "GP";
+                case "businesscritical":
+                    return "BC";
+                case "hyperscale":
+                    return "HS";
+                default:
+                    return null;
+            }
+        }
+    }
+}

src/Sql/Sql/Database/Services/AzureSqlDatabaseAdapter.cs

@@ -23,6 +23,7 @@
 using System.Collections.Generic;
 using System.Globalization;
 using System.Linq;
+using Microsoft.Azure.Commands.Sql.Common;
 using Microsoft.Azure.Management.Sql.Models;
 
 namespace Microsoft.Azure.Commands.Sql.Database.Services
@@ -333,6 +334,7 @@ public void RenameDatabase(string resourceGroupName, string serverName, string d
         ///    Edition              | SkuName
         ///    GeneralPurpose       | GP
         ///    BusinessCritical     | BC
+        ///    Hyperscale           | HS
         ///    Standard             | Standard
         ///    Basic                | Basic
         ///    Premium              | Premium
@@ -342,17 +344,11 @@ public void RenameDatabase(string resourceGroupName, string serverName, string d
         public static string GetDatabaseSkuName(string tier)
         {
             if (string.IsNullOrWhiteSpace(tier))
-                return null;
-
-            switch(tier.ToLowerInvariant())
             {
-                case "generalpurpose":
-                    return "GP";
-                case "businesscritical":
-                    return "BC";
-                default:
-                    return tier;
+                return null;
             }
+
+            return SqlSkuUtils.GetVcoreSkuPrefix(tier) ?? tier;
         }
 
         /// <summary>

src/Sql/Sql/Elastic Pools/Services/AzureSqlElasticPoolAdapter.cs

@@ -25,6 +25,7 @@
 using Microsoft.Azure.Commands.ResourceManager.Common.Tags;
 using Microsoft.Azure.Commands.Common.Authentication.Abstractions;
 using System.Globalization;
+using Microsoft.Azure.Commands.Sql.Common;
 
 namespace Microsoft.Azure.Commands.Sql.ElasticPool.Services
 {
@@ -420,17 +421,11 @@ private AzureSqlElasticPoolModel CreateElasticPoolModelFromResponse(string resou
         public static string GetPoolSkuName(string tier)
         {
             if (string.IsNullOrWhiteSpace(tier))
-                return null;
-
-            switch (tier.ToLowerInvariant())
             {
-                case "generalpurpose":
-                    return "GP";
-                case "businesscritical":
-                    return "BC";
-                default:
-                    return string.Format("{0}Pool", tier);
+                return null;
             }
+
+            return SqlSkuUtils.GetVcoreSkuPrefix(tier) ?? string.Format("{0}Pool", tier);
         }
     }
 }