[Networking] - Add Connection Draining support for load balancer backend address (#24872)

* [Networking] - Add Connection Draining support for load balancer backend address

* generate record json

Author: phrazfipho

src/Network/Network.Test/ScenarioTests/LoadBalancerBackendPoolTests.ps1

@@ -106,6 +106,7 @@ function Test-LoadBalancerBackendPoolCreate
     $testIpAddress1 = "10.0.0.5"
     $testIpAddress2 = "10.0.1.6"
     $testIpAddress3 = "10.0.0.7"
+    $adminState = "Up"
 
     $backendAddressConfigName1 = Get-ResourceName
     $backendAddressConfigName2 = Get-ResourceName
@@ -128,7 +129,7 @@ function Test-LoadBalancerBackendPoolCreate
         # Create Standard Azure load balancer
         $lb = New-AzLoadBalancer -Name $lbName -ResourceGroupName $rgname -Location $location -SKU Standard
 
-        $ip1 = New-AzLoadBalancerBackendAddressConfig -IpAddress $testIpAddress1 -Name $backendAddressConfigName1 -VirtualNetworkId $vnet.Id
+        $ip1 = New-AzLoadBalancerBackendAddressConfig -IpAddress $testIpAddress1 -Name $backendAddressConfigName1 -VirtualNetworkId $vnet.Id -AdminState $adminState
         $ip2 = New-AzLoadBalancerBackendAddressConfig -IpAddress $testIpAddress2 -Name $backendAddressConfigName2 -SubnetId $vnet.Subnets[0].Id 
         $ip3 = New-AzLoadBalancerBackendAddressConfig -IpAddress $testIpAddress3 -Name $backendAddressConfigName3 -VirtualNetworkId $vnet.Id
 
@@ -144,6 +145,7 @@ function Test-LoadBalancerBackendPoolCreate
 
         Assert-NotNull $create2
         Assert-True { @($create2.LoadBalancerBackendAddresses).Count -eq 2}
+        Assert-True { $create2.LoadBalancerBackendAddresses[0].AdminState -eq "Up"}
 
         ## create by Name without ip's
         $create3 = New-AzLoadBalancerBackendAddressPool -ResourceGroupName $rgname -LoadBalancerName $lbName -Name $backendPool3
@@ -502,6 +504,8 @@ function Test-LoadBalancerBackendAddressConfig
     $validIpAddress = "10.0.0.5"
     $invalidIpAddress2 = "xxxxx"
 
+    $adminState = "Up"
+
     $backendAddressConfigName1 = Get-ResourceName
     $backendAddressConfigName2 = Get-ResourceName
 
@@ -514,11 +518,12 @@ function Test-LoadBalancerBackendAddressConfig
         $subnet = New-AzVirtualNetworkSubnetConfig -Name $subnetName -AddressPrefix 10.0.1.0/24
         $virtualNetwork = New-AzVirtualNetwork -Name $vnetName -ResourceGroupName $rgname -Location $location -AddressPrefix 10.0.0.0/16 -Subnet $subnet
 
-        $ipconfig1 = New-AzLoadBalancerBackendAddressConfig -IpAddress $validIpAddress -Name $backendAddressConfigName1 -VirtualNetworkId $virtualNetwork.Id
+        $ipconfig1 = New-AzLoadBalancerBackendAddressConfig -IpAddress $validIpAddress -Name $backendAddressConfigName1 -VirtualNetworkId $virtualNetwork.Id -AdminState $adminState
 
         Assert-AreEqual $ipconfig1.Name $backendAddressConfigName1
         Assert-AreEqual $ipconfig1.IpAddress $validIpAddress
         Assert-AreEqual $ipconfig1.VirtualNetwork.Id $virtualNetwork.Id
+        Assert-AreEqual $ipconfig1.AdminState $adminState
 
         $ipconfig2 = New-AzLoadBalancerBackendAddressConfig -IpAddress $validIpAddress -Name $backendAddressConfigName1 -SubnetId virtualNetwork.Subnets[0].Id
 

src/Network/Network.Test/SessionRecords/Commands.Network.Test.ScenarioTests.LoadBalancerBackendPoolTests/TestLoadBalancerBackendAddressConfig.json

@@ -24,6 +24,8 @@
     - `New-AzApplicationGatewayFirewallPolicySetting`
 * Added optional property `HeaderValueMatcher` to `New-AzApplicationGatewayRewriteRuleHeaderConfiguration`
 * Added new cmdlet `New-AzApplicationGatewayHeaderValueMatcher` to support for the new property `HeaderValueMatcher`
+* Added AdminState parameter to Load Balancer Backend Address
+    - `New-AzLoadBalancerBackendAddressConfig`
 * Updated PS SDK to older SDK removing identity field
 
 ## Version 7.5.0

src/Network/Network.Test/SessionRecords/Commands.Network.Test.ScenarioTests.LoadBalancerBackendPoolTests/TestLoadBalancerBackendPoolCreate.json

@@ -74,13 +74,25 @@ public partial class NewAzureLoadBalancerBackendAddress : NetworkBaseCmdlet
         [ValidateNotNullOrEmpty]
         public string LoadBalancerFrontendIPConfigurationId { get; set; }
 
+        [Parameter(
+            Mandatory = false,
+            HelpMessage = "The admin state associated with the Backend Address config",
+            ValueFromPipelineByPropertyName = true)]
+        [PSArgumentCompleter(
+            "Up",
+            "Down",
+            "None"
+        )]
+        public string AdminState { get; set; }
+
         public override void Execute()
         {
             base.Execute();
 
             var loadBalancerBackendAddress = new PSLoadBalancerBackendAddress
             {
-                Name = this.Name
+                Name = this.Name,
+                AdminState = this.AdminState
             };
 
             if (string.Equals(ParameterSetName, "SetByIpAndVnet"))

src/Network/Network/ChangeLog.md

@@ -40,6 +40,9 @@ public partial class PSLoadBalancerBackendAddress : PSChildResource
         [JsonProperty(Order = 6)]
         public List<PSNatRulePortMapping> InboundNatRulesPortMapping { get; set; }
 
+        [JsonProperty(Order = 7)]
+        public string AdminState { get; set; }
+
         [JsonIgnore]
         public string NetworkInterfaceIpConfigurationIdText
         {

src/Network/Network/LoadBalancer/NewAzLoadBalancerBackendAddressConfig.cs

@@ -15,19 +15,19 @@ Returns a load balancer backend address config.
 ### SetByIpAndSubnet (Default)
 ```
 New-AzLoadBalancerBackendAddressConfig -IpAddress <String> -Name <String> -SubnetId <String>
- [-DefaultProfile <IAzureContextContainer>] [-WhatIf] [-Confirm] [<CommonParameters>]
+ [-AdminState <String>] [-DefaultProfile <IAzureContextContainer>] [-WhatIf] [-Confirm] [<CommonParameters>]
 ```
 
 ### SetByIpAndVnet
 ```
 New-AzLoadBalancerBackendAddressConfig -IpAddress <String> -Name <String> -VirtualNetworkId <String>
- [-DefaultProfile <IAzureContextContainer>] [-WhatIf] [-Confirm] [<CommonParameters>]
+ [-AdminState <String>] [-DefaultProfile <IAzureContextContainer>] [-WhatIf] [-Confirm] [<CommonParameters>]
 ```
 
 ### SetByResourceFrontendIPConfiguration
 ```
 New-AzLoadBalancerBackendAddressConfig -Name <String> -LoadBalancerFrontendIPConfigurationId <String>
- [-DefaultProfile <IAzureContextContainer>] [-WhatIf] [-Confirm] [<CommonParameters>]
+ [-AdminState <String>] [-DefaultProfile <IAzureContextContainer>] [-WhatIf] [-Confirm] [<CommonParameters>]
 ```
 
 ## DESCRIPTION
@@ -56,6 +56,21 @@ New-AzLoadBalancerBackendAddressConfig -LoadBalancerFrontendIPConfigurationId $f
 
 ## PARAMETERS
 
+### -AdminState
+The admin state associated with the Backend Address config
+
+```yaml
+Type: System.String
+Parameter Sets: (All)
+Aliases:
+
+Required: False
+Position: Named
+Default value: None
+Accept pipeline input: True (ByPropertyName)
+Accept wildcard characters: False
+```
+
 ### -DefaultProfile
 The credentials, account, tenant, and subscription used for communication with Azure.