Merge branch 'dev' into neha-dev

Author: NehaRaje

src/ServiceManagement/RecoveryServices/Commands.RecoveryServices/Properties/Resources.Designer.cs

@@ -194,4 +194,10 @@ ClientRequestId: {3}</value>
   <data name="SubscriptionIsNotAssociatedWithTheAccount" xml:space="preserve">
     <value>Subscription {0} is not associated with the account</value>
   </data>
+  <data name="ProtectionEntityAlreadyDisabled" xml:space="preserve">
+    <value>Protection entity {0} is already disabled</value>
+  </data>
+  <data name="ProtectionEntityAlreadyEnabled" xml:space="preserve">
+    <value>Protection entity {0} is already enabled</value>
+  </data>
 </root>

src/ServiceManagement/RecoveryServices/Commands.RecoveryServices/Properties/Resources.resx

@@ -35,6 +35,11 @@ public class SetAzureSiteRecoveryProtectionEntity : RecoveryServicesCmdletBase
         /// </summary>
         private JobResponse jobResponse = null;
 
+        /// <summary>
+        /// Protection Status of the entity.
+        /// </summary>
+        private bool alreadyEnabled = false;
+
         /// <summary>
         /// Holds either Name (if object is passed) or ID (if IDs are passed) of the PE.
         /// </summary>
@@ -52,7 +57,7 @@ public class SetAzureSiteRecoveryProtectionEntity : RecoveryServicesCmdletBase
         /// </summary>
         [Parameter(ParameterSetName = ASRParameterSets.ByIDs, Mandatory = true)]
         [ValidateNotNullOrEmpty]
-        public string ProtectionContianerId { get; set; }
+        public string ProtectionContainerId { get; set; }
 
         /// <summary>
         /// Gets or sets Protection Entity Object.
@@ -93,14 +98,38 @@ public override void ExecuteCmdlet()
             {
                 case ASRParameterSets.ByPEObject:
                     this.Id = this.ProtectionEntity.ID;
-                    this.ProtectionContianerId = this.ProtectionEntity.ProtectionContainerId;
+                    this.ProtectionContainerId = this.ProtectionEntity.ProtectionContainerId;
                     this.targetNameOrId = this.ProtectionEntity.Name;
+                    this.alreadyEnabled = this.ProtectionEntity.Protected;
+
                     break;
                 case ASRParameterSets.ByIDs:
                     this.targetNameOrId = this.Id;
+                    ProtectionEntityResponse protectionEntityResponse =
+                        RecoveryServicesClient.GetAzureSiteRecoveryProtectionEntity(
+                        this.ProtectionContainerId,
+                        this.Id);
+                    this.alreadyEnabled = protectionEntityResponse.ProtectionEntity.Protected;
+                    this.targetNameOrId = protectionEntityResponse.ProtectionEntity.Name;
+
                     break;
             }
 
+            if (this.alreadyEnabled &&
+                this.Protection.Equals(PSRecoveryServicesClient.EnableProtection, StringComparison.OrdinalIgnoreCase))
+            {
+                throw new ArgumentException(
+                    Properties.Resources.ProtectionEntityAlreadyEnabled,
+                    this.targetNameOrId);
+            }
+            else if (!this.alreadyEnabled &&
+                this.Protection.Equals(PSRecoveryServicesClient.DisableProtection, StringComparison.OrdinalIgnoreCase))
+            {
+                throw new ArgumentException(
+                    Properties.Resources.ProtectionEntityAlreadyDisabled,
+                    this.targetNameOrId);
+            }
+
             this.ConfirmAction(
                 this.Force.IsPresent || 0 != string.CompareOrdinal(this.Protection, PSRecoveryServicesClient.DisableProtection),
                 string.Format(Properties.Resources.DisableProtectionWarning, this.targetNameOrId),
@@ -112,7 +141,7 @@ public override void ExecuteCmdlet()
                         {
                             this.jobResponse =
                                 RecoveryServicesClient.SetProtectionOnProtectionEntity(
-                                this.ProtectionContianerId,
+                                this.ProtectionContainerId,
                                 this.Id,
                                 this.Protection);