Fixing pipelining issues

Author: avneeshraiusit

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/Network/GetAzureRMSiteRecoveryNetwork.cs

@@ -32,8 +32,8 @@ public class GetAzureRMSiteRecoveryNetwork : SiteRecoveryCmdletBase
         /// Gets or sets Server object.
         /// </summary>
         [Parameter(ParameterSetName = ASRParameterSets.ByServerObject, Mandatory = true, ValueFromPipeline = true)]
-        [Parameter(ParameterSetName = ASRParameterSets.ByName, Mandatory = true)]
-        [Parameter(ParameterSetName = ASRParameterSets.ByFriendlyName, Mandatory = true)]
+        [Parameter(ParameterSetName = ASRParameterSets.ByName, Mandatory = true, ValueFromPipeline = true)]
+        [Parameter(ParameterSetName = ASRParameterSets.ByFriendlyName, Mandatory = true, ValueFromPipeline = true)]
         [ValidateNotNullOrEmpty]
         public ASRServer Server { get; set; }
 

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/Network/GetAzureRMSiteRecoveryNetworkMapping.cs

@@ -31,8 +31,8 @@ public class GetAzureRMSiteRecoveryNetworkMapping : SiteRecoveryCmdletBase
         /// <summary>
         /// Gets or sets Primary Server object.
         /// </summary>
-        [Parameter(ParameterSetName = ASRParameterSets.EnterpriseToEnterprise, Mandatory = true)]
-        [Parameter(ParameterSetName = ASRParameterSets.EnterpriseToAzure, Mandatory = true)]
+        [Parameter(ParameterSetName = ASRParameterSets.EnterpriseToEnterprise, Mandatory = true, ValueFromPipeline = true)]
+        [Parameter(ParameterSetName = ASRParameterSets.EnterpriseToAzure, Mandatory = true, ValueFromPipeline = true)]
         [ValidateNotNullOrEmpty]
         public ASRServer PrimaryServer { get; set; }
 

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/Network/NewAzureRMSiteRecoveryNetworkMapping.cs

@@ -32,8 +32,8 @@ public class NewAzureRMSiteRecoveryNetworkMapping : SiteRecoveryCmdletBase
         /// <summary>
         /// Gets or sets Primary Network object.
         /// </summary>
-        [Parameter(ParameterSetName = ASRParameterSets.EnterpriseToEnterprise, Mandatory = true)]
-        [Parameter(ParameterSetName = ASRParameterSets.EnterpriseToAzure, Mandatory = true)]
+        [Parameter(ParameterSetName = ASRParameterSets.EnterpriseToEnterprise, Mandatory = true, ValueFromPipeline = true)]
+        [Parameter(ParameterSetName = ASRParameterSets.EnterpriseToAzure, Mandatory = true, ValueFromPipeline = true)]
         [ValidateNotNullOrEmpty]
         public ASRNetwork PrimaryNetwork { get; set; }
 

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/Policy/RemoveAzureSiteRecoveryPolicy.cs

@@ -32,7 +32,7 @@ public class RemoveAzureSiteRecoveryPolicy : SiteRecoveryCmdletBase
         /// <summary>
         /// Gets or sets Name of the Policy.
         /// </summary>
-        [Parameter(Mandatory = true)]
+        [Parameter(Mandatory = true, ValueFromPipeline = true)]
         public ASRPolicy Policy { get; set; }
 
         #endregion Parameters

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/ProtectionEntity/UpdateAzureSiteRecoveryProtectionDirection.cs

@@ -62,8 +62,8 @@ public class UpdateAzureSiteRecoveryProtection : SiteRecoveryCmdletBase
         /// <summary>
         /// Gets or sets Failover direction for the recovery plan.
         /// </summary>
-        [Parameter(ParameterSetName = ASRParameterSets.ByRPObject, Mandatory = true, ValueFromPipeline = true)]
-        [Parameter(ParameterSetName = ASRParameterSets.ByPEObject, Mandatory = true, ValueFromPipeline = false)]
+        [Parameter(ParameterSetName = ASRParameterSets.ByRPObject, Mandatory = true)]
+        [Parameter(ParameterSetName = ASRParameterSets.ByPEObject, Mandatory = true)]
         [ValidateSet(
             Constants.PrimaryToRecovery,
             Constants.RecoveryToPrimary)]

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/RecoveryPlan/EditAzureSiteRecoveryRecoveryPlan.cs

@@ -34,7 +34,7 @@ public class EditAzureSiteRecoveryRecoveryPlan : SiteRecoveryCmdletBase
         /// <summary>
         /// Gets or sets Name of the Recovery Plan.
         /// </summary>
