AzSKPreview

4.0.0

Framework/Core/SVT/Services/LoadBalancer.ps1

Set-StrictMode -Version Latest 
class LoadBalancer: AzSVTBase
{       
    hidden [PSObject] $ResourceObject;

    LoadBalancer([string] $subscriptionId, [SVTResource] $svtResource): 
        Base($subscriptionId, $svtResource) 
    { 
    }

    hidden [ControlResult] CheckPublicIP([ControlResult] $controlResult)
    {    
        $publicIps = @();
        $loadBalancer = Get-AzLoadBalancer -ResourceGroupName $this.ResourceContext.ResourceGroupName -Name $this.ResourceContext.ResourceName
        if([Helpers]::CheckMember($loadBalancer,"FrontendIpConfigurations"))
        {
            $loadBalancer.FrontendIpConfigurations | 
                ForEach-Object {
                    Set-Variable -Name feIpConfigurations -Scope Local -Value $_
                    if(($feIpConfigurations | Get-Member -Name "PublicIpAddress") -and $feIpConfigurations.PublicIpAddress)
                    {
                        $ipResource = Get-AzResource -ResourceId $feIpConfigurations.PublicIpAddress.Id 
                        if($ipResource)
                        {
                           $publicIpObject = Get-AzPublicIpAddress -Name $ipResource.Name -ResourceGroupName $ipResource.ResourceGroupName
                           if($publicIpObject)
                           {
                                $_.PublicIpAddress = $publicIpObject;
                                $publicIps += $publicIpObject;
                            }
                        }
                    }
                }
         }
        if($publicIps.Count -gt 0)
        {              
            $controlResult.AddMessage([VerificationResult]::Verify, "Validate Public IP(s) associated with Load Balancer. Total - $($publicIps.Count)", $publicIps);  
            $controlResult.SetStateData("Public IP(s) associated with Load Balancer", $publicIps);
        }
        else
        {
            $controlResult.AddMessage([VerificationResult]::Passed, "No Public IP is associated with Load Balancer");
        }
        return $controlResult;
    }
}