Add a validation mode (#27)

Fixes #26

Author: nitisht

README.md

@@ -28,12 +28,11 @@ Positional arguments for the 'smoke' or 'load' mode:
 4. Password 
 ```
 
-
 Additional positional arguments for the 'load' mode
 ```
-5. (Optional) Number of different json formats to send to a stream
-6. (Optional) Schema count
-7. (Optional) Duration
+5. (Optional) Number of different json schemas to send to a stream
+6. (Optional) Number of virtual users (Refer K6 [documentation on VUs](https://k6.io/docs/get-started/running-k6/#adding-more-vus))
+7. (Optional) Duration of the test
 ```
 
 Example usage:

main.sh

@@ -39,9 +39,16 @@ run_load_test () {
   return $?
 }
 
+run_validation_test () {
+  ./testcases/validate_test.sh "$endpoint" "$stream_name" "$username" "$password" "$schema_count" "$vus" "$duration"
+  return $?
+}
+
 case "$mode" in
    "smoke") run_smoke_test 
    ;;
    "load") run_load_test 
    ;;
+   "validate") run_validation_test 
+   ;;
 esac

testcases/validate_test.sh

@@ -0,0 +1,66 @@
+#!/bin/sh
+#
+# Parseable Server (C) 2023 Cloudnatively Pvt. Ltd.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as
+# published by the Free Software Foundation, either version 3 of the
+# License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+
+parseable_url=$1
+stream_name="backend"
+username=$3
+password=$4
+count=$5
+vus=$6
+duration=$7
+
+curl_std_opts=( -sS --header 'Content-Type: application/json' -w '\n\n%{http_code}' -u "$username":"$password" )
+
+# Create stream
+create_stream () {
+  response=$(curl "${curl_std_opts[@]}" --request PUT "$parseable_url"/api/v1/logstream/"$stream_name")
+
+  if [ $? -ne 0 ]; then
+    printf "Failed to create log stream %s with exit code: %s\n" "$stream_name" "$?"
+    printf "Test create_stream: failed\n"
+    exit 1
+  fi
+
+  http_code=$(tail -n1 <<< "$response")
+  if [ "$http_code" -ne 200 ]; then
+    printf "Failed to create log stream %s with http code: %s and response: %s\n" "$stream_name" "$http_code" "$content"
+    printf "Test create_stream: failed\n"
+    exit 1
+  fi
+
+  content=$(sed '$ d' <<< "$response")
+  if [ "$content" != "log stream created" ]; then
+    printf "Failed to create log stream $stream_name with response: %s\n" "$content"
+    printf "Test create_stream: failed\n"
+    exit 1
+  fi
+
+  printf "Test create_stream: successful\n"
+  return 0
+}
+
+run_k6() {
+  k6 run -e P_URL="$parseable_url" -e P_STREAM="$stream_name" -e P_USERNAME="$username" -e P_PASSWORD="$password" -e P_SCHEMA_COUNT="$count" "/tests/testcases/smoke.js" --vus="$vus" --duration="$duration"
+}
+
+printf "======= Starting validation tests with k6 =======\n"
+printf "** Log stream name: %s **\n" "$stream_name"
+printf "====================================\n"
+create_stream
+run_k6
+printf "======= Load tests completed ======\n"