Update display style and add flags

Author: subeshb1

api-test.sh

@@ -4,7 +4,7 @@ set -o pipefail
 RED=$(tput setaf 1)
 GREEN=$(tput setaf 2)
 BOLD=$(tput bold)
-RESET=$(tput sgr0)
+RESET=$(tput sgr 0)
 
 ACTION=""
 
@@ -18,21 +18,34 @@ ACCESS_TOKEN=""
 ID_TOKEN=""
 URL=""
 
+SHOW_HEADER=0
+HEADER_ONLY=0
+SILENT=0
+
 echo_v() {
   if [ $VERBOSE -eq 1 ]; then
     echo $1
   fi
 }
 
 run() {
-  cat $FILE | jq empty
-  if [ $? -ne 0 ]; then
-    exit
-  fi
-  URL=$(jq -r '.url' $FILE)
-  ACCESS_TOKEN=$(jq -r '.accessToken' $FILE)
-  ID_TOKEN=$(jq -r '.idToken' $FILE)
-  COMMON_HEADER=$(cat $FILE | jq -r -c ". | .header | if  . != null then . else {} end   | to_entries | map(\"\(.key): \(.value|tostring)\") | join(\"\n\") | if ( . | length) != 0 then \"-H\" + .  else \"-H \" end")
+  for arg in "$@"; do
+    case $arg in
+    -i | --include)
+      SHOW_HEADER=1
+      shift
+      ;;
+    -I | --header-only)
+      HEADER_ONLY=1
+      shift
+      ;;
+    -s | --silent)
+      SILENT=1
+      shift
+      ;;
+    esac
+  done
+
   case $1 in
   all)
     api_factory "$(jq -r '.testCases | keys[]' $FILE)"
@@ -47,18 +60,49 @@ api_factory() {
   for TEST_CASE in $@; do
     echo "${BOLD}Running Case:${RESET} $TEST_CASE"
     echo_v "${BOLD}Description: ${RESET}$(jq -r ".testCases.$TEST_CASE.description" $FILE)"
-    echo_v
     ROUTE=$(jq -r ".testCases.$TEST_CASE.path" $FILE)
     BODY="$(jq -r ".testCases.$TEST_CASE.body" $FILE)"
     QUERY_PARAMS=$(cat $FILE | jq -r ".testCases.$TEST_CASE | select(.query != null) | .query  | to_entries | map(\"\(.key)=\(.value|tostring)\") | join(\"&\") | \"?\" + . ")
     REQUEST_HEADER=$(cat $FILE | jq -r ".testCases.$TEST_CASE | .header | if  . != null then . else {} end   | to_entries | map(\"\(.key): \(.value|tostring)\") | join(\"\n\") | if ( . | length) != 0 then \"-H\" + .  else \"-H \" end")
     METHOD="$(jq -r ".testCases.$TEST_CASE.method //\"GET\" | ascii_upcase" $FILE)"
     call_api
-    echo ""
-    echo ""
+    display_results
   done
 }
 
+display_results() {
+  local res=$(jq -r '.http_status + " " + .http_message ' <<<"$HEADER")
+  local status=$(jq -r '.http_status' <<<"$HEADER")
+  echo "${BOLD}$(color_response $status)$res${RESET}"
+  if [[ $HEADER_ONLY == 1 ]]; then
+    echo "HEADER:"
+    echo "$HEADER" | jq -C
+  else
+    if [[ $SHOW_HEADER == 1 ]]; then
+      echo "HEADER:"
+      echo "$HEADER" | jq -C
+    fi
+    if [[ $SILENT == 0 ]]; then
+      echo "BODY:"
+      echo "$BODY" | jq -C
+    fi
+
+  fi
+
+  echo "META:"
+  echo "$META" | jq -C
+  echo ""
+  echo ""
+}
+
+color_response() {
+  case $1 in
+  2[0-9][0-9]) echo $GREEN ;;
+  [45][0-9][0-9]) echo $RED ;;
+  *) ;;
+  esac
+}
+
 call_api() {
   # curl -ivs --request $METHOD "$URL$ROUTE$QUERY_PARAMS" \
   #   --data "$BODY" \
@@ -80,12 +124,6 @@ call_api() {
   BODY=$(sed -n 1p <<<"$json")
   META=$(sed 1d <<<"$json")
   parse_header "$header"
-  echo "HEADER:"
-  echo "$HEADER" | jq -C
-  echo "BODY:"
-  echo "$BODY" | jq -C
-  echo "META:"
-  echo "$META" | jq -C
 }
 
 function parse_header() {
@@ -144,6 +182,16 @@ if [ ! -f "$FILE" ]; then
   exit 1
 fi
 
+cat $FILE | jq empty
+if [ $? -ne 0 ]; then
+  echo "Empty file"
+  exit
+fi
+URL=$(jq -r '.url' $FILE)
+ACCESS_TOKEN=$(jq -r '.accessToken' $FILE)
+ID_TOKEN=$(jq -r '.idToken' $FILE)
+COMMON_HEADER=$(cat $FILE | jq -r -c ". | .header | if  . != null then . else {} end   | to_entries | map(\"\(.key): \(.value|tostring)\") | join(\"\n\") | if ( . | length) != 0 then \"-H\" + .  else \"-H \" end")
+
 case $ACTION in
 run)
   run $@