From 2eec738172cc653fcbe82e77703494ed4f2fb982 Mon Sep 17 00:00:00 2001
From: Xiang Li <lixiang1992@gwu.edu>
Date: Wed, 24 Mar 2021 11:23:21 -0700
Subject: [PATCH] track driver start time in the e2e test

---
 hack/e2e/run.sh | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/hack/e2e/run.sh b/hack/e2e/run.sh
index 75e62e6bda..96e5046787 100755
--- a/hack/e2e/run.sh
+++ b/hack/e2e/run.sh
@@ -25,6 +25,7 @@ source "${BASE_DIR}"/util.sh
 
 DRIVER_NAME=${DRIVER_NAME:-aws-ebs-csi-driver}
 CONTAINER_NAME=${CONTAINER_NAME:-ebs-plugin}
+DRIVER_START_TIME_THRESHOLD=20
 
 TEST_ID=${TEST_ID:-$RANDOM}
 CLUSTER_NAME=test-cluster-${TEST_ID}.k8s.local
@@ -104,11 +105,13 @@ if [[ $? -ne 0 ]]; then
 fi
 
 loudecho "Deploying driver"
+startSec=$(date +'%s')
 "${HELM_BIN}" upgrade --install "${DRIVER_NAME}" \
   --namespace kube-system \
   --set image.repository="${IMAGE_NAME}" \
   --set image.tag="${IMAGE_TAG}" \
   -f "${HELM_VALUES_FILE}" \
+  --wait
   ./charts/"${DRIVER_NAME}"
 
 if [[ -r "${EBS_SNAPSHOT_CRD}" ]]; then
@@ -116,6 +119,15 @@ if [[ -r "${EBS_SNAPSHOT_CRD}" ]]; then
   kubectl apply -f "$EBS_SNAPSHOT_CRD"
   # TODO deploy snapshot controller too instead of including in helm chart
 fi
+endSec=$(date +'%s')
+secondUsed=$(( (endSec-startSec)/1 ))
+# Set timeout threshold as 20 seconds for now, usually it takes less than 10s to startup
+if [ $secondUsed -gt $DRIVER_START_TIME_THRESHOLD ]; then
+  loudecho "Driver start timeout, test fail!"
+  exit 1
+fi
+loudecho "Driver deployment complete, time used: $secondUsed seconds"
+
 
 loudecho "Testing focus ${GINKGO_FOCUS}"
 eval "EXPANDED_TEST_EXTRA_FLAGS=$TEST_EXTRA_FLAGS"