Added test case for Set Vault Context cmdlet usage.

Author: Samuel Anudeep

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

@@ -238,6 +238,9 @@
     <None Include="SessionRecords\Microsoft.Azure.Commands.RecoveryServices.Backup.Test.ScenarioTests.ItemTests\TestAzureSqlGetRPs.json">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>
+    <None Include="SessionRecords\Microsoft.Azure.Commands.RecoveryServices.Backup.Test.ScenarioTests.ItemTests\TestAzureVMSetVaultContext.json">
+      <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+    </None>
     <None Include="SessionRecords\Microsoft.Azure.Commands.RecoveryServices.Backup.Test.ScenarioTests.JobTests\TestAzureVMCancelJob.json">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>

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

@@ -90,5 +90,14 @@ public void TestAzureVMRPMountScript()
 
             Assert.True(AzureSession.Instance.DataStore.FileExists(mountScriptDetails.FilePath));
         }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        [Trait(TestConstants.Workload, TestConstants.AzureVM)]
+        public void TestAzureVMSetVaultContext()
+        {
+            TestController.NewInstance.RunPsTest(
+                PsBackupProviderTypes.IaasVm, "Test-AzureVMSetVaultContext");
+        }
     }
 }

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

@@ -321,4 +321,48 @@ function Test-AzureVMBackup
 		# Cleanup
 		Cleanup-ResourceGroup $resourceGroupName
 	}
+}
+
+function Test-AzureVMSetVaultContext
+{
+	$location = Get-ResourceGroupLocation
+	$resourceGroupName = Create-ResourceGroup $location
+
+	try
+	{
+		# Setup
+		$vm = Create-VM $resourceGroupName $location
+		$vault = Create-RecoveryServicesVault $resourceGroupName $location
+
+		Set-AzureRmRecoveryServicesVaultContext -Vault $vault
+
+		# Get default policy
+		$policy = Get-AzureRmRecoveryServicesBackupProtectionPolicy `
+			-Name "DefaultPolicy";
+	
+		# Enable protection
+		Enable-AzureRmRecoveryServicesBackupProtection `
+			-Policy $policy `
+			-Name $vm.Name `
+			-ResourceGroupName $vm.ResourceGroupName;
+
+		$container = Get-AzureRmRecoveryServicesBackupContainer `
+			-ContainerType AzureVM `
+			-Status Registered;
+
+		$item = Get-AzureRmRecoveryServicesBackupItem `
+			-Container $container `
+			-WorkloadType AzureVM
+
+		# Disable protection
+		Disable-AzureRmRecoveryServicesBackupProtection `
+			-Item $item `
+			-RemoveRecoveryPoints `
+			-Force;
+	}
+	finally
+	{
+		# Cleanup
+		Cleanup-ResourceGroup $resourceGroupName
+	}
 }

src/ResourceManager/RecoveryServices/Commands.RecoveryServices.Backup.Test/SessionRecords/Microsoft.Azure.Commands.RecoveryServices.Backup.Test.ScenarioTests.ItemTests/TestAzureVMSetVaultContext.json

@@ -1 +1 @@
-# Placeholder for script template
+# Placeholder for script template