Add instance pool, instance pool usages cmdlets, and update managed instance cmdlets to support instance pools.

src/Sql/Sql.Test/ScenarioTests/AdvancedDataSecurityManagedInstanceTests.cs

@@ -14,6 +14,7 @@
 
 using Microsoft.Azure.Commands.ScenarioTest.SqlTests;
 using Microsoft.WindowsAzure.Commands.ScenarioTest;
+using System.Collections.Generic;
 using Xunit;
 using Xunit.Abstractions;
 using RestTestFramework = Microsoft.Rest.ClientRuntime.Azure.TestFramework;
@@ -32,6 +33,9 @@ protected override void SetupManagementClients(RestTestFramework.MockContext con
 
         public AdvancedDataSecurityManagedInstanceTests(ITestOutputHelper output) : base(output)
         {
+            base.resourceTypesToIgnoreApiVersion = new string[] {
+                "Microsoft.Sql/managedInstances"
+            };
         }
 
         [Fact]

src/Sql/Sql.Test/ScenarioTests/Common.ps1

@@ -201,8 +201,8 @@ Creates the basic test environment needed to perform the Sql data security tests
 function Create-BasicManagedTestEnvironmentWithParams ($params, $location)
 {
 	New-AzureRmResourceGroup -Name $params.rgname -Location $location
-	
-	# Setup VNET 
+
+	# Setup VNET
 	$vnetName = "cl_initial"
 	$subnetName = "Cool"
 	$virtualNetwork1 = CreateAndGetVirtualNetworkForManagedInstance $vnetName $subnetName
@@ -235,7 +235,7 @@ function Create-DataMaskingTestEnvironment ($testSuffix)
     New-AzSqlServer -ResourceGroupName  $params.rgname -ServerName $params.serverName -ServerVersion "12.0" -Location "West Central US" -SqlAdministratorCredentials $credentials
 	New-AzSqlServerFirewallRule -ResourceGroupName  $params.rgname -ServerName $params.serverName -StartIpAddress 0.0.0.0 -EndIpAddress 255.255.255.255 -FirewallRuleName "ddmRule"
 	New-AzSqlDatabase -ResourceGroupName $params.rgname -ServerName $params.serverName -DatabaseName $params.databaseName
-	
+
 	if ([Microsoft.Azure.Test.HttpRecorder.HttpMockServer]::Mode -eq "Record")
 	{
 		$fullServerName = $params.serverName + ".database.windows.net"
@@ -358,11 +358,11 @@ function Get-ManagedInstanceForTdeTest ($params)
 	$rg = Create-ResourceGroupForTest
 	$vnetName = "cl_initial"
 	$subnetName = "Cool"
-	
-	# Setup VNET 
+
+	# Setup VNET
 	$virtualNetwork1 = CreateAndGetVirtualNetworkForManagedInstance $vnetName $subnetName $rg.Location
 	$subnetId = $virtualNetwork1.Subnets.where({ $_.Name -eq $subnetName })[0].Id
-	
+
 	$managedInstance = Create-ManagedInstanceForTest $rg $subnetId
 	Set-AzKeyVaultAccessPolicy -VaultName $params.vaultName -ObjectId $managedInstance.Identity.PrincipalId -PermissionsToKeys get, list, wrapKey, unwrapKey
 
@@ -495,7 +495,7 @@ function Get-ProviderLocation($provider)
 			if ($location -eq $null)
 			{
 				return "East US"
-			} 
+			}
             else
 			{
 				return $location.Locations[0]
@@ -539,7 +539,7 @@ function Get-ServerCredential
 	$serverLogin = "testusername"
 	<#[SuppressMessage("Microsoft.Security", "CS002:SecretInNextLine", Justification="Test passwords only valid for the duration of the test")]#>
 	$serverPassword = "t357ingP@s5w0rd!"
-	$credentials = new-object System.Management.Automation.PSCredential($serverLogin, ($serverPassword | ConvertTo-SecureString -asPlainText -Force)) 
+	$credentials = new-object System.Management.Automation.PSCredential($serverLogin, ($serverPassword | ConvertTo-SecureString -asPlainText -Force))
 	return $credentials
 }
 
@@ -738,6 +738,59 @@ function Create-ManagedInstanceForTest ($resourceGroup, $subnetId)
 	return $managedInstance
 }
 
+<#
+	.SYNOPSIS
+	Creates a managed instance in an instance pool
+#>
+function Create-ManagedInstanceInInstancePoolForTest ($instancePool)
+{
+    $managedInstanceName = Get-ManagedInstanceName
+    $credentials = Get-ServerCredential
+    $vCore = 2
+    $managedInstance = $instancePool | New-AzSqlInstance -Name $managedInstanceName -VCore $vCore -AdministratorCredential $credentials -StorageSizeInGb 32 -PublicDataEndpointEnabled
+    return $managedInstance
+}
+
+function Remove-ManagedInstancesInInstancePool($instancePool)
+{
+    $instancePool | Get-AzSqlInstance | Remove-AzSqlInstance -Force
+}
+
+
+function Get-InstancePoolTestProperties()
+{
+    $tags = @{ instance="Pools" };
+    $instancePoolTestProperties = @{
+        resourceGroup = "instancePoolCSSdemo"
+        name = "cssinstancepool0"
+        subnetName = "InstancePool"
+        vnetName = "vnet-cssinstancepool0"
+        tags = $tags
+        computeGen = "Gen5"
+        edition = "GeneralPurpose"
+        location = "canadacentral"
+        licenseType = "LicenseIncluded"
+        vCores = 16
+    }
+    return $instancePoolTestProperties
+}
+
+<#
+	.SYNOPSIS
+	Creates an instance pool for Sql instance pool CRUD tests
+#>
+function Create-InstancePoolForTest()
+{
+    $props = Get-InstancePoolTestProperties
+    $virtualNetwork = CreateAndGetVirtualNetworkForManagedInstance $props.vnetName $props.subnetName $props.location $props.resourceGroup
+    $subnetId = $virtualNetwork.Subnets.where({ $_.Name -eq $props.subnetName })[0].Id
+    $instancePool = New-AzSqlInstancePool -ResourceGroupName $props.resourceGroup -Name $props.name `
+                -Location $props.location -SubnetId $subnetId -VCore $props.vCores `
+                -Edition $props.Edition -ComputeGeneration $props.computeGen `
+                -LicenseType $props.licenseType -Tag $props.tags
+    return $instancePool
+}
+
 <#
 	.SYNOPSIS
 	Create a virtual network

src/Sql/Sql.Test/ScenarioTests/DataClassificationTests.cs

@@ -23,6 +23,9 @@ public class DataClassificationTests : SqlTestsBase
     {
         public DataClassificationTests(ITestOutputHelper output) : base(output)
         {
+            base.resourceTypesToIgnoreApiVersion = new string[] {
+                "Microsoft.Sql/managedInstances"
+            };
         }
 
         protected override void SetupManagementClients(RestTestFramework.MockContext context)

src/Sql/Sql.Test/ScenarioTests/InstancePoolCrudTests.cs

@@ -0,0 +1,136 @@
+// ----------------------------------------------------------------------------------
+//
+// 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.
+// ----------------------------------------------------------------------------------
+
+using Microsoft.Azure.Commands.ScenarioTest.SqlTests;
+using Microsoft.WindowsAzure.Commands.ScenarioTest;
+using Xunit;
+using Xunit.Abstractions;
+using RestTestFramework = Microsoft.Rest.ClientRuntime.Azure.TestFramework;
+
+namespace Microsoft.Azure.Commands.Sql.Test.ScenarioTests
+{
+    public class InstancePoolCrudTests : SqlTestsBase
+    {
+        public InstancePoolCrudTests(ITestOutputHelper output) : base(output)
+        {
+        }
+
+        protected override void SetupManagementClients(RestTestFramework.MockContext context)
+        {
+            var sqlClient = GetSqlClient(context);
+            var newResourcesClient = GetResourcesClient(context);
+            var networkClient = GetNetworkClient(context);
+            Helper.SetupSomeOfManagementClients(sqlClient, newResourcesClient, networkClient);
+        }
+
+        #region Instance pool
+
+        /// <summary>
+        /// Tests creation of an instance pool
+        /// </summary>
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestCreateInstancePool()
+        {
+            RunPowerShellTest("Test-CreateInstancePool");
+        }
+
+        /// <summary>
+        /// Tests updating an instance pool
+        /// </summary>
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestUpdateInstancePool()
+        {
+            RunPowerShellTest("Test-UpdateInstancePool");
+        }
+
+        /// <summary>
+        /// Tests getting an instance pool
+        /// </summary>
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestGetInstancePool()
+        {
+            RunPowerShellTest("Test-GetInstancePool");
+        }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestRemoveInstancePool()
+        {
+            RunPowerShellTest("Test-RemoveInstancePool");
+        }
+
+        #endregion
+
+        #region Managed Instance
+
+        /// <summary>
+        /// Tests creation of a managed instance in an instance pool
+        /// </summary>
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestCreateManagedInstanceInInstancePool()
+        {
+            RunPowerShellTest("Test-CreateManagedInstanceInInstancePool");
+        }
+
+        /// <summary>
+        /// Tests getting all managed instances in an instance pool
+        /// </summary>
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestGetManagedInstanceInInstancePool()
+        {
+            RunPowerShellTest("Test-GetManagedInstanceInInstancePool");
+        }
+
+        /// <summary>
+        /// Tests updating a managed instance in an instance pool
+        /// </summary>
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestUpdateManagedInstanceInInstancePool()
+        {
+            RunPowerShellTest("Test-UpdateManagedInstanceInInstancePool");
+        }
+
+        /// <summary>
+        /// Tests updating a managed instance in an instance pool
+        /// </summary>
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestDeleteManagedInstanceInInstancePool()
+        {
+            RunPowerShellTest("Test-DeleteManagedInstanceInInstancePool");
+        }
+
+        #endregion
+
+        #region Instance Pool Usages
+
+        /// <summary>
+        /// Tests getting the instance pool usage
+        /// </summary>
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestGetInstancePoolUsage()
+        {
+            RunPowerShellTest("Test-GetInstancePoolUsage");
+        }
+
+        #endregion
+    }
+}