PS Enable-Disable Protected Item Test cases

Author: pragrawa

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

@@ -31,5 +31,19 @@ public void TestGetItemScenario()
         {
             this.RunPowerShellTest(PsBackupProviderTypes.IaasVm.ToString(), "Test-GetItemScenario");
         }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestEnableAzureVMProtectionScenario()
+        {
+            this.RunPowerShellTest(PsBackupProviderTypes.IaasVm.ToString(), "Test-EnableAzureVMProtectionScenario");
+        }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestDisableAzureVMProtectionScenario()
+        {
+            this.RunPowerShellTest(PsBackupProviderTypes.IaasVm.ToString(), "Test-DisableAzureVMProtectionScenario");
+        }
     }
 }

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

@@ -22,4 +22,30 @@ function Test-GetItemScenario
 
 	$item = Get-AzureRmRecoveryServicesItem -Container $namedContainer -WorkloadType "AzureVM";
 	echo $item.Name;
+}
+
+function Test-EnableAzureVMProtectionScenario
+{
+	$vault = Get-AzureRmRecoveryServicesVault -ResourceGroupName "phaniktRSV" -Name "phaniktRs1";
+	Set-AzureRmRecoveryServicesVaultContext -Vault $vault;
+
+	$policy = Get-AzureRmRecoveryServicesProtectionPolicy -Name "pwtest1"	
+
+	$job = Enable-AzureRmRecoveryServicesProtection -Name "mylinux1" -ResourceGroupName "mylinux1"  -WorkloadType "AzureVM" -Policy $policy;
+	Assert-AreEqual $job.Status "Completed";
+
+}
+
+function Test-DisableAzureVMProtectionScenario
+{
+	$vault = Get-AzureRmRecoveryServicesVault -ResourceGroupName "phaniktRSV" -Name "phaniktRs1";
+	Set-AzureRmRecoveryServicesVaultContext -Vault $vault;
+	
+	$namedContainer = Get-AzureRmRecoveryServicesContainer -ContainerType "AzureVM" -Status "Registered" -Name "mylinux1";
+	Assert-AreEqual $namedContainer.FriendlyName "mylinux1";
+
+	$item = Get-AzureRmRecoveryServicesItem -Container $namedContainer -WorkloadType "AzureVM";
+
+	$job = Disable-AzureRmRecoveryServicesProtection -Item $item -Force;
+	Assert-AreEqual $job.Status "Completed";
 }