use standard locust env vars

locust/docker-entrypoint.sh

@@ -2,40 +2,16 @@
 
 set -e
 
-LOCUST_MODE=${LOCUST_MODE:-standalone}
-LOCUST_MASTER_BIND_PORT=${LOCUST_MASTER_BIND_PORT:-5557}
-LOCUST_FILE=${LOCUST_FILE:-locustfile.py}
-
 if [ "x" != "${LOCUST_S3_PATH}x" ] ; then
-    aws s3 cp s3://${LOCUST_S3_PATH}/${LOCUST_FILE} . --region ${AWS_REGION}
-    chmod +x ${LOCUST_FILE}
+    aws s3 cp s3://${LOCUST_S3_PATH}/${LOCUST_LOCUSTFILE} . --region ${AWS_REGION}
+    chmod +x ${LOCUST_LOCUSTFILE}
 fi
 
 if [ ${1} = "locust" ] ; then 
     echo "Starting locust......."
-    if [ -z ${TEST_HOST_URL+x} ] ; then
-        echo "You need to set the URL of the host to be tested (TEST_HOST_URL)."
+    if [ -z ${LOCUST_HOST+x} ] ; then
+        echo "You need to set the URL of the host to be tested (LOCUST_HOST)."
         exit 1
     fi
-
-    LOCUST_OPTS="--host=${TEST_HOST_URL} --no-reset-stats $LOCUST_OPTS"
-
-    case `echo ${LOCUST_MODE} | tr 'a-z' 'A-Z'` in
-    "MASTER")
-        echo "Setting master node config"
-        LOCUST_OPTS="-f ${LOCUST_FILE} --master --master-bind-port=${LOCUST_MASTER_BIND_PORT} $LOCUST_OPTS"
-        ;;
-
-    "WORKER")
-        echo "Setting worker node config"
-        LOCUST_OPTS="-f - --worker --master-host=${LOCUST_MASTER} --master-port=${LOCUST_MASTER_BIND_PORT} --processes -1 $LOCUST_OPTS"
-        if [ -z ${LOCUST_MASTER+x} ] ; then
-            echo "You need to set LOCUST_MASTER."
-            exit 1
-        fi
-        ;;
-    esac
-    exec "locust ${LOCUST_OPTS}"
-else
-    exec "$@"
-fi
+fi
+exec "$@"