[AKS] support EdgeZone (#20576)

Co-authored-by: Jin Lei <[email protected]>

Author: YanaXu

src/Aks/Aks.Test/ScenarioTests/KubernetesTests.cs

@@ -170,5 +170,12 @@ public void TestEnableUptimeSLA()
         {
             TestRunner.RunTestScript("Test-EnableUptimeSLA");
         }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestEdgeZone()
+        {
+            TestRunner.RunTestScript("Test-EdgeZone");
+        }
     }
 }

src/Aks/Aks.Test/ScenarioTests/KubernetesTests.ps1

@@ -934,7 +934,6 @@ function Test-GpuInstanceProfile {
     }
 }
 
-
 function Test-EnableUptimeSLA {
     # Setup
     $resourceGroupName = Get-RandomResourceGroupName
@@ -950,7 +949,7 @@ function Test-EnableUptimeSLA {
         Assert-AreEqual "Basic" $cluster.Sku.Name
         Assert-AreEqual "Paid" $cluster.Sku.Tier
 
-        # create a 2nd nodepool
+        # update the aks cluster
         Set-AzAksCluster -ResourceGroupName $resourceGroupName -Name $kubeClusterName -EnableUptimeSLA:$false
         $cluster = Get-AzAksCluster -ResourceGroupName $resourceGroupName -Name $kubeClusterName
         Assert-AreEqual "Basic" $cluster.Sku.Name
@@ -966,4 +965,27 @@ function Test-EnableUptimeSLA {
     finally {
         Remove-AzResourceGroup -Name $resourceGroupName -Force
     }
-}
+}
+
+function Test-EdgeZone {
+    # Setup
+    $resourceGroupName = Get-RandomResourceGroupName
+    $kubeClusterName = Get-RandomClusterName
+    $location = 'eastus2euap'
+
+    try {
+        New-AzResourceGroup -Name $resourceGroupName -Location $location
+
+        # create aks cluster with default nodepool
+        New-AzAksCluster -ResourceGroupName $resourceGroupName -Name $kubeClusterName -NodeCount 1 -EdgeZone 'microsoftrrdclab1'
+        $cluster = Get-AzAksCluster -ResourceGroupName $resourceGroupName -Name $kubeClusterName
+        Assert-AreEqual "microsoftrrdclab1" $cluster.ExtendedLocation.Name
+        Assert-AreEqual "edgezone" $cluster.ExtendedLocation.Type
+
+        $cluster | Remove-AzAksCluster -Force
+    }
+    finally {
+        Remove-AzResourceGroup -Name $resourceGroupName -Force
+    }
+}
+