Merge pull request #5 from ZianWang02/feature/Cdn

Add Get-AzCdnEndpointResourceUsage test
Azure · Mar 23, 2022 · 8619c6d · 8619c6d
2 parents a7e97b6 + c5107d4
commit 8619c6d
Showing 1 changed file with 53 additions and 2 deletions.
diff --git a/src/Cdn/Cdn.Autorest/test/Get-AzCdnEndpointResourceUsage.Tests.ps1 b/src/Cdn/Cdn.Autorest/test/Get-AzCdnEndpointResourceUsage.Tests.ps1
@@ -15,7 +15,58 @@ if(($null -eq $TestName) -or ($TestName -contains 'Get-AzCdnEndpointResourceUsag
 }
 
 Describe 'Get-AzCdnEndpointResourceUsage' {
-    It 'List' -skip {
-        { throw [System.NotImplementedException] } | Should -Not -Throw
+    It 'List' {
+        { 
+            $ResourceGroupName = 'testps-rg-' + (RandomString -allChars $false -len 6)
+            try
+            {
+                Write-Host -ForegroundColor Green "Create test group $($ResourceGroupName)"
+                New-AzResourceGroup -Name $ResourceGroupName -Location $env.location
+
+                $cdnProfileName = 'p-' + (RandomString -allChars $false -len 6);
+                Write-Host -ForegroundColor Green "Use cdnProfileName : $($cdnProfileName)"
+
+                $profileSku = "Standard_Akamai";
+                New-AzCdnProfile -SkuName $profileSku -Name $cdnProfileName -ResourceGroupName $ResourceGroupName -Location Global
+
+                $endpointName = 'e-' + (RandomString -allChars $false -len 6);
+                $origin = @{
+                    Name = "origin1"
+                    HostName = "host1.hello.com"
+                };
+                $location = "westus"
+                Write-Host -ForegroundColor Green "Create endpointName : $($endpointName), origin.Name : $($origin.Name), origin.HostName : $($origin.HostName)"
+
+                New-AzCdnEndpoint -Name $endpointName -ResourceGroupName $ResourceGroupName -ProfileName $cdnProfileName -Location $location -Origin $origin
+                $endpointResourceUsages = Get-AzCdnEndpointResourceUsage -EndpointName $endpointName -ProfileName $cdnProfileName -ResourceGroupName $ResourceGroupName
+                $geofilterUsage = $endpointResourceUsages | Where-Object -Property ResourceType -eq 'geofilter'
+
+                $endpointResourceUsages.Count | Should -Be 3
+                $geofilterUsage.Limit | Should -Be 25
+                $geofilterUsage.CurrentValue | Should -Be 0
+
+                $geofilters = @(
+                    @{
+                        RelativePath = "/mycar" 
+                        Action =  [Microsoft.Azure.PowerShell.Cmdlets.Cdn.Support.GeoFilterActions]::Allow
+                        CountryCode = "AU"
+                    },
+                    @{
+                        RelativePath = "/mycars" 
+                        Action =  [Microsoft.Azure.PowerShell.Cmdlets.Cdn.Support.GeoFilterActions]::Allow
+                        CountryCode = "AU"
+                    })
+                Update-AzCdnEndpoint -Name $endpointName -ResourceGroupName $ResourceGroupName -ProfileName $cdnProfileName -GeoFilter $geofilters
+                $endpointResourceUsages = Get-AzCdnEndpointResourceUsage -EndpointName $endpointName -ProfileName $cdnProfileName -ResourceGroupName $ResourceGroupName
+                $geofilterUsage = $endpointResourceUsages | Where-Object -Property ResourceType -eq 'geofilter'
+
+                $endpointResourceUsages.Count | Should -Be 3
+                $geofilterUsage.Limit | Should -Be 25
+                $geofilterUsage.CurrentValue | Should -Be 2
+            } Finally
+            {
+                Remove-AzResourceGroup -Name $ResourceGroupName -NoWait
+            }
+        } | Should -Not -Throw
     }
 }