-        [Parameter(Mandatory = true)]
+        [Parameter(Mandatory = true, ValueFromPipeline = true)]
         [ValidateNotNullOrEmpty]
         public ASRRecoveryPlan RecoveryPlan { get; set; }
 

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/RecoveryPlan/NewAzureSiteRecoveryRecoveryPlan.cs

@@ -100,9 +100,9 @@ public class NewAzureSiteRecoveryRecoveryPlan : SiteRecoveryCmdletBase
         /// <summary>
         /// Gets or sets Replication Frequency of the Policy in seconds.
         /// </summary>
-        [Parameter(ParameterSetName = ASRParameterSets.EnterpriseToEnterprise, Mandatory = true)]
-        [Parameter(ParameterSetName = ASRParameterSets.EnterpriseToAzure, Mandatory = true)]
-        [Parameter(ParameterSetName = ASRParameterSets.HyperVSiteToAzure, Mandatory = true)]
+        [Parameter(ParameterSetName = ASRParameterSets.EnterpriseToEnterprise, Mandatory = true, ValueFromPipeline = true)]
+        [Parameter(ParameterSetName = ASRParameterSets.EnterpriseToAzure, Mandatory = true, ValueFromPipeline = true)]
+        [Parameter(ParameterSetName = ASRParameterSets.HyperVSiteToAzure, Mandatory = true, ValueFromPipeline = true)]
         [ValidateNotNullOrEmpty]
         public ASRProtectionEntity[] ProtectionEntityList { get; set; }
 

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/RecoveryPlan/RemoveAzureSiteRecoveryRecoveryPlan.cs

@@ -38,7 +38,7 @@ public class RemoveAzureSiteRecoveryRecoveryPlan : SiteRecoveryCmdletBase
         /// <summary>
         /// Gets or sets Name of the Recovery Plan.
         /// </summary>
-        [Parameter(Mandatory = true, ParameterSetName = ASRParameterSets.ByObject)]
+        [Parameter(Mandatory = true, ParameterSetName = ASRParameterSets.ByObject, ValueFromPipeline = true)]
         public ASRRecoveryPlan RecoveryPlan { get; set; }
 
         #endregion Parameters

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/RecoveryPlan/UpdateAzureSiteRecoveryRecoveryPlan.cs

@@ -36,7 +36,7 @@ public class UpdateAzureSiteRecoveryRecoveryPlan : SiteRecoveryCmdletBase
         /// <summary>
         /// Gets or sets Name of the Recovery Plan.
         /// </summary>
-        [Parameter(ParameterSetName = ASRParameterSets.ByRPObject, Mandatory = true)]
+        [Parameter(ParameterSetName = ASRParameterSets.ByRPObject, Mandatory = true, ValueFromPipeline = true)]
         public ASRRecoveryPlan RecoveryPlan { get; set; }
 
         /// <summary>

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/Storage/Classification/GetAzureSiteRecoveryStorageClassificationMapping.cs

@@ -19,7 +19,7 @@ public class GetAzureSiteRecoveryStorageClassificationMapping : SiteRecoveryCmdl
         /// <summary>
         /// Gets or sets name of classification.
         /// </summary>
-        [Parameter(ParameterSetName = ASRParameterSets.ByName, Mandatory = true, ValueFromPipeline = true)]
+        [Parameter(ParameterSetName = ASRParameterSets.ByName, Mandatory = true)]
         [ValidateNotNullOrEmpty]
         public string Name { get; set; }
         #endregion

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/VM/SetAzureSiteRecoveryVM.cs

@@ -33,7 +33,7 @@ public class SetAzureSiteRecoveryVM : SiteRecoveryCmdletBase
         /// <summary>
         /// Gets or sets ID of the Virtual Machine.
         /// </summary>
-        [Parameter(Mandatory = true)]
+        [Parameter(Mandatory = true, ValueFromPipeline = true)]
         [ValidateNotNullOrEmpty]
         public ASRVirtualMachine VirtualMachine { get; set; }
 
@@ -195,7 +195,7 @@ public override void ExecuteSiteRecoveryCmdlet()
             LongRunningOperationResponse response = RecoveryServicesClient.UpdateVmProperties(
                 Utilities.GetValueFromArmId(this.VirtualMachine.ID, ARMResourceTypeConstants.ReplicationFabrics),
                 Utilities.GetValueFromArmId(this.VirtualMachine.ID, ARMResourceTypeConstants.ReplicationProtectionContainers),
-                this.VirtualMachine.Name,
+                replicationProtectedItemResponse.ReplicationProtectedItem.Name,
                 input);
 
             JobResponse jobResponse =