From 09860d95d2ddb39e9709a66108c94cd6949ff46b Mon Sep 17 00:00:00 2001
From: Evgeniy Dmitriev <edmitriev@wallarm.com>
Date: Mon, 25 Dec 2023 18:04:26 +0700
Subject: [PATCH] Add WALLARM_API_HOST in e2e tests

---
 .github/workflows/ci.yaml  | 2 --
 test/e2e/run-e2e-suite.sh  | 1 +
 test/e2e/run-kind-e2e.sh   | 5 +++++
 test/e2e/wait-for-nginx.sh | 2 ++
 4 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index f8ecb4e37a..3ebedb77f8 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -309,8 +309,6 @@ jobs:
           WALLARM_ENABLED: true
           WALLARM_API_TOKEN: ${{ steps.secrets.outputs.api_token }}
           WALLARM_API_HOST: audit.api.wallarm.com # TODO: tmp
-          WALLARM_API_PRESET: audit # TODO: tmp
-          CLIENT_ID: "55146" # TODO: tmp
         run: |
           kind get kubeconfig > $HOME/.kube/kind-config-kind
           make E2E_NODES=6 kind-e2e-test
diff --git a/test/e2e/run-e2e-suite.sh b/test/e2e/run-e2e-suite.sh
index aaa2dc7fd8..4b1b6f5ad5 100755
--- a/test/e2e/run-e2e-suite.sh
+++ b/test/e2e/run-e2e-suite.sh
@@ -83,6 +83,7 @@ kubectl run --rm \
   --env="NGINX_BASE_IMAGE=${NGINX_BASE_IMAGE}" \
   --env="WALLARM_ENABLED=${WALLARM_ENABLED:-false}" \
   --env="WALLARM_API_TOKEN=${WALLARM_API_TOKEN:-}" \
+  --env="WALLARM_API_HOST=${WALLARM_API_HOST:-}" \
   --env="HTTPBUN_IMAGE=${HTTPBUN_IMAGE}" \
   --overrides='{ "apiVersion": "v1", "spec":{"serviceAccountName": "ingress-nginx-e2e"}}' \
   e2e --image=nginx-ingress-controller:e2e
diff --git a/test/e2e/run-kind-e2e.sh b/test/e2e/run-kind-e2e.sh
index 0502455d3b..4deae9bec9 100755
--- a/test/e2e/run-kind-e2e.sh
+++ b/test/e2e/run-kind-e2e.sh
@@ -55,6 +55,7 @@ export DOCKER_CLI_EXPERIMENTAL=enabled
 export KUBECONFIG="${KUBECONFIG:-$HOME/.kube/kind-config-$KIND_CLUSTER_NAME}"
 export WALLARM_ENABLED="${WALLARM_ENABLED:-false}"
 export WALLARM_API_TOKEN="${WALLARM_API_TOKEN:-}"
+export WALLARM_API_HOST="${WALLARM_API_HOST:-api.wallarm.com}"
 SKIP_INGRESS_IMAGE_CREATION="${SKIP_INGRESS_IMAGE_CREATION:-false}"
 SKIP_E2E_IMAGE_CREATION="${SKIP_E2E_IMAGE_CREATION:=false}"
 SKIP_CLUSTER_CREATION="${SKIP_CLUSTER_CREATION:-false}"
@@ -129,6 +130,10 @@ if [ "${WALLARM_ENABLED}" == "true" ]; then
     echo "WALLARM_API_TOKEN must be set! Exiting ..."
     exit 1
   fi
+    if [ -z "${WALLARM_API_HOST}" ]; then
+      echo "WALLARM_API_HOST must be set! Exiting ..."
+      exit 1
+    fi
 fi
 
 echo "[dev-env] running e2e tests..."
diff --git a/test/e2e/wait-for-nginx.sh b/test/e2e/wait-for-nginx.sh
index e984010c5b..7318e8c3e7 100755
--- a/test/e2e/wait-for-nginx.sh
+++ b/test/e2e/wait-for-nginx.sh
@@ -67,6 +67,7 @@ if [[ ! -z "$NAMESPACE_OVERLAY" && -d "$DIR/namespace-overlays/$NAMESPACE_OVERLA
         --set controller.image.chroot="${IS_CHROOT}" \
         --set controller.wallarm.enabled="${WALLARM_ENABLED}" \
         --set controller.wallarm.token="${WALLARM_API_TOKEN}" \
+        --set controller.wallarm.apiHost="${WALLARM_API_HOST}" \
         --set controller.wallarm.fallback="off"
 else
     cat << EOF | helm install nginx-ingress ${DIR}/charts/ingress-nginx --namespace=$NAMESPACE --values -
@@ -77,6 +78,7 @@ controller:
   wallarm:
     enabled: ${WALLARM_ENABLED}
     token: ${WALLARM_API_TOKEN}
+    apiHost: ${WALLARM_API_HOST}
     fallback: "off"
   image:
     repository: wallarm/ingress-controller