Add PassThru to Remove cmd and DefaultParamSet to Update. Updated ScenarioTests accordingly

Author: madsd

src/Websites/Websites.Test/ScenarioTests/AccessRestrictionTests.ps1

@@ -285,7 +285,7 @@ function Test-RemoveWebAppAccessRestriction
 		Assert-AreEqual "Deny" $actual.MainSiteAccessRestrictions[1].Action
 
 		# Run Tests
-		$actual = Remove-AzWebAppAccessRestrictionRule -ResourceGroupName $rgname -WebAppName $wname -Name developers
+		$actual = Remove-AzWebAppAccessRestrictionRule -ResourceGroupName $rgname -WebAppName $wname -Name developers -PassThru
 
 		# Assert
 		Assert-AreEqual 1 $actual.MainSiteAccessRestrictions.Count
@@ -379,7 +379,7 @@ function Test-RemoveWebAppAccessRestrictionScm
 		Assert-AreEqual "Deny" $actual.ScmSiteAccessRestrictions[1].Action
 
 		# Run Tests
-		$actual = Remove-AzWebAppAccessRestrictionRule -ResourceGroupName $rgname -WebAppName $wname -Name developers -TargetScmSite
+		$actual = Remove-AzWebAppAccessRestrictionRule -ResourceGroupName $rgname -WebAppName $wname -Name developers -TargetScmSite -PassThru
 
 		# Assert
 		Assert-AreEqual 1 $actual.ScmSiteAccessRestrictions.Count

src/Websites/Websites.Test/SessionRecords/Microsoft.Azure.Commands.Websites.Test.ScenarioTests.AccessRestrictionTests/TestAddWebAppAccessRestriction.json

@@ -49,6 +49,9 @@ public class RemoveAzureWebAppAccessRestrictionRuleCmdlet : WebAppBaseClientCmdL
         [Parameter(Mandatory = false, HelpMessage = "Deployment Slot name.")]
         public string SlotName { get; set; }
 
+        [Parameter(Mandatory = false, HelpMessage = "Return the access restriction config object.")]
+        public SwitchParameter PassThru { get; set; }
+
         public override void ExecuteCmdlet()
         {
             if (!string.IsNullOrWhiteSpace(ResourceGroupName) && !string.IsNullOrWhiteSpace(WebAppName))
@@ -78,10 +81,13 @@ public override void ExecuteCmdlet()
                     // Update web app configuration
                     WebsitesClient.UpdateWebAppConfiguration(ResourceGroupName, webApp.Location, WebAppName, SlotName, siteConfig);
 
-                    // Refresh object to get the final state
-                    webApp = new PSSite(WebsitesClient.GetWebApp(ResourceGroupName, WebAppName, SlotName));
-                    var accessRestrictionConfig = new PSAccessRestrictionConfig(ResourceGroupName, WebAppName, webApp.SiteConfig, SlotName);
-                    WriteObject(accessRestrictionConfig);
+                    if (PassThru)
+                    {
+                        // Refresh object to get the final state
+                        webApp = new PSSite(WebsitesClient.GetWebApp(ResourceGroupName, WebAppName, SlotName));
+                        var accessRestrictionConfig = new PSAccessRestrictionConfig(ResourceGroupName, WebAppName, webApp.SiteConfig, SlotName);
+                        WriteObject(accessRestrictionConfig);
+                    }
                 }
             }
         }

src/Websites/Websites.Test/SessionRecords/Microsoft.Azure.Commands.Websites.Test.ScenarioTests.AccessRestrictionTests/TestAddWebAppAccessRestrictionScm.json

@@ -23,7 +23,7 @@ namespace Microsoft.Azure.Commands.WebApps.Cmdlets.WebApps
     /// <summary>
     /// this commandlet will let you update the access restriction settings of an Azure Website
     /// </summary>
-    [Cmdlet("Update", ResourceManager.Common.AzureRMConstants.AzureRMPrefix + "WebAppAccessRestrictionConfig", SupportsShouldProcess = true)]
+    [Cmdlet("Update", ResourceManager.Common.AzureRMConstants.AzureRMPrefix + "WebAppAccessRestrictionConfig", SupportsShouldProcess = true, DefaultParameterSetName = InputValuesParameterSet)]
     [OutputType(typeof(PSAccessRestrictionConfig))]
     public class UpdateAzureWebAppAccessRestrictionConfigCmdlet : WebAppBaseClientCmdLet
     {

src/Websites/Websites.Test/SessionRecords/Microsoft.Azure.Commands.Websites.Test.ScenarioTests.AccessRestrictionTests/TestAddWebAppAccessRestrictionSlot.json

@@ -13,8 +13,8 @@ Removes an Access Restriction rule from an Azure Web App.
 
 ```
 Remove-AzWebAppAccessRestrictionRule [-ResourceGroupName] <String> [-WebAppName] <String> -Name <String>
- [-TargetScmSite] [-SlotName <String>] [-DefaultProfile <IAzureContextContainer>] [-WhatIf] [-Confirm]
- [<CommonParameters>]
+ [-TargetScmSite] [-SlotName <String>] [-PassThru] [-DefaultProfile <IAzureContextContainer>] [-WhatIf]
+ [-Confirm] [<CommonParameters>]
 ```
 
 ## DESCRIPTION
@@ -61,6 +61,21 @@ Accept pipeline input: False
 Accept wildcard characters: False
 ```
 
+### -PassThru
+Return the access restriction config object.
+
+```yaml
+Type: System.Management.Automation.SwitchParameter
+Parameter Sets: (All)
+Aliases:
+
+Required: False
+Position: Named
+Default value: None
+Accept pipeline input: False
+Accept wildcard characters: False
+```
+
 ### -ResourceGroupName
 Resource Group Name
 

src/Websites/Websites.Test/SessionRecords/Microsoft.Azure.Commands.Websites.Test.ScenarioTests.AccessRestrictionTests/TestGetWebAppAccessRestriction.json

@@ -11,7 +11,7 @@ Updates the inheritance of Main site Access Restiction config to SCM Site for an
 
 ## SYNTAX
 
-### InputValuesParameterSet
+### InputValuesParameterSet (Default)
 ```
 Update-AzWebAppAccessRestrictionConfig [-ResourceGroupName] <String> [-Name] <String>
  [-ScmSiteUseMainSiteRestrictionConfig] [-SlotName <String>] [-PassThru]