Updated docker script for published test to detect errors

Author: Chris Park

examples/docker/run_python.sh

@@ -1,5 +1,10 @@
 #!/bin/bash
 
+retcode=0
+ping_url="https://api.rosette.com/rest/v1"
+
+#------------------ Functions ----------------------------------------------------
+
 #Gets called when the user doesn't provide any args
 function HELP {
     echo -e "\nusage: source_file.py --key API_KEY [--url ALT_URL]"
@@ -10,6 +15,59 @@ function HELP {
     exit 1
 }
 
+#Checks if Rosette API key is valid
+function checkAPI() {
+    match=$(curl "${ping_url}/ping" -H "X-RosetteAPI-Key: ${API_KEY}" |  grep -o "forbidden")
+    if [ ! -z $match ]; then
+        echo -e "\nInvalid Rosette API Key"
+        exit 1
+    fi  
+}
+
+function cleanURL() {
+    # strip the trailing slash off of the alt_url if necessary
+    if [ ! -z "${ALT_URL}" ]; then
+        case ${ALT_URL} in
+            */) ALT_URL=${ALT_URL::-1}
+                echo "Slash detected"
+                ;;
+        esac
+        ping_url=${ALT_URL}
+    fi
+}
+
+function validateURL() {
+    match=$(curl "${ping_url}/ping" -H "X-RosetteAPI-Key: ${API_KEY}" -H "user_key: ${API_KEY}" |  grep -o "Rosette API")
+    if [ "${match}" = "" ]; then
+        echo -e "\n${ping_url} server not responding\n"
+        exit 1
+    fi  
+}
+
+function runExample() {
+    echo -e "\n---------- ${1} start -------------"
+    result=""
+    if [ -z ${ALT_URL} ]; then
+        result="$(python ${1} --key ${API_KEY} 2>&1 )"
+    else
+        result="$(python ${1} --key ${API_KEY} --url ${ALT_URL} 2>&1 )"
+    fi
+    echo "${result}"
+    echo -e "\n---------- ${1} end -------------"
+    if [[ "${result}" == *"Exception"* ]]; then
+        echo "Exception found"
+        retcode=1
+    elif [[ "$result" == *"processingFailure"* ]]; then
+        retcode=1
+    elif [[ "$result" == *"AttributeError"* ]]; then
+        retcode=1
+    elif [[ "$result" == *"ImportError"* ]]; then
+        retcode=1
+    fi
+}
+
+#------------------ Functions End ------------------------------------------------
+
 #Gets API_KEY, FILENAME and ALT_URL if present
 while getopts ":API_KEY:FILENAME:ALT_URL" arg; do
     case "${arg}" in
@@ -27,33 +85,10 @@ while getopts ":API_KEY:FILENAME:ALT_URL" arg; do
             ;;
     esac
 done
-ping_url="https://api.rosette.com/rest/v1"
-
-# strip the trailing slash off of the alt_url if necessary
-if [ ! -z "${ALT_URL}" ]; then
-    case ${ALT_URL} in
-        */) ALT_URL=${ALT_URL::-1}
-            echo "Slash detected"
-            ;;
-    esac
-    ping_url=${ALT_URL}
-fi
 
-#Checks for valid url
-match=$(curl "${ping_url}/ping" -H "X-RosetteAPI-Key: ${API_KEY}" |  grep -o "Rosette API")
-if [ "${match}" = "" ]; then
-    echo -e "\n${ping_url} server not responding\n"
-    exit 1
-fi  
+cleanURL
 
-#Checks if Rosette API key is valid
-function checkAPI {
-    match=$(curl "${ping_url}/ping" -H "X-RosetteAPI-Key: ${API_KEY}" |  grep -o "forbidden")
-    if [ ! -z $match ]; then
-        echo -e "\nInvalid Rosette API Key"
-        exit 1
-    fi  
-}
+validateURL
 
 #Copy the examples from the mounted content in /source to current WORKDIR
 cp  /source/examples/*.* .
@@ -62,16 +97,14 @@ cp  /source/examples/*.* .
 if [ ! -z ${API_KEY} ]; then
     checkAPI
     if [ ! -z ${FILENAME} ]; then
-        if [ ! -z ${ALT_URL} ]; then
-	    python ${FILENAME} --key ${API_KEY} --url ${ALT_URL} 
-	else
-	    python ${FILENAME} --key ${API_KEY} 
-   	fi
-    elif [ ! -z ${ALT_URL} ]; then
-    	find -maxdepth 1  -name '*.py' -print -exec python {} --key ${API_KEY} --url ${ALT_URL} \;
+        runExample ${FILENAME}
     else
-	find -maxdepth 1  -name '*.py' -print -exec python {} --key ${API_KEY} \;
-    fi
+        for file in *.py; do
+            runExample ${file}
+        done
+   	fi
 else 
     HELP
 fi
+
+exit ${retcode}