Geo-Restore Instance database cmdlet

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

@@ -63,5 +63,19 @@ public void TestRestoreManagedDatabase()
         {
             RunPowerShellTest("Test-RestoreManagedDatabase");
         }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestGetManagedDatabaseGeoBackup()
+        {
+            RunPowerShellTest("Test-GetManagedDatabaseGeoBackup");
+        }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestGeoRestoreManagedDatabase()
+        {
+            RunPowerShellTest("Test-GeoRestoreManagedDatabase");
+        }
     }
 }

src/Sql/Sql.Test/ScenarioTests/ManagedDatabaseCrudScenarioTests.ps1

@@ -240,4 +240,92 @@ function Test-RestoreManagedDatabase
 	{
 		Remove-ResourceGroupForTest $rg
 	}
+}
+
+<#
+.SYNOPSIS
+	Tests Getting a managed database geo-redundant backups
+#>
+function Test-GetManagedDatabaseGeoBackup
+{
+	# Setup
+	$rgName = "restore-rg"	
+	$managedInstanceName = "testbrinstance"
+	$managedDatabaseName = "sourcedb"
+
+	# Test Get using all parameters
+	$gdb1 = Get-AzSqlInstanceDatabaseGeoBackup -ResourceGroupName $rgName -InstanceName $managedInstanceName -Name $managedDatabaseName
+	Assert-NotNull $gdb1
+	Assert-AreEqual $managedDatabaseName $gdb1.Name
+
+	# Test Get using ResourceGroupName and InstanceName
+	$all = Get-AzSqlInstanceDatabaseGeoBackup -ResourceGroupName $rgName -InstanceName $managedInstanceName 
+
+	Assert-NotNull $all
+	if($all.Count -le 1)
+	{
+        throw "Should get more than 1 backup geo-redundant backups"
+    }
+}
+
+<#
+	.SYNOPSIS
+	Tests geo-restoring a managed database
+#>
+function Test-GeoRestoreManagedDatabase
+{
+	# Setup
+    $rgName = "restore-rg"	
+	$managedInstanceName = "testbrinstance"
+	$managedDatabaseName = "sourcedb"
+
+	$targetRgName = "restore-rg"
+	$targetInstanceName = "testbrinstance"
+	try
+	{
+		$sourceDbGeoBackup = Get-AzSqlInstanceDatabaseGeoBackup -ResourceGroupName $rgName -InstanceName $managedInstanceName -Name $managedDatabaseName
+
+		Assert-NotNull $sourceDbGeoBackup
+
+	   $targetManagedDatabaseName1 = Get-ManagedDatabaseName		
+		$targetManagedDatabaseName2 = Get-ManagedDatabaseName
+		$targetManagedDatabaseName3 = Get-ManagedDatabaseName
+		$targetManagedDatabaseName4 = Get-ManagedDatabaseName
+
+		# geo-restore managed database using resourceID
+		$restoredDb1 = Restore-AzSqlInstanceDatabase -FromGeoBackup -ResourceId $sourceDbGeoBackup.RecoverableDatabaseId -TargetInstanceDatabaseName $targetManagedDatabaseName1 -TargetInstanceName $targetInstanceName -TargetResourceGroupName $targetRgName
+		Assert-NotNull $restoredDb1
+		Assert-AreEqual $restoredDb1.Name $targetManagedDatabaseName1
+		Assert-AreEqual $restoredDb1.ResourceGroupName $targetRgName
+		Assert-AreEqual $restoredDb1.ManagedInstanceName $targetInstanceName
+
+		# geo-restore managed database using name, instance and resource group name 
+		$restoredDb2 = Restore-AzSqlInstanceDatabase -FromGeoBackup -ResourceGroupName $rgName -InstanceName $managedInstanceName -Name $managedDatabaseName -TargetInstanceDatabaseName $targetManagedDatabaseName2 -TargetInstanceName $targetInstanceName -TargetResourceGroupName $targetRgName
+		Assert-NotNull $restoredDb2
+		Assert-AreEqual $restoredDb2.Name $targetManagedDatabaseName2
+		Assert-AreEqual $restoredDb2.ResourceGroupName $targetRgName
+		Assert-AreEqual $restoredDb2.ManagedInstanceName $targetInstanceName
+
+		# geo-restore managed database using GeoBackupObject
+		$restoredDb3 = Restore-AzSqlInstanceDatabase -FromGeoBackup -GeoBackupObject $sourceDbGeoBackup -TargetInstanceDatabaseName $targetManagedDatabaseName3 -TargetInstanceName $targetInstanceName -TargetResourceGroupName $targetRgName
+		Assert-NotNull $restoredDb3
+		Assert-AreEqual $restoredDb3.Name $targetManagedDatabaseName3
+		Assert-AreEqual $restoredDb3.ResourceGroupName $targetRgName
+		Assert-AreEqual $restoredDb3.ManagedInstanceName $targetInstanceName
+
+		# geo-restore managed database using piping
+		$restoredDb4 = $sourceDbGeoBackup | Restore-AzSqlInstanceDatabase -FromGeoBackup -TargetInstanceDatabaseName $targetManagedDatabaseName4 -TargetInstanceName $targetInstanceName -TargetResourceGroupName $targetRgName
+		Assert-NotNull $restoredDb4
+		Assert-AreEqual $restoredDb4.Name $targetManagedDatabaseName4
+		Assert-AreEqual $restoredDb4.ResourceGroupName $targetRgName
+	   Assert-AreEqual $restoredDb4.ManagedInstanceName $targetInstanceName	
+
+	}
+	finally
+	{
+     	$restoredDb1 | Remove-AzSqlInstanceDatabase -Force
+		$restoredDb2 | Remove-AzSqlInstanceDatabase -Force
+		$restoredDb3 | Remove-AzSqlInstanceDatabase -Force
+		$restoredDb4 | Remove-AzSqlInstanceDatabase -Force
+	}
 }