AFS Backup-Item

Author: siddharth7

src/ResourceManager/RecoveryServices/Commands.RecoveryServices.Backup.Providers/Commands.RecoveryServices.Backup.Providers.csproj

@@ -54,7 +54,7 @@
     <Compile Include="AzureWorkloadProviderHelper.cs" />
     <Compile Include="IPsBackupProvider.cs" />
     <Compile Include="Properties\AssemblyInfo.cs" />
-    <Compile Include="Providers\AzureFilesPsBackupBackupProvider.cs" />
+    <Compile Include="Providers\AzureFilesPsBackupProvider.cs" />
     <Compile Include="Providers\AzureSqlPsBackupProvider.cs" />
     <Compile Include="Providers\DpmPsBackupProvider.cs" />
     <Compile Include="Providers\MabPsBackupProvider.cs" />

src/ResourceManager/RecoveryServices/Commands.RecoveryServices.Backup.Providers/Providers/AzureFilesPsBackupBackupProvider.cs renamed to src/ResourceManager/RecoveryServices/Commands.RecoveryServices.Backup.Providers/Providers/AzureFilesPsBackupProvider.cs

@@ -89,7 +89,18 @@ public RestAzureNS.AzureOperationResponse TriggerBackup()
 
         public RestAzureNS.AzureOperationResponse TriggerRestore()
         {
-            throw new NotImplementedException();
+            string vaultName = (string)ProviderData[VaultParams.VaultName];
+            string resourceGroupName = (string)ProviderData[VaultParams.ResourceGroupName];
+            ItemBase item = (ItemBase)ProviderData[ItemParams.Item];
+            DateTime? expiryDateTime = (DateTime?)ProviderData[ItemParams.ExpiryDateTimeUTC];
+            AzureFileShareItem azureFileShareItem = item as AzureFileShareItem;
+
+            return ServiceClientAdapter.TriggerBackup(
+                IdUtils.GetValueByName(azureFileShareItem.Id, IdUtils.IdNames.ProtectionContainerName),
+                IdUtils.GetValueByName(azureFileShareItem.Id, IdUtils.IdNames.ProtectedItemName),
+                expiryDateTime,
+                vaultName: vaultName,
+                resourceGroupName: resourceGroupName);
         }
 
         public ProtectedItemResource GetProtectedItem()

src/ResourceManager/RecoveryServices/Commands.RecoveryServices.Backup.ServiceClientAdapter/BMSAPIs/ItemAPIs.cs

@@ -136,9 +136,18 @@ public RestAzureNS.AzureOperationResponse TriggerBackup(
             string resourceGroupName = null)
         {
             BackupRequestResource triggerBackupRequest = new BackupRequestResource();
-            IaasVMBackupRequest iaasVmBackupRequest = new IaasVMBackupRequest();
-            iaasVmBackupRequest.RecoveryPointExpiryTimeInUTC = expiryDateTimeUtc;
-            triggerBackupRequest.Properties = iaasVmBackupRequest;
+            if (string.Compare(containerName.Split(';')[0], "IaasVMContainer") == 0)
+            {
+                IaasVMBackupRequest iaasVmBackupRequest = new IaasVMBackupRequest();
+                iaasVmBackupRequest.RecoveryPointExpiryTimeInUTC = expiryDateTimeUtc;
+                triggerBackupRequest.Properties = iaasVmBackupRequest;
+            }
+            else if (string.Compare(containerName.Split(';')[0], "StorageContainer") == 0)
+            {
+                AzureFileShareBackupRequest azureFileShareBackupRequest = new AzureFileShareBackupRequest();
+                azureFileShareBackupRequest.RecoveryPointExpiryTimeInUTC = expiryDateTimeUtc;
+                triggerBackupRequest.Properties = azureFileShareBackupRequest;
+            }
 
             return BmsAdapter.Client.Backups.TriggerWithHttpMessagesAsync(
                 vaultName ?? BmsAdapter.GetResourceName(),

src/ResourceManager/RecoveryServices/Commands.RecoveryServices.Backup.Test/ScenarioTests/AzureFiles/ItemTests.cs

@@ -29,5 +29,14 @@ public void TestAzureFileItem()
             TestController.NewInstance.RunPsTest(
                 _logger, PsBackupProviderTypes.AzureFiles, "Test-AzureFileItem");
         }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        [Trait(TestConstants.Workload, TestConstants.AzureVM)]
+        public void TestAzureFileShareBackup()
+        {
+            TestController.NewInstance.RunPsTest(
+                _logger, PsBackupProviderTypes.IaasVm, "Test-AzureFileShareBackup");
+        }
     }
 }

src/ResourceManager/RecoveryServices/Commands.RecoveryServices.Backup.Test/ScenarioTests/AzureFiles/ItemTests.ps1

@@ -27,4 +27,29 @@ function Test-AzureFileItem
 	{
 		# Cleanup
 	}
+}
+
+function Test-AzureFileShareBackup
+{
+	$location = "westus"
+	$resourceGroupName = "sisi-RSV"
+
+	try
+	{
+		# Setup
+		$vm = Create-VM $resourceGroupName $location
+		$vault = Create-RecoveryServicesVault $resourceGroupName $location
+		$item = Enable-Protection $vault $vm
+		
+		# Trigger backup and wait for completion
+		$backupJob = Backup-AzureRmRecoveryServicesBackupItem `
+			-VaultId $vault.ID `
+			-Item $item | Wait-AzureRmRecoveryServicesBackupJob -VaultId $vault.ID
+
+		Assert-True { $backupJob.Status -eq "Completed" }
+	}
+	finally
+	{
+		# Cleanup
+	}
 }