Add configureable timeout to test-go.sh

justaugustus · Sep 18, 2020 · 6e427eb · 6e427eb
1 parent c3e0263
commit 6e427eb
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/hack/test-go.sh b/hack/test-go.sh
@@ -16,8 +16,25 @@
 
 set -euo pipefail
 
+# Default timeout is 900s
+TEST_TIMEOUT=900
+
+for arg in "$@"
+do
+    case $arg in
+        -t=*|--timeout=*)
+        TEST_TIMEOUT="${arg#*=}"
+        shift
+        ;;
+        -t|--timeout)
+        TEST_TIMEOUT="$2"
+        shift
+        shift
+    esac
+done
+
 REPO_ROOT=$(git rev-parse --show-toplevel)
 cd "${REPO_ROOT}"
 
-GO111MODULE=on go test -v -count=1 -cover -coverprofile coverage.out ./...
+GO111MODULE=on go test -v -timeout="${TEST_TIMEOUT}s" -count=1 -cover -coverprofile coverage.out ./...
 go tool cover -html coverage.out -o coverage.html