LTR MI cmdlets and tests

src/Sql/Sql.Test/ScenarioTests/DataSyncTests.cs

@@ -28,7 +28,7 @@ public DataSyncTests(ITestOutputHelper output) : base(output)
 
             base.resourceTypesToIgnoreApiVersion = new string[] {
                 "Microsoft.Sql/servers",
-                 "Microsoft.Sql/managedInstances/databases"
+                "Microsoft.Sql/managedInstances/databases"
             };
         }
 
@@ -135,6 +135,6 @@ public void TestSyncMemberSchemaRefreshAndGet()
         public void TestSyncMemberRemove()
         {
             RunPowerShellTest("Test-RemoveSyncMember");
-        }                       
+        }
     }
 }

src/Sql/Sql.Test/ScenarioTests/ManagedDatabaseBackupTests.cs

@@ -52,5 +52,26 @@ public void ManagedDeletedDatabaseShortTermRetentionPolicy()
         {
             RunPowerShellTest("Test-ManagedDeletedDatabaseShortTermRetentionPolicy");
         }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestManagedInstanceLongTermRetentionPolicy()
+        {
+            RunPowerShellTest("Test-ManagedInstanceLongTermRetentionPolicy");
+        }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestManagedInstanceLongTermRetentionBackup()
+        {
+            RunPowerShellTest("Test-ManagedInstanceLongTermRetentionBackup");
+        }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestManagedInstanceLongTermRetentionResourceGroupBasedBackup()
+        {
+            RunPowerShellTest("Test-ManagedInstanceLongTermRetentionResourceGroupBasedBackup");
+        }
     }
 }

src/Sql/Sql.Test/ScenarioTests/ManagedDatabaseBackupTests.ps1

@@ -23,7 +23,7 @@ function Test-ManagedLiveDatabaseShortTermRetentionPolicy
 	$vnetName = "cl_initial"
 	$subnetName = "Cool"
 
-	# Setup VNET 
+	# Setup VNET
 	$virtualNetwork1 = CreateAndGetVirtualNetworkForManagedInstance $vnetName $subnetName $rg.Location
 	$subnetId = $virtualNetwork1.Subnets.where({ $_.Name -eq $subnetName })[0].Id
 
@@ -88,14 +88,18 @@ function Test-ManagedLiveDatabaseShortTermRetentionPolicy
 	}
 }
 
