Adding support for ShouldProcess in Removal (destructive) cmdlets.

Author: avneeshraiusit

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/Fabrics/RemoveAzureRmSiteRecoveryFabric.cs

@@ -25,7 +25,7 @@ namespace Microsoft.Azure.Commands.SiteRecovery
     /// <summary>
     /// Creates Azure Site Recovery Policy object in memory.
     /// </summary>
-    [Cmdlet(VerbsCommon.Remove, "AzureRmSiteRecoveryFabric", DefaultParameterSetName = ASRParameterSets.Default)]
+    [Cmdlet(VerbsCommon.Remove, "AzureRmSiteRecoveryFabric", DefaultParameterSetName = ASRParameterSets.Default, SupportsShouldProcess = true)]
     [OutputType(typeof(ASRJob))]
     public class RemoveAzureRmSiteRecoveryFabric : SiteRecoveryCmdletBase
     {
@@ -53,22 +53,25 @@ public override void ExecuteSiteRecoveryCmdlet()
         {
             base.ExecuteSiteRecoveryCmdlet();
 
-            LongRunningOperationResponse response;
-
-            if (!this.Force.IsPresent)
-            {
-                response = RecoveryServicesClient.DeleteAzureSiteRecoveryFabric(this.Fabric.Name);
-            }
-            else
+            if (ShouldProcess(this.Fabric.FriendlyName, VerbsCommon.Remove))
             {
-                response = RecoveryServicesClient.PurgeAzureSiteRecoveryFabric(this.Fabric.Name);
-            }
-           
-            JobResponse jobResponse =
-                RecoveryServicesClient
-                .GetAzureSiteRecoveryJobDetails(PSRecoveryServicesClient.GetJobIdFromReponseLocation(response.Location));
+                LongRunningOperationResponse response;
 
-            WriteObject(new ASRJob(jobResponse.Job));
+                if (!this.Force.IsPresent)
+                {
+                    response = RecoveryServicesClient.DeleteAzureSiteRecoveryFabric(this.Fabric.Name);
+                }
+                else
+                {
+                    response = RecoveryServicesClient.PurgeAzureSiteRecoveryFabric(this.Fabric.Name);
+                }
+
+                JobResponse jobResponse =
+                    RecoveryServicesClient
+                    .GetAzureSiteRecoveryJobDetails(PSRecoveryServicesClient.GetJobIdFromReponseLocation(response.Location));
+
+                WriteObject(new ASRJob(jobResponse.Job));
+            }
         }
     }
 }

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/ProtectionContainerMapping/RemoveAzureRmSiteRecoveryProtectionContainerMapping.cs

@@ -23,7 +23,7 @@ namespace Microsoft.Azure.Commands.SiteRecovery
     /// <summary>
     /// Adds Azure Site Recovery Policy settings to a Protection Container.
     /// </summary>
-    [Cmdlet(VerbsCommon.Remove, "AzureRmSiteRecoveryProtectionContainerMapping", DefaultParameterSetName = ASRParameterSets.ByObject)]
+    [Cmdlet(VerbsCommon.Remove, "AzureRmSiteRecoveryProtectionContainerMapping", DefaultParameterSetName = ASRParameterSets.ByObject, SupportsShouldProcess = true)]
     [OutputType(typeof(ASRJob))]
     public class RemoveAzureRmSiteRecoveryProtectionContainerMapping : SiteRecoveryCmdletBase
     {
@@ -52,39 +52,42 @@ public override void ExecuteSiteRecoveryCmdlet()
         {
             base.ExecuteSiteRecoveryCmdlet();
 
-            LongRunningOperationResponse response = null;
-
-            if (!this.Force.IsPresent)
+            if (ShouldProcess(this.ProtectionContainerMapping.Name, VerbsCommon.Remove))
             {
-                RemoveProtectionContainerMappingInputProperties inputProperties = new RemoveProtectionContainerMappingInputProperties()
-                {
-                    ProviderSpecificInput = new ReplicationProviderContainerUnmappingInput()
-                };
+                LongRunningOperationResponse response = null;
 
-                RemoveProtectionContainerMappingInput input = new RemoveProtectionContainerMappingInput()
+                if (!this.Force.IsPresent)
                 {
-                    Properties = inputProperties
-                };
+                    RemoveProtectionContainerMappingInputProperties inputProperties = new RemoveProtectionContainerMappingInputProperties()
+                    {
+                        ProviderSpecificInput = new ReplicationProviderContainerUnmappingInput()
+                    };
 
-                response = RecoveryServicesClient.UnConfigureProtection(
-                    Utilities.GetValueFromArmId(this.ProtectionContainerMapping.ID, ARMResourceTypeConstants.ReplicationFabrics),
-                    Utilities.GetValueFromArmId(this.ProtectionContainerMapping.ID, ARMResourceTypeConstants.ReplicationProtectionContainers),
-                    this.ProtectionContainerMapping.Name,
-                    input);
-            }
-            else
-            {
-                response = RecoveryServicesClient.PurgeCloudMapping(
-                    Utilities.GetValueFromArmId(this.ProtectionContainerMapping.ID, ARMResourceTypeConstants.ReplicationFabrics),
-                    Utilities.GetValueFromArmId(this.ProtectionContainerMapping.ID, ARMResourceTypeConstants.ReplicationProtectionContainers),
-                    this.ProtectionContainerMapping.Name);
-            }
+                    RemoveProtectionContainerMappingInput input = new RemoveProtectionContainerMappingInput()
+                    {
+                        Properties = inputProperties
+                    };
+
+                    response = RecoveryServicesClient.UnConfigureProtection(
+                        Utilities.GetValueFromArmId(this.ProtectionContainerMapping.ID, ARMResourceTypeConstants.ReplicationFabrics),
+                        Utilities.GetValueFromArmId(this.ProtectionContainerMapping.ID, ARMResourceTypeConstants.ReplicationProtectionContainers),
+                        this.ProtectionContainerMapping.Name,
+                        input);
+                }
+                else
+                {
+                    response = RecoveryServicesClient.PurgeCloudMapping(
+                        Utilities.GetValueFromArmId(this.ProtectionContainerMapping.ID, ARMResourceTypeConstants.ReplicationFabrics),
+                        Utilities.GetValueFromArmId(this.ProtectionContainerMapping.ID, ARMResourceTypeConstants.ReplicationProtectionContainers),
+                        this.ProtectionContainerMapping.Name);
+                }
 
-            JobResponse jobResponse =
-                RecoveryServicesClient
-                .GetAzureSiteRecoveryJobDetails(PSRecoveryServicesClient.GetJobIdFromReponseLocation(response.Location));
+                JobResponse jobResponse =
+                    RecoveryServicesClient
+                    .GetAzureSiteRecoveryJobDetails(PSRecoveryServicesClient.GetJobIdFromReponseLocation(response.Location));
 
-            this.WriteObject(new ASRJob(jobResponse.Job));
+                this.WriteObject(new ASRJob(jobResponse.Job));
+            }
         }
     }
 }