From 44bd166b3d9d39d8f1bcd987023c7b8de797e81a Mon Sep 17 00:00:00 2001
From: Jakub Guzik <jguzik@redhat.com>
Date: Wed, 6 Jul 2022 20:26:31 +0200
Subject: test-apiv2: streamed response testing based on response duration

Signed-off-by: Jakub Guzik <jguzik@redhat.com>
---
 test/apiv2/test-apiv2 | 26 +++++++++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

(limited to 'test/apiv2/test-apiv2')

diff --git a/test/apiv2/test-apiv2 b/test/apiv2/test-apiv2
index 8548d84e5..0fd282854 100755
--- a/test/apiv2/test-apiv2
+++ b/test/apiv2/test-apiv2
@@ -56,6 +56,9 @@ fi
 # Path to podman binary
 PODMAN_BIN=${PODMAN:-${CONTAINERS_HELPER_BINARY_DIR}/podman}
 
+# Timeout for streamed responses
+CURL_TIMEOUT=0
+
 # Cleanup handlers
 clean_up_server() {
     if [ -n "$service_pid" ]; then
@@ -216,6 +219,21 @@ function jsonify() {
     (IFS=','; echo "{${settings_out[*]}}")
 }
 
+#######
+#  t_timeout  #  Timeout wrapper for test helper
+#######
+function t_timeout() {
+    CURL_TIMEOUT=$1; shift
+    local min_runtime=$((CURL_TIMEOUT - 1))
+    start=`date +%s`
+    t $@
+    local end=`date +%s`
+    local runtime=$((end-start))
+    if ! [[ "$runtime" -ge "$min_runtime" ]]; then
+        die "Error: Streaming time should be greater or equal to '$min_runtime'"
+    fi
+}
+
 #######
 #  t  #  Main test helper
 #######
@@ -226,6 +244,12 @@ function t() {
     local content_type="application/json"
 
     local testname="$method $path"
+
+    if [[ $CURL_TIMEOUT != 0 ]]; then
+        local c_timeout=$CURL_TIMEOUT
+        curl_args+=("-m $CURL_TIMEOUT")
+        CURL_TIMEOUT=0 # 'consume' timeout
+    fi
     # POST and PUT requests may be followed by one or more key=value pairs.
     # Slurp the command line until we see a 3-digit status code.
     if [[ $method = "POST" || $method == "PUT" ]]; then
@@ -291,7 +315,7 @@ function t() {
                     -o $WORKDIR/curl.result.out "$url"); rc=$?; } || :
 
     # Any error from curl is instant bad news, from which we can't recover
-    if [[ $rc -ne 0 ]]; then
+    if [[ $rc -ne 0 ]] && [[ $c_timeout -eq 0 ]]; then
         die "curl failure ($rc) on $url - cannot continue"
     fi
 
-- 
cgit v1.2.3-54-g00ecf