+<#
+	.SYNOPSIS
+	Test LTR Policy functions for MI
+#>
 function Test-ManagedDeletedDatabaseShortTermRetentionPolicy
 {
 	# Setup
 	$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
 
@@ -130,7 +134,7 @@ function Test-ManagedDeletedDatabaseShortTermRetentionPolicy
 		Remove-AzSqlInstanceDatabase -ResourceGroupName $rg.ResourceGroupName -InstanceName $managedInstance.ManagedInstanceName -Name $managedDatabaseName -Force
 
 		# Get deleted database
-		$deletedDatabases = Get-AzSqlDeletedInstanceDatabaseBackup -ResourceGroupName $rg.ResourceGroupName -InstanceName $managedInstance.ManagedInstanceName -DatabaseName $managedDatabaseName 
+		$deletedDatabases = Get-AzSqlDeletedInstanceDatabaseBackup -ResourceGroupName $rg.ResourceGroupName -InstanceName $managedInstance.ManagedInstanceName -DatabaseName $managedDatabaseName
 
  		# Set retention to 29, test default parameter providing.
 		$retention = 29
@@ -173,4 +177,143 @@ function Test-ManagedDeletedDatabaseShortTermRetentionPolicy
 	{
 		Remove-ResourceGroupForTest $rg
 	}
-}
+}
+<#
+	.SYNOPSIS
+	Test long term retention for managed databases.
+#>
+function Test-ManagedInstanceLongTermRetentionPolicy()
+{
+	# Setup
+	$resourceGroupName = "cl_stage_sea_cv"
+	$managedInstanceName = "seageodr-gen5-gp"
+	$weeklyRetention = "P1W"
+	$zeroRetention = "PT0S"
+
+	try
+	{
+		# create test database
+		$databaseName = "ps-ltr-policy-test-1"
+		$database = New-AzSqlInstanceDatabase -ResourceGroupName $resourceGroupName -InstanceName $managedInstanceName -Name $databaseName
+
+		Set-AzSqlInstanceDatabaseBackupLongTermRetentionPolicy -ResourceGroupName $resourceGroupName -InstanceName $managedInstanceName -DatabaseName $databaseName -WeeklyRetention $weeklyRetention
+		$policy = Get-AzSqlInstanceDatabaseBackupLongTermRetentionPolicy -ResourceGroup $resourceGroupName -InstanceName $managedInstanceName -DatabaseName $databaseName
+		Assert-AreEqual $policy.WeeklyRetention $weeklyRetention
+		Assert-AreEqual $policy.MonthlyRetention $zeroRetention
+		Assert-AreEqual $policy.YearlyRetention $zeroRetention
+	}
+	finally
+	{
+		Remove-ResourceGroupForTest $resourceGroup
+	}
+}
+
+<#
+	.SYNOPSIS
+	Test long term retention backup commands for managed databases.
+#>
+function Test-ManagedInstanceLongTermRetentionBackup
+{
+
+	# MANUAL INSTRUCTIONS
+	# Create a server and database and fill in the appropriate information below
+	# Set the weekly retention on the database so that the first backup gets picked up, for example:
+	# Set-AzSqlInstanceDatabaseBackupLongTermRetentionPolicy -ResourceGroup $resourceGroup -ServerName $serverName -DatabaseName $databaseName -WeeklyRetention P1W
+	# Wait about 18 hours until it gets properly copied and you see the backup when run get backups, for example:
+	# Get-AzSqlInstanceDatabaseLongTermRetentionBackup -Location $locationName -InstanceName $managedInstanceName -DatabaeName $databaseName
+	$resourceGroup = "cl_stage_sea_cv"
+	$locationName = "southeastasia"
+	$managedInstanceName = "seageodr-gen5-gp"
+	$databaseName = "ps-test-1"
+	$databaseWithRemovableBackup = "ps-test-2";
+
+	# Basic Get Tests
+	$backups = Get-AzSqlInstanceDatabaseLongTermRetentionBackup -Location $locationName
+	Assert-AreNotEqual $backups.Count 0
+	$backups = Get-AzSqlInstanceDatabaseLongTermRetentionBackup -Location $locationName -InstanceName $managedInstanceName
+	Assert-AreNotEqual $backups.Count 0
+	$backups = Get-AzSqlInstanceDatabaseLongTermRetentionBackup -Location $locationName -InstanceName $managedInstanceName -DatabaseName $databaseName
+	Assert-AreNotEqual $backups.Count 0
+	$backups = Get-AzSqlInstanceDatabaseLongTermRetentionBackup -Location $locationName -InstanceName $managedInstanceName -DatabaseName $databaseName -BackupName $backups[0].BackupName
+	Assert-AreNotEqual $backups.Count 0
+
+	# Test Get Optional Parameters
+	$backups = Get-AzSqlInstanceDatabaseLongTermRetentionBackup -Location $locationName -InstanceName $managedInstanceName -DatabaseName $databaseName -OnlyLatestPerDatabase
+	Assert-AreNotEqual $backups.Count 0
+	$backups = Get-AzSqlInstanceDatabaseLongTermRetentionBackup -Location $locationName -InstanceName $managedInstanceName -OnlyLatestPerDatabase -DatabaseState All
+	Assert-AreNotEqual $backups.Count 0
+
+	# Test Get Piping with Optional Parameters
+	$backups = Get-AzSqlInstanceDatabase -ResourceGroup $resourceGroup -InstanceName $managedInstanceName -Name $databaseName | Get-AzSqlInstanceDatabaseLongTermRetentionBackup -OnlyLatestPerDatabase
+	Assert-AreNotEqual $backups.Count 0
+
+	# Restore Test
+	$backups = Get-AzSqlInstanceDatabaseLongTermRetentionBackup -Location $locationName
+	$restoredDatabase = "ps-test-restore-2"
+	$db = Restore-AzSqlInstanceDatabase -FromLongTermRetentionBackup -ResourceId $backups[0].ResourceId -TargetResourceGroupName $resourceGroup -TargetInstanceName $managedInstanceName -TargetInstanceDatabaseName $restoredDatabase
+	Assert-AreEqual $db.Name $restoredDatabase
+
+	# Test Remove Backup
+	$backups = Get-AzSqlInstanceDatabaseLongTermRetentionBackup -Location $locationName -InstanceName $managedInstanceName -DatabaseName $databaseWithRemovableBackup
+	$initialBackups = $backups.Count
+	Remove-AzSqlInstanceDatabaseLongTermRetentionBackup -Location $locationName -InstanceName $managedInstanceName -DatabaseName $databaseWithRemovableBackup -BackupName $backups[0].BackupName -Force
+	$backups = Get-AzSqlInstanceDatabase -ResourceGroup $resourceGroup -InstanceName $managedInstanceName -Name $databaseWithRemovableBackup | Get-AzSqlInstanceDatabaseLongTermRetentionBackup
+	$expectedBackups = $initialBackups-1
+	Assert-AreEqual $expectedBackups $backups.Count
+
+	# drop the restored db
+	Remove-AzSqlInstanceDatabase -ResourceGroupName $resourceGroup -InstanceName $managedInstanceName -Name $restoredDatabase -Force
+}
+
+<#
+	.SYNOPSIS
+	Test long term retention backup commands for managed databases (using resource group).
+#>
+function Test-ManagedInstanceLongTermRetentionResourceGroupBasedBackup
+{
+	# MANUAL INSTRUCTIONS
+	# Create a server and database and fill in the appropriate information below
+	# Set the weekly retention on the database so that the first backup gets picked up, for example:
+	# Set-AzSqlInstanceDatabaseBackupLongTermRetentionPolicy -ResourceGroup $resourceGroup -ServerName $serverName -DatabaseName $databaseName -WeeklyRetention P1W
+	# Wait about 18 hours until it gets properly copied and you see the backup when run get backups, for example:
+	# Get-AzSqlInstanceDatabaseLongTermRetentionBackup -Location $locationName -ServerName $serverName -DatabaeName $databaseName -ResourceGroupName $resourceGroup
+	$resourceGroup = "cl_stage_sea_cv"
+	$locationName = "southeastasia"
+	$managedInstanceName = "seageodr-gen5-gp"
+	$databaseName = "ps-test-3"
+
+	# Basic Get Tests
+	$backups = Get-AzSqlInstanceDatabaseLongTermRetentionBackup -Location $locationName -ResourceGroupName $resourceGroup
+	Assert-AreNotEqual $backups.Count 0
+	$backups = Get-AzSqlInstanceDatabaseLongTermRetentionBackup -Location $locationName -InstanceName $managedInstanceName -ResourceGroupName $resourceGroup
+	Assert-AreNotEqual $backups.Count 0
+	$backups = Get-AzSqlInstanceDatabaseLongTermRetentionBackup -Location $locationName -InstanceName $managedInstanceName -DatabaseName $databaseName -ResourceGroupName $resourceGroup
+	Assert-AreNotEqual $backups.Count 0
+	$backups = Get-AzSqlInstanceDatabaseLongTermRetentionBackup -Location $locationName -InstanceName $managedInstanceName -DatabaseName $databaseName -BackupName $backups[0].BackupName -ResourceGroupName $resourceGroup
+	Assert-AreNotEqual $backups.Count 0
+
+	# Test Get Piping
+	$backups = Get-AzSqlInstanceDatabase -ResourceGroup $resourceGroup -InstanceName $managedInstanceName -Name $databaseName | Get-AzSqlInstanceDatabaseLongTermRetentionBackup
+	Assert-AreNotEqual $backups.Count 0
+	$backups = Get-AzSqlInstanceDatabase -ResourceGroup $resourceGroup -InstanceName $managedInstanceName -Name $databaseName | Get-AzSqlInstanceDatabaseLongTermRetentionBackup -BackupName $backups[0].BackupName
+	Assert-AreNotEqual $backups.Count 0
+
+	# Test Get Optional Parameters
+	$backups = Get-AzSqlInstanceDatabaseLongTermRetentionBackup -Location $locationName -InstanceName $managedInstanceName -DatabaseName $databaseName -ResourceGroupName $resourceGroup -OnlyLatestPerDatabase
+	$backups = Get-AzSqlInstanceDatabaseLongTermRetentionBackup -Location $locationName -InstanceName $managedInstanceName -ResourceGroupName $resourceGroup -DatabaseState All
+	Assert-AreNotEqual $backups.Count 0
+
+	# Test Get Piping with Optional Parameters
+	$backups = Get-AzSqlInstanceDatabase -ResourceGroup $resourceGroup -InstanceName $managedInstanceName -Name $databaseName | Get-AzSqlInstanceDatabaseLongTermRetentionBackup -OnlyLatestPerDatabase
+	Assert-AreNotEqual $backups.Count 0
+
+	# Restore Test
+	$restoredDatabase = "ps-test-restore-with-rg-2"
+	$backups = Get-AzSqlInstanceDatabaseLongTermRetentionBackup -Location $locationName -ResourceGroupName $resourceGroup
+	$db = Restore-AzSqlInstanceDatabase -FromLongTermRetentionBackup -ResourceId $backups[0].ResourceId -TargetResourceGroupName $resourceGroup -TargetInstanceName $managedInstanceName -TargetInstanceDatabaseName $restoredDatabase
+	Assert-AreEqual $db.Name $restoredDatabase
+
+	# drop the restored db
+	Remove-AzSqlInstanceDatabase -ResourceGroupName $resourceGroup -InstanceName $managedInstanceName -Name $restoredDatabase -Force
+}
+

src/Sql/Sql.Test/ScenarioTests/ManagedDatabaseCrudScenarioTests.cs

@@ -62,7 +62,7 @@ public void TestRemoveManagedDatabase()
             RunPowerShellTest("Test-RemoveManagedDatabase");
         }
 
-        [Fact]
+        [Fact(Skip = "Skip due to long setup time for managed instance")]
         [Trait(Category.AcceptanceType, Category.CheckIn)]
         public void TestRestoreManagedDatabase()
         {

src/Sql/Sql.Test/ScenarioTests/VulnerabilityAssessmentTests.cs

@@ -33,7 +33,9 @@ protected override void SetupManagementClients(RestTestFramework.MockContext con
         public VulnerabilityAssessmentTests(ITestOutputHelper output) : base(output)
         {
             base.resourceTypesToIgnoreApiVersion = new string[] {
-                "Microsoft.Sql/servers"
+                "Microsoft.Sql/servers",
+                "Microsoft.Sql/managedInstances",
+                "Microsoft.Sql/managedInstances/databases"
             };
         }
 
@@ -95,4 +97,4 @@ public void VulnerabilityAssessmentScanConvertTest()
 
         #endregion
     }
-}
+}