Define AZURE_LOCATION_LOAD for load test regions

Author: nojnhuh

hack/util.sh

@@ -76,6 +76,11 @@ capz::util::get_random_region_edgezone() {
     local REGIONS=("canadacentral")
     echo "${REGIONS[${RANDOM} % ${#REGIONS[@]}]}"
 }
+# all regions below must have sufficient quota to run load tests
+capz::util::get_random_region_load() {
+    local REGIONS=("canadacentral" "francecentral" "northeurope")
+    echo "${REGIONS[${RANDOM} % ${#REGIONS[@]}]}"
+}
 
 capz::util::generate_ssh_key() {
     # Generate SSH key.

scripts/ci-conformance.sh

@@ -68,6 +68,7 @@ export GINKGO_NODES=1
 export AZURE_LOCATION="${AZURE_LOCATION:-$(capz::util::get_random_region)}"
 export AZURE_LOCATION_GPU="${AZURE_LOCATION_GPU:-$(capz::util::get_random_region_gpu)}"
 export AZURE_LOCATION_EDGEZONE="${AZURE_LOCATION_EDGEZONE:-$(capz::util::get_random_region_edgezone)}"
+export AZURE_LOCATION_LOAD="${AZURE_LOCATION_LOAD:-$(capz::util::get_random_region_load)}"
 export AZURE_CONTROL_PLANE_MACHINE_TYPE="${AZURE_CONTROL_PLANE_MACHINE_TYPE:-"Standard_B2s"}"
 export AZURE_NODE_MACHINE_TYPE="${AZURE_NODE_MACHINE_TYPE:-"Standard_B2s"}"
 export WINDOWS="${WINDOWS:-false}"

scripts/ci-entrypoint.sh

@@ -100,6 +100,8 @@ setup() {
     echo "Using AZURE_LOCATION_GPU: ${AZURE_LOCATION_GPU}"
     export AZURE_LOCATION_EDGEZONE="${AZURE_LOCATION_EDGEZONE:-$(capz::util::get_random_region_edgezone)}"
     echo "Using AZURE_LOCATION_EDGEZONE: ${AZURE_LOCATION_EDGEZONE}"
+    export AZURE_LOCATION_LOAD="${AZURE_LOCATION_LOAD:-$(capz::util::get_random_region_load)}"
+    echo "Using AZURE_LOCATION_LOAD: ${AZURE_LOCATION_LOAD}"
     # Need a cluster with at least 2 nodes
     export CONTROL_PLANE_MACHINE_COUNT="${CONTROL_PLANE_MACHINE_COUNT:-1}"
     export CCM_COUNT="${CCM_COUNT:-1}"

templates/test/dev/cluster-template-custom-builds-load-dra.yaml

@@ -5,6 +5,7 @@ resources:
 - ../custom-builds
 - storageclass-resource-set.yaml
 patches:
+- path: patches/location.yaml
 - path: patches/cluster-label-storageclass.yaml
 - path: patches/cluster-label-azuredisk.yaml
 - path: patches/kcp-scheduler.yaml

templates/test/dev/cluster-template-custom-builds-load.yaml

@@ -0,0 +1,6 @@
+apiVersion: infrastructure.cluster.x-k8s.io/v1beta1
+kind: AzureCluster
+metadata:
+  name: ${CLUSTER_NAME}
+spec:
+  location: ${AZURE_LOCATION_LOAD}