load_test.sh

#!/bin/bash -e
# Runs a load test against a deployed RDR. See load_test_locustfile for instructions.

if ! which locust
then
  echo "Install Locust with \"sudo pip install locustio\"."
  exit 1
fi
function usage {
  echo "Usage: load_test.sh --project all-of-us-rdr-test --account $USER@pmi-ops.org"
  exit 1
}
while true
do
  case "$1" in
    --project) PROJECT=$2; shift 2;;
    --account) ACCOUNT=$2; shift 2;;
    *) break;;
  esac
done
if [ "$@" ]
then
  echo "Unrecognized arguments: $@"
  usage
fi
if [ -z "$ACCOUNT" ]
then
  usage
fi
if [ -z "$PROJECT" ]
then
  usage
fi
if [[ "$PROJECT" =~ "prod" ]]
then
  echo "Forbidden to load test against production instance $PROJECT."
  exit 1
fi

CREDS_ACCOUNT=$ACCOUNT
. tools/auth_setup.sh

# Some dependency data is loaded via DAOs before load testing via API.
run_cloud_sql_proxy
set_db_connection_string

# Prefer system-wide installs of some packages (specifically "requests"). The
# gcloud libraries added by set_path.sh include an older version.
OLDPATH=$PYTHONPATH
. tools/set_path.sh;

GAE_SDK_ROOT="${sdk_dir}/platform/google_appengine"

# The next line enables Python libraries for Google Cloud SDK
GAEPATH=${GAE_SDK_ROOT}

# * OPTIONAL STEP *
# If you wish to import all Python modules, you may iterate in the directory
# tree and import each module.
#
# * WARNING *
# Some modules have two or more versions available (Ex. django), so the loop
# will import always its latest version.
for module in ${GAE_SDK_ROOT}/lib/*; do
  if [ -r ${module} ];
  then
    GAEPATH=${module}:${GAEPATH}
  fi
done
export PYTHONPATH=/usr/local/lib/python2.7/dist-packages:$OLDPATH:$PYTHONPATH:${GAEPATH}

LOCUST_CREDS_FILE="$CREDS_FILE" \
LOCUST_TARGET_INSTANCE="https://$PROJECT.appspot.com" \
locust -f "$BASE_DIR"/tools/load_test_locustfile.py