build/run-tests.sh

   1 #!/bin/bash
   2 # Copyright (C) The Arvados Authors. All rights reserved.
   3 #
   4 # SPDX-License-Identifier: AGPL-3.0
   5
   6 COLUMNS=80
   7 . `dirname "$(readlink -f "$0")"`/run-library.sh
   8
   9 read -rd "\000" helpmessage <<EOF
  10 $(basename $0): Install and test Arvados components.
  11
  12 Exit non-zero if any tests fail.
  13
  14 Syntax:
  15         $(basename $0) WORKSPACE=/path/to/arvados [options]
  16
  17 Options:
  18
  19 --skip FOO     Do not test the FOO component.
  20 --skip sanity  Skip initial dev environment sanity checks.
  21 --skip install Do not run any install steps. Just run tests.
  22                You should provide GOPATH, GEMHOME, and VENVDIR options
  23                from a previous invocation if you use this option.
  24 --only FOO     Do not test anything except the FOO component.
  25 --temp DIR     Install components and dependencies under DIR instead of
  26                making a new temporary directory. Implies --leave-temp.
  27 --leave-temp   Do not remove GOPATH, virtualenv, and other temp dirs at exit.
  28                Instead, show the path to give as --temp to reuse them in
  29                subsequent invocations.
  30 --repeat N     Repeat each install/test step until it succeeds N times.
  31 --retry        Prompt to retry if an install or test suite fails.
  32 --only-install Run specific install step
  33 --short        Skip (or scale down) some slow tests.
  34 --interactive  Set up, then prompt for test/install steps to perform.
  35 WORKSPACE=path Arvados source tree to test.
  36 CONFIGSRC=path Dir with config.yml file containing PostgreSQL section for use by tests.
  37 services/api_test="TEST=test/functional/arvados/v1/collections_controller_test.rb"
  38                Restrict apiserver tests to the given file
  39 sdk/python_test="--test-suite tests.test_keep_locator"
  40                Restrict Python SDK tests to the given class
  41 apps/workbench_test="TEST=test/integration/pipeline_instances_test.rb"
  42                Restrict Workbench tests to the given file
  43 services/arv-git-httpd_test="-check.vv"
  44                Show all log messages, even when tests pass (also works
  45                with services/keepstore_test etc.)
  46 ARVADOS_DEBUG=1
  47                Print more debug messages
  48 envvar=value   Set \$envvar to value. Primarily useful for WORKSPACE,
  49                *_test, and other examples shown above.
  50
  51 Assuming "--skip install" is not given, all components are installed
  52 into \$GOPATH, \$VENDIR, and \$GEMHOME before running any tests. Many
  53 test suites depend on other components being installed, and installing
  54 everything tends to be quicker than debugging dependencies.
  55
  56 As a special concession to the current CI server config, CONFIGSRC
  57 defaults to $HOME/arvados-api-server if that directory exists.
  58
  59 More information and background:
  60
  61 https://arvados.org/projects/arvados/wiki/Running_tests
  62
  63 Available tests:
  64
  65 apps/workbench (*)
  66 apps/workbench_units (*)
  67 apps/workbench_functionals (*)
  68 apps/workbench_integration (*)
  69 apps/workbench_benchmark
  70 apps/workbench_profile
  71 cmd/arvados-client
  72 cmd/arvados-server
  73 doc
  74 lib/cli
  75 lib/cmd
  76 lib/controller
  77 lib/controller/federation
  78 lib/controller/railsproxy
  79 lib/controller/router
  80 lib/controller/rpc
  81 lib/crunchstat
  82 lib/crunch-run
  83 lib/cloud
  84 lib/cloud/azure
  85 lib/cloud/cloudtest
  86 lib/dispatchcloud
  87 lib/dispatchcloud/container
  88 lib/dispatchcloud/scheduler
  89 lib/dispatchcloud/sshexecutor
  90 lib/dispatchcloud/worker
  91 lib/mount
  92 lib/pam
  93 lib/service
  94 services/api
  95 services/arv-git-httpd
  96 services/crunchstat
  97 services/dockercleaner
  98 services/fuse
  99 services/fuse:py3
 100 services/health
 101 services/keep-web
 102 services/keepproxy
 103 services/keepstore
 104 services/keep-balance
 105 services/login-sync
 106 services/crunch-dispatch-local
 107 services/crunch-dispatch-slurm
 108 services/ws
 109 sdk/cli
 110 sdk/python
 111 sdk/python:py3
 112 sdk/ruby
 113 sdk/go/arvados
 114 sdk/go/arvadosclient
 115 sdk/go/auth
 116 sdk/go/dispatch
 117 sdk/go/keepclient
 118 sdk/go/health
 119 sdk/go/httpserver
 120 sdk/go/manifest
 121 sdk/go/blockdigest
 122 sdk/go/asyncbuf
 123 sdk/go/stats
 124 sdk/go/crunchrunner
 125 sdk/cwl:py3
 126 sdk/R
 127 sdk/java-v2
 128 tools/sync-groups
 129 tools/crunchstat-summary
 130 tools/crunchstat-summary:py3
 131 tools/keep-exercise
 132 tools/keep-rsync
 133 tools/keep-block-check
 134
 135 (*) apps/workbench is shorthand for apps/workbench_units +
 136     apps/workbench_functionals + apps/workbench_integration
 137
 138 EOF
 139
 140 # First make sure to remove any ARVADOS_ variables from the calling
 141 # environment that could interfere with the tests.
 142 unset $(env | cut -d= -f1 | grep \^ARVADOS_)
 143
 144 # Reset other variables that could affect our [tests'] behavior by
 145 # accident.
 146 GITDIR=
 147 GOPATH=
 148 VENVDIR=
 149 VENV3DIR=
 150 PYTHONPATH=
 151 GEMHOME=
 152 PERLINSTALLBASE=
 153 R_LIBS=
 154 export LANG=en_US.UTF-8
 155
 156 short=
 157 only_install=
 158 temp=
 159 temp_preserve=
 160
 161 clear_temp() {
 162     if [[ -z "$temp" ]]; then
 163         # we did not even get as far as making a temp dir
 164         :
 165     elif [[ -z "$temp_preserve" ]]; then
 166         # Go creates readonly dirs in the module cache, which cause
 167         # "rm -rf" to fail unless we chmod first.
 168         chmod -R u+w "$temp"
 169         rm -rf "$temp"
 170     else
 171         echo "Leaving behind temp dirs in $temp"
 172     fi
 173 }
 174
 175 fatal() {
 176     clear_temp
 177     echo >&2 "Fatal: $* (encountered in ${FUNCNAME[1]} at ${BASH_SOURCE[1]} line ${BASH_LINENO[0]})"
 178     exit 1
 179 }
 180
 181 exit_cleanly() {
 182     trap - INT
 183     if which create-plot-data-from-log.sh >/dev/null; then
 184         create-plot-data-from-log.sh $BUILD_NUMBER "$WORKSPACE/apps/workbench/log/test.log" "$WORKSPACE/apps/workbench/log/"
 185     fi
 186     rotate_logfile "$WORKSPACE/apps/workbench/log/" "test.log"
 187     stop_services
 188     rotate_logfile "$WORKSPACE/services/api/log/" "test.log"
 189     report_outcomes
 190     clear_temp
 191     exit ${#failures}
 192 }
 193
 194 sanity_checks() {
 195     [[ -n "${skip[sanity]}" ]] && return 0
 196     ( [[ -n "$WORKSPACE" ]] && [[ -d "$WORKSPACE/services" ]] ) \
 197         || fatal "WORKSPACE environment variable not set to a source directory (see: $0 --help)"
 198     [[ -z "$CONFIGSRC" ]] || [[ -s "$CONFIGSRC/config.yml" ]] \
 199         || fatal "CONFIGSRC is $CONFIGSRC but '$CONFIGSRC/config.yml' is empty or not found (see: $0 --help)"
 200     echo Checking dependencies:
 201     echo "locale: ${LANG}"
 202     [[ "$(locale charmap)" = "UTF-8" ]] \
 203         || fatal "Locale '${LANG}' is broken/missing. Try: echo ${LANG} | sudo tee -a /etc/locale.gen && sudo locale-gen"
 204     echo -n 'ruby: '
 205     ruby -v \
 206         || fatal "No ruby. Install >=2.1.9 (using rbenv, rvm, or source)"
 207     echo -n 'go: '
 208     go version \
 209         || fatal "No go binary. See http://golang.org/doc/install"
 210     [[ $(go version) =~ go1.([0-9]+) ]] && [[ ${BASH_REMATCH[1]} -ge 12 ]] \
 211         || fatal "Go >= 1.12 required. See http://golang.org/doc/install"
 212     echo -n 'gcc: '
 213     gcc --version | egrep ^gcc \
 214         || fatal "No gcc. Try: apt-get install build-essential"
 215     echo -n 'fuse.h: '
 216     find /usr/include -path '*fuse/fuse.h' | egrep --max-count=1 . \
 217         || fatal "No fuse/fuse.h. Try: apt-get install libfuse-dev"
 218     echo -n 'gnutls.h: '
 219     find /usr/include -path '*gnutls/gnutls.h' | egrep --max-count=1 . \
 220         || fatal "No gnutls/gnutls.h. Try: apt-get install libgnutls28-dev"
 221     echo -n 'virtualenv: '
 222     python3 -m venv -h | egrep --max-count=1 . \
 223         || fatal "No virtualenv. Try: apt-get install python3-venv"
 224     echo -n 'Python3 pyconfig.h: '
 225     find /usr/include -path '*/python3*/pyconfig.h' | egrep --max-count=1 . \
 226         || fatal "No Python3 pyconfig.h. Try: apt-get install python3-dev"
 227     which netstat \
 228         || fatal "No netstat. Try: apt-get install net-tools"
 229     echo -n 'nginx: '
 230     PATH="$PATH:/sbin:/usr/sbin:/usr/local/sbin" nginx -v \
 231         || fatal "No nginx. Try: apt-get install nginx"
 232     echo -n 'perl: '
 233     perl -v | grep version \
 234         || fatal "No perl. Try: apt-get install perl"
 235     for mod in ExtUtils::MakeMaker JSON LWP Net::SSL; do
 236         echo -n "perl $mod: "
 237         perl -e "use $mod; print \"\$$mod::VERSION\\n\"" \
 238             || fatal "No $mod. Try: apt-get install perl-modules libcrypt-ssleay-perl libjson-perl libwww-perl"
 239     done
 240     echo -n 'gitolite: '
 241     which gitolite \
 242         || fatal "No gitolite. Try: apt-get install gitolite3"
 243     echo -n 'npm: '
 244     npm --version \
 245         || fatal "No npm. Try: wget -O- https://nodejs.org/dist/v10.23.1/node-v10.23.1-linux-x64.tar.xz | sudo tar -C /usr/local -xJf - && sudo ln -s ../node-v10.23.1-linux-x64/bin/{node,npm} /usr/local/bin/"
 246     echo -n 'cadaver: '
 247     cadaver --version | grep -w cadaver \
 248           || fatal "No cadaver. Try: apt-get install cadaver"
 249     echo -n 'libattr1 xattr.h: '
 250     find /usr/include -path '*/attr/xattr.h' | egrep --max-count=1 . \
 251         || fatal "No libattr1 xattr.h. Try: apt-get install libattr1-dev"
 252     echo -n 'libcurl curl.h: '
 253     find /usr/include -path '*/curl/curl.h' | egrep --max-count=1 . \
 254         || fatal "No libcurl curl.h. Try: apt-get install libcurl4-gnutls-dev"
 255     echo -n 'libpq libpq-fe.h: '
 256     find /usr/include -path '*/postgresql/libpq-fe.h' | egrep --max-count=1 . \
 257         || fatal "No libpq libpq-fe.h. Try: apt-get install libpq-dev"
 258     echo -n 'libpam pam_appl.h: '
 259     find /usr/include -path '*/security/pam_appl.h' | egrep --max-count=1 . \
 260         || fatal "No libpam pam_appl.h. Try: apt-get install libpam0g-dev"
 261     echo -n 'postgresql: '
 262     psql --version || fatal "No postgresql. Try: apt-get install postgresql postgresql-client-common"
 263     echo -n 'phantomjs: '
 264     phantomjs --version || fatal "No phantomjs. Try: apt-get install phantomjs"
 265     echo -n 'xvfb: '
 266     which Xvfb || fatal "No xvfb. Try: apt-get install xvfb"
 267     echo -n 'graphviz: '
 268     dot -V || fatal "No graphviz. Try: apt-get install graphviz"
 269     echo -n 'geckodriver: '
 270     geckodriver --version | grep ^geckodriver || echo "No geckodriver. Try: wget -O- https://github.com/mozilla/geckodriver/releases/download/v0.23.0/geckodriver-v0.23.0-linux64.tar.gz | sudo tar -C /usr/local/bin -xzf - geckodriver"
 271
 272     if [[ "$NEED_SDK_R" = true ]]; then
 273       # R SDK stuff
 274       echo -n 'R: '
 275       which Rscript || fatal "No Rscript. Try: apt-get install r-base"
 276       echo -n 'testthat: '
 277       Rscript -e "library('testthat')" || fatal "No testthat. Try: apt-get install r-cran-testthat"
 278       # needed for roxygen2, needed for devtools, needed for R sdk
 279       pkg-config --exists libxml-2.0 || fatal "No libxml2. Try: apt-get install libxml2-dev"
 280       # needed for pkgdown, builds R SDK doc pages
 281       which pandoc || fatal "No pandoc. Try: apt-get install pandoc"
 282     fi
 283     echo 'procs with /dev/fuse open:'
 284     find /proc/*/fd -lname /dev/fuse 2>/dev/null | cut -d/ -f3 | xargs --no-run-if-empty ps -lywww
 285     echo 'grep fuse /proc/self/mountinfo:'
 286     grep fuse /proc/self/mountinfo
 287 }
 288
 289 rotate_logfile() {
 290   # i.e.  rotate_logfile "$WORKSPACE/apps/workbench/log/" "test.log"
 291   # $BUILD_NUMBER is set by Jenkins if this script is being called as part of a Jenkins run
 292   if [[ -f "$1/$2" ]]; then
 293     THEDATE=`date +%Y%m%d%H%M%S`
 294     mv "$1/$2" "$1/$THEDATE-$BUILD_NUMBER-$2"
 295     gzip "$1/$THEDATE-$BUILD_NUMBER-$2"
 296   fi
 297 }
 298
 299 declare -a failures
 300 declare -A skip
 301 declare -A only
 302 declare -A testargs
 303 skip[apps/workbench_profile]=1
 304
 305 while [[ -n "$1" ]]
 306 do
 307     arg="$1"; shift
 308     case "$arg" in
 309         --help)
 310             echo >&2 "$helpmessage"
 311             echo >&2
 312             exit 1
 313             ;;
 314         --skip)
 315             skip[$1]=1; shift
 316             ;;
 317         --only)
 318             only[$1]=1; skip[$1]=""; shift
 319             ;;
 320         --short)
 321             short=1
 322             ;;
 323         --interactive)
 324             interactive=1
 325             ;;
 326         --skip-install)
 327             skip[install]=1
 328             ;;
 329         --only-install)
 330             only_install="$1"; shift
 331             ;;
 332         --temp)
 333             temp="$1"; shift
 334             temp_preserve=1
 335             ;;
 336         --leave-temp)
 337             temp_preserve=1
 338             ;;
 339         --repeat)
 340             repeat=$((${1}+0)); shift
 341             ;;
 342         --retry)
 343             retry=1
 344             ;;
 345         *_test=*)
 346             suite="${arg%%_test=*}"
 347             args="${arg#*=}"
 348             testargs["$suite"]="$args"
 349             ;;
 350         *=*)
 351             eval export $(echo $arg | cut -d= -f1)=\"$(echo $arg | cut -d= -f2-)\"
 352             ;;
 353         *)
 354             echo >&2 "$0: Unrecognized option: '$arg'. Try: $0 --help"
 355             exit 1
 356             ;;
 357     esac
 358 done
 359
 360 # R SDK installation is very slow (~360s in a clean environment) and only
 361 # required when testing it. Skip that step if it is not needed.
 362 NEED_SDK_R=true
 363
 364 if [[ ${#only[@]} -ne 0 ]] &&
 365    [[ -z "${only['sdk/R']}" && -z "${only['doc']}" ]]; then
 366   NEED_SDK_R=false
 367 fi
 368
 369 if [[ ${skip["sdk/R"]} == 1 && ${skip["doc"]} == 1 ]]; then
 370   NEED_SDK_R=false
 371 fi
 372
 373 if [[ $NEED_SDK_R == false ]]; then
 374         echo "R SDK not needed, it will not be installed."
 375 fi
 376
 377 checkpidfile() {
 378     svc="$1"
 379     pid="$(cat "$WORKSPACE/tmp/${svc}.pid")"
 380     if [[ -z "$pid" ]] || ! kill -0 "$pid"; then
 381         tail $WORKSPACE/tmp/${1}*.log
 382         echo "${svc} pid ${pid} not running"
 383         return 1
 384     fi
 385     echo "${svc} pid ${pid} ok"
 386 }
 387
 388 checkhealth() {
 389     svc="$1"
 390     base=$("${VENV3DIR}/bin/python3" -c "import yaml; print(list(yaml.safe_load(open('$ARVADOS_CONFIG','r'))['Clusters']['zzzzz']['Services']['$1']['InternalURLs'].keys())[0])")
 391     url="$base/_health/ping"
 392     if ! curl -Ss -H "Authorization: Bearer e687950a23c3a9bceec28c6223a06c79" "${url}" | tee -a /dev/stderr | grep '"OK"'; then
 393         echo "${url} failed"
 394         return 1
 395     fi
 396 }
 397
 398 checkdiscoverydoc() {
 399     dd="https://${1}/discovery/v1/apis/arvados/v1/rest"
 400     if ! (set -o pipefail; curl -fsk "$dd" | grep -q ^{ ); then
 401         echo >&2 "ERROR: could not retrieve discovery doc from RailsAPI at $dd"
 402         tail -v $WORKSPACE/tmp/railsapi.log
 403         return 1
 404     fi
 405     echo "${dd} ok"
 406 }
 407
 408 start_services() {
 409     if [[ -n "$ARVADOS_TEST_API_HOST" ]]; then
 410         return 0
 411     fi
 412     . "$VENV3DIR/bin/activate"
 413     echo 'Starting API, controller, keepproxy, keep-web, arv-git-httpd, ws, and nginx ssl proxy...'
 414     if [[ ! -d "$WORKSPACE/services/api/log" ]]; then
 415         mkdir -p "$WORKSPACE/services/api/log"
 416     fi
 417     # Remove empty api.pid file if it exists
 418     if [[ -f "$WORKSPACE/tmp/api.pid" && ! -s "$WORKSPACE/tmp/api.pid" ]]; then
 419         rm -f "$WORKSPACE/tmp/api.pid"
 420     fi
 421     all_services_stopped=
 422     fail=1
 423
 424     cd "$WORKSPACE" \
 425         && eval $(python3 sdk/python/tests/run_test_server.py start --auth admin) \
 426         && export ARVADOS_TEST_API_HOST="$ARVADOS_API_HOST" \
 427         && export ARVADOS_TEST_API_INSTALLED="$$" \
 428         && checkpidfile api \
 429         && checkdiscoverydoc $ARVADOS_API_HOST \
 430         && eval $(python3 sdk/python/tests/run_test_server.py start_nginx) \
 431         && checkpidfile nginx \
 432         && python3 sdk/python/tests/run_test_server.py start_controller \
 433         && checkpidfile controller \
 434         && checkhealth Controller \
 435         && checkdiscoverydoc $ARVADOS_API_HOST \
 436         && python3 sdk/python/tests/run_test_server.py start_keep_proxy \
 437         && checkpidfile keepproxy \
 438         && python3 sdk/python/tests/run_test_server.py start_keep-web \
 439         && checkpidfile keep-web \
 440         && checkhealth WebDAV \
 441         && python3 sdk/python/tests/run_test_server.py start_arv-git-httpd \
 442         && checkpidfile arv-git-httpd \
 443         && checkhealth GitHTTP \
 444         && python3 sdk/python/tests/run_test_server.py start_ws \
 445         && checkpidfile ws \
 446         && export ARVADOS_TEST_PROXY_SERVICES=1 \
 447         && (env | egrep ^ARVADOS) \
 448         && fail=0
 449     deactivate
 450     if [[ $fail != 0 ]]; then
 451         unset ARVADOS_TEST_API_HOST
 452     fi
 453     return $fail
 454 }
 455
 456 stop_services() {
 457     if [[ -n "$all_services_stopped" ]]; then
 458         return
 459     fi
 460     unset ARVADOS_TEST_API_HOST ARVADOS_TEST_PROXY_SERVICES
 461     . "$VENV3DIR/bin/activate" || return
 462     cd "$WORKSPACE" \
 463         && python3 sdk/python/tests/run_test_server.py stop_nginx \
 464         && python3 sdk/python/tests/run_test_server.py stop_arv-git-httpd \
 465         && python3 sdk/python/tests/run_test_server.py stop_ws \
 466         && python3 sdk/python/tests/run_test_server.py stop_keep-web \
 467         && python3 sdk/python/tests/run_test_server.py stop_keep_proxy \
 468         && python3 sdk/python/tests/run_test_server.py stop_controller \
 469         && python3 sdk/python/tests/run_test_server.py stop \
 470         && all_services_stopped=1
 471     deactivate
 472     unset ARVADOS_CONFIG
 473 }
 474
 475 interrupt() {
 476     failures+=("($(basename $0) interrupted)")
 477     exit_cleanly
 478 }
 479 trap interrupt INT
 480
 481 setup_ruby_environment() {
 482     if [[ -s "$HOME/.rvm/scripts/rvm" ]] ; then
 483         source "$HOME/.rvm/scripts/rvm"
 484         using_rvm=true
 485     elif [[ -s "/usr/local/rvm/scripts/rvm" ]] ; then
 486         source "/usr/local/rvm/scripts/rvm"
 487         using_rvm=true
 488     else
 489         using_rvm=false
 490     fi
 491
 492     if [[ "$using_rvm" == true ]]; then
 493         # If rvm is in use, we can't just put separate "dependencies"
 494         # and "gems-under-test" paths to GEM_PATH: passenger resets
 495         # the environment to the "current gemset", which would lose
 496         # our GEM_PATH and prevent our test suites from running ruby
 497         # programs (for example, the Workbench test suite could not
 498         # boot an API server or run arv). Instead, we have to make an
 499         # rvm gemset and use it for everything.
 500
 501         [[ `type rvm | head -n1` == "rvm is a function" ]] \
 502             || fatal 'rvm check'
 503
 504         # Put rvm's favorite path back in first place (overriding
 505         # virtualenv, which just put itself there). Ignore rvm's
 506         # complaint about not being in first place already.
 507         rvm use @default 2>/dev/null
 508
 509         # Create (if needed) and switch to an @arvados-tests-* gemset,
 510         # salting the gemset name so it doesn't interfere with
 511         # concurrent builds in other workspaces. Leave the choice of
 512         # ruby to the caller.
 513         gemset="arvados-tests-$(echo -n "${WORKSPACE}" | md5sum | head -c16)"
 514         rvm use "@${gemset}" --create \
 515             || fatal 'rvm gemset setup'
 516
 517         rvm env
 518         (bundle version | grep -q 2.2.19) || gem install bundler -v 2.2.19
 519         bundle="$(which bundle)"
 520         echo "$bundle"
 521         "$bundle" version | grep 2.2.19 || fatal 'install bundler'
 522     else
 523         # When our "bundle install"s need to install new gems to
 524         # satisfy dependencies, we want them to go where "gem install
 525         # --user-install" would put them. (However, if the caller has
 526         # already set GEM_HOME, we assume that's where dependencies
 527         # should be installed, and we should leave it alone.)
 528
 529         if [ -z "$GEM_HOME" ]; then
 530             user_gempath="$(gem env gempath)"
 531             export GEM_HOME="${user_gempath%%:*}"
 532         fi
 533         PATH="$(gem env gemdir)/bin:$PATH"
 534
 535         # When we build and install our own gems, we install them in our
 536         # $GEMHOME tmpdir, and we want them to be at the front of GEM_PATH and
 537         # PATH so integration tests prefer them over other versions that
 538         # happen to be installed in $user_gempath, system dirs, etc.
 539
 540         tmpdir_gem_home="$(env - PATH="$PATH" HOME="$GEMHOME" gem env gempath | cut -f1 -d:)"
 541         PATH="$tmpdir_gem_home/bin:$PATH"
 542         export GEM_PATH="$tmpdir_gem_home:$(gem env gempath)"
 543
 544         echo "Will install dependencies to $(gem env gemdir)"
 545         echo "Will install bundler and arvados gems to $tmpdir_gem_home"
 546         echo "Gem search path is GEM_PATH=$GEM_PATH"
 547         bundle="$tmpdir_gem_home/bin/bundle"
 548         (
 549             export HOME=$GEMHOME
 550             bundlers="$(gem list --details bundler)"
 551             versions=(2.2.19)
 552             for v in ${versions[@]}; do
 553                 if ! echo "$bundlers" | fgrep -q "($v)"; then
 554                     gem install --user $(for v in ${versions[@]}; do echo bundler:${v}; done)
 555                     break
 556                 fi
 557             done
 558             "$bundle" version | tee /dev/stderr | grep -q 'version 2'
 559         ) || fatal 'install bundler'
 560     fi
 561 }
 562
 563 with_test_gemset() {
 564     if [[ "$using_rvm" == true ]]; then
 565         "$@"
 566     else
 567         GEM_HOME="$tmpdir_gem_home" GEM_PATH="$tmpdir_gem_home" "$@"
 568     fi
 569 }
 570
 571 gem_uninstall_if_exists() {
 572     if gem list "$1\$" | egrep '^\w'; then
 573         gem uninstall --force --all --executables "$1"
 574     fi
 575 }
 576
 577 setup_virtualenv() {
 578     local venvdest="$1"; shift
 579     if ! [[ -e "$venvdest/bin/activate" ]] || ! [[ -e "$venvdest/bin/pip3" ]]; then
 580         python3 -m venv "$@" "$venvdest" || fatal "virtualenv $venvdest failed"
 581     elif [[ -n "$short" ]]; then
 582         return
 583     fi
 584     "$venvdest/bin/pip3" install --no-cache-dir 'setuptools>=18.5' 'pip>=7'
 585 }
 586
 587 initialize() {
 588     # If dependencies like ruby, go, etc. are installed in
 589     # /var/lib/arvados -- presumably by "arvados-server install" --
 590     # then we want to use those versions, instead of whatever happens
 591     # to be installed in /usr.
 592     PATH="/var/lib/arvados/bin:${PATH}"
 593     sanity_checks
 594
 595     echo "WORKSPACE=$WORKSPACE"
 596
 597     # Clean up .pyc files that may exist in the workspace
 598     cd "$WORKSPACE"
 599     find -name '*.pyc' -delete
 600
 601     if [[ -z "$temp" ]]; then
 602         temp="$(mktemp -d)"
 603     fi
 604
 605     # Set up temporary install dirs (unless existing dirs were supplied)
 606     for tmpdir in VENV3DIR GOPATH GEMHOME PERLINSTALLBASE R_LIBS
 607     do
 608         if [[ -z "${!tmpdir}" ]]; then
 609             eval "$tmpdir"="$temp/$tmpdir"
 610         fi
 611         if ! [[ -d "${!tmpdir}" ]]; then
 612             mkdir "${!tmpdir}" || fatal "can't create ${!tmpdir} (does $temp exist?)"
 613         fi
 614     done
 615
 616     rm -vf "${WORKSPACE}/tmp/*.log"
 617
 618     export PERLINSTALLBASE
 619     export PERL5LIB="$PERLINSTALLBASE/lib/perl5${PERL5LIB:+:$PERL5LIB}"
 620
 621     export R_LIBS
 622
 623     export GOPATH
 624     # Make sure our compiled binaries under test override anything
 625     # else that might be in the environment.
 626     export PATH=$GOPATH/bin:$PATH
 627
 628     # Jenkins config requires that glob tmp/*.log match something. Ensure
 629     # that happens even if we don't end up running services that set up
 630     # logging.
 631     mkdir -p "${WORKSPACE}/tmp/" || fatal "could not mkdir ${WORKSPACE}/tmp"
 632     touch "${WORKSPACE}/tmp/controller.log" || fatal "could not touch ${WORKSPACE}/tmp/controller.log"
 633
 634     unset http_proxy https_proxy no_proxy
 635
 636     # Note: this must be the last time we change PATH, otherwise rvm will
 637     # whine a lot.
 638     setup_ruby_environment
 639
 640     echo "PATH is $PATH"
 641 }
 642
 643 install_env() {
 644     go mod download || fatal "Go deps failed"
 645     which goimports >/dev/null || go install golang.org/x/tools/cmd/goimports@latest || fatal "Go setup failed"
 646
 647     setup_virtualenv "$VENV3DIR"
 648     . "$VENV3DIR/bin/activate"
 649
 650     # Needed for run_test_server.py which is used by certain (non-Python) tests.
 651     # pdoc3 needed to generate the Python SDK documentation.
 652     (
 653         set -e
 654         "${VENV3DIR}/bin/pip3" install wheel
 655         "${VENV3DIR}/bin/pip3" install PyYAML
 656         "${VENV3DIR}/bin/pip3" install httplib2
 657         "${VENV3DIR}/bin/pip3" install future
 658         "${VENV3DIR}/bin/pip3" install google-api-python-client
 659         "${VENV3DIR}/bin/pip3" install ciso8601
 660         "${VENV3DIR}/bin/pip3" install pycurl
 661         "${VENV3DIR}/bin/pip3" install ws4py
 662         "${VENV3DIR}/bin/pip3" install pdoc3
 663         cd "$WORKSPACE/sdk/python"
 664         python3 setup.py install
 665     ) || fatal "installing PyYAML and sdk/python failed"
 666 }
 667
 668 retry() {
 669     remain="${repeat}"
 670     while :
 671     do
 672         if ${@}; then
 673             if [[ "$remain" -gt 1 ]]; then
 674                 remain=$((${remain}-1))
 675                 title "(repeating ${remain} more times)"
 676             else
 677                 break
 678             fi
 679         elif [[ "$retry" == 1 ]]; then
 680             read -p 'Try again? [Y/n] ' x
 681             if [[ "$x" != "y" ]] && [[ "$x" != "" ]]
 682             then
 683                 break
 684             fi
 685         else
 686             break
 687         fi
 688     done
 689 }
 690
 691 do_test() {
 692     case "${1}" in
 693         apps/workbench_units | apps/workbench_functionals | apps/workbench_integration)
 694             suite=apps/workbench
 695             ;;
 696         *)
 697             suite="${1}"
 698             ;;
 699     esac
 700     if [[ -n "${skip[$suite]}" || \
 701               -n "${skip[$1]}" || \
 702               (${#only[@]} -ne 0 && ${only[$suite]} -eq 0 && ${only[$1]} -eq 0) ]]; then
 703         return 0
 704     fi
 705     case "${1}" in
 706         services/api)
 707             stop_services
 708             check_arvados_config "$1"
 709             ;;
 710         gofmt | doc | lib/cli | lib/cloud/azure | lib/cloud/ec2 | lib/cloud/cloudtest | lib/cmd | lib/dispatchcloud/sshexecutor | lib/dispatchcloud/worker)
 711             check_arvados_config "$1"
 712             # don't care whether services are running
 713             ;;
 714         *)
 715             check_arvados_config "$1"
 716             if ! start_services; then
 717                 checkexit 1 "$1 tests"
 718                 title "test $1 -- failed to start services"
 719                 return 1
 720             fi
 721             ;;
 722     esac
 723     retry do_test_once ${@}
 724 }
 725
 726 go_ldflags() {
 727     version=${ARVADOS_VERSION:-$(git log -n1 --format=%H)-dev}
 728     echo "-X git.arvados.org/arvados.git/lib/cmd.version=${version} -X main.version=${version}"
 729 }
 730
 731 do_test_once() {
 732     unset result
 733
 734     title "test $1"
 735     timer_reset
 736
 737     result=
 738     if which deactivate >/dev/null; then deactivate; fi
 739     if ! . "$VENV3DIR/bin/activate"
 740     then
 741         result=1
 742     elif [[ "$2" == "go" ]]
 743     then
 744         covername="coverage-$(echo "$1" | sed -e 's/\//_/g')"
 745         coverflags=("-covermode=count" "-coverprofile=$WORKSPACE/tmp/.$covername.tmp")
 746         # We do "go install" here to catch compilation errors
 747         # before trying "go test". Otherwise, coverage-reporting
 748         # mode makes Go show the wrong line numbers when reporting
 749         # compilation errors.
 750         go install -ldflags "$(go_ldflags)" "$WORKSPACE/$1" && \
 751             cd "$WORKSPACE/$1" && \
 752             if [[ -n "${testargs[$1]}" ]]
 753         then
 754             # "go test -check.vv giturl" doesn't work, but this
 755             # does:
 756             go test ${short:+-short} ${testargs[$1]}
 757         else
 758             # The above form gets verbose even when testargs is
 759             # empty, so use this form in such cases:
 760             go test ${short:+-short} ${coverflags[@]} "git.arvados.org/arvados.git/$1"
 761         fi
 762         result=${result:-$?}
 763         if [[ -f "$WORKSPACE/tmp/.$covername.tmp" ]]
 764         then
 765             go tool cover -html="$WORKSPACE/tmp/.$covername.tmp" -o "$WORKSPACE/tmp/$covername.html"
 766             rm "$WORKSPACE/tmp/.$covername.tmp"
 767         fi
 768         [[ $result = 0 ]] && gofmt -e -d *.go
 769     elif [[ "$2" == "pip" ]]
 770     then
 771         tries=0
 772         cd "$WORKSPACE/$1" && while :
 773         do
 774             tries=$((${tries}+1))
 775             # $3 can name a path directory for us to use, including trailing
 776             # slash; e.g., the bin/ subdirectory of a virtualenv.
 777             if [[ -e "${3}activate" ]]; then
 778                 . "${3}activate"
 779             fi
 780             python setup.py ${short:+--short-tests-only} test ${testargs[$1]}
 781             result=$?
 782             if [[ ${tries} < 3 && ${result} == 137 ]]
 783             then
 784                 printf '\n*****\n%s tests killed -- retrying\n*****\n\n' "$1"
 785                 continue
 786             else
 787                 break
 788             fi
 789         done
 790     elif [[ "$2" != "" ]]
 791     then
 792         "test_$2"
 793     else
 794         "test_$1"
 795     fi
 796     result=${result:-$?}
 797     checkexit $result "$1 tests"
 798     title "test $1 -- `timer`"
 799     return $result
 800 }
 801
 802 check_arvados_config() {
 803     if [[ "$1" = "env" ]] ; then
 804         return
 805     fi
 806     if [[ -z "$ARVADOS_CONFIG" ]] ; then
 807         # Create config file.  The run_test_server script requires PyYAML,
 808         # so virtualenv needs to be active.  Downstream steps like
 809         # workbench install which require a valid config.yml.
 810         if [[ ! -s "$VENV3DIR/bin/activate" ]] ; then
 811             install_env
 812         fi
 813         . "$VENV3DIR/bin/activate"
 814         cd "$WORKSPACE"
 815         eval $(python3 sdk/python/tests/run_test_server.py setup_config)
 816         deactivate
 817     fi
 818 }
 819
 820 do_install() {
 821     if [[ -n "${skip[install]}" || ( -n "${only_install}" && "${only_install}" != "${1}" && "${only_install}" != "${2}" ) ]]; then
 822         return 0
 823     fi
 824     check_arvados_config "$1"
 825     retry do_install_once ${@}
 826 }
 827
 828 do_install_once() {
 829     title "install $1"
 830     timer_reset
 831
 832     result=
 833     if which deactivate >/dev/null; then deactivate; fi
 834     if [[ "$1" != "env" ]] && ! . "$VENV3DIR/bin/activate"; then
 835         result=1
 836     elif [[ "$2" == "go" ]]
 837     then
 838         go install -ldflags "$(go_ldflags)" "$WORKSPACE/$1"
 839     elif [[ "$2" == "pip" ]]
 840     then
 841         # $3 can name a path directory for us to use, including trailing
 842         # slash; e.g., the bin/ subdirectory of a virtualenv.
 843
 844         # Need to change to a different directory after creating
 845         # the source dist package to avoid a pip bug.
 846         # see https://arvados.org/issues/5766 for details.
 847
 848         # Also need to install twice, because if it believes the package is
 849         # already installed, pip it won't install it.  So the first "pip
 850         # install" ensures that the dependencies are met, the second "pip
 851         # install" ensures that we've actually installed the local package
 852         # we just built.
 853         cd "$WORKSPACE/$1" \
 854             && "${3}python3" setup.py sdist rotate --keep=1 --match .tar.gz \
 855             && cd "$WORKSPACE" \
 856             && "${3}pip3" install --no-cache-dir "$WORKSPACE/$1/dist"/*.tar.gz \
 857             && "${3}pip3" install --no-cache-dir --no-deps --ignore-installed "$WORKSPACE/$1/dist"/*.tar.gz
 858     elif [[ "$2" != "" ]]
 859     then
 860         "install_$2"
 861     else
 862         "install_$1"
 863     fi
 864     result=${result:-$?}
 865     checkexit $result "$1 install"
 866     title "install $1 -- `timer`"
 867     return $result
 868 }
 869
 870 bundle_install_trylocal() {
 871     (
 872         set -e
 873         echo "(Running bundle install --local. 'could not find package' messages are OK.)"
 874         if ! "$bundle" install --local --no-deployment; then
 875             echo "(Running bundle install again, without --local.)"
 876             "$bundle" install --no-deployment
 877         fi
 878         "$bundle" package
 879     )
 880 }
 881
 882 install_doc() {
 883     cd "$WORKSPACE/doc" \
 884         && bundle_install_trylocal \
 885         && rm -rf .site
 886 }
 887
 888 install_gem() {
 889     gemname=$1
 890     srcpath=$2
 891     with_test_gemset gem_uninstall_if_exists "$gemname" \
 892         && cd "$WORKSPACE/$srcpath" \
 893         && bundle_install_trylocal \
 894         && gem build "$gemname.gemspec" \
 895         && with_test_gemset gem install --no-document $(ls -t "$gemname"-*.gem|head -n1)
 896 }
 897
 898 install_sdk/ruby() {
 899     install_gem arvados sdk/ruby
 900 }
 901
 902 install_sdk/R() {
 903   if [[ "$NEED_SDK_R" = true ]]; then
 904     cd "$WORKSPACE/sdk/R" \
 905        && Rscript --vanilla install_deps.R
 906   fi
 907 }
 908
 909 install_sdk/perl() {
 910     cd "$WORKSPACE/sdk/perl" \
 911         && perl Makefile.PL INSTALL_BASE="$PERLINSTALLBASE" \
 912         && make install INSTALLDIRS=perl
 913 }
 914
 915 install_sdk/cli() {
 916     install_gem arvados-cli sdk/cli
 917 }
 918
 919 install_services/login-sync() {
 920     install_gem arvados-login-sync services/login-sync
 921 }
 922
 923 install_services/api() {
 924     stop_services
 925     check_arvados_config "services/api"
 926     cd "$WORKSPACE/services/api" \
 927         && RAILS_ENV=test bundle_install_trylocal \
 928             || return 1
 929
 930     rm -f config/environments/test.rb
 931     cp config/environments/test.rb.example config/environments/test.rb
 932
 933     # Clear out any lingering postgresql connections to the test
 934     # database, so that we can drop it. This assumes the current user
 935     # is a postgresql superuser.
 936     cd "$WORKSPACE/services/api" \
 937         && test_database=$("${VENV3DIR}/bin/python3" -c "import yaml; print(yaml.safe_load(open('$ARVADOS_CONFIG','r'))['Clusters']['zzzzz']['PostgreSQL']['Connection']['dbname'])") \
 938         && psql "$test_database" -c "SELECT pg_terminate_backend (pg_stat_activity.pid::int) FROM pg_stat_activity WHERE pg_stat_activity.datname = '$test_database';" 2>/dev/null
 939
 940     mkdir -p "$WORKSPACE/services/api/tmp/pids"
 941
 942     cert="$WORKSPACE/services/api/tmp/self-signed"
 943     if [[ ! -e "$cert.pem" || "$(date -r "$cert.pem" +%s)" -lt 1512659226 ]]; then
 944         (
 945             dir="$WORKSPACE/services/api/tmp"
 946             set -e
 947             openssl req -newkey rsa:2048 -nodes -subj '/C=US/ST=State/L=City/CN=localhost' -out "$cert.csr" -keyout "$cert.key" </dev/null
 948             openssl x509 -req -in "$cert.csr" -signkey "$cert.key" -out "$cert.pem" -days 3650 -extfile <(printf 'subjectAltName=DNS:localhost,DNS:::1,DNS:0.0.0.0,DNS:127.0.0.1,IP:::1,IP:0.0.0.0,IP:127.0.0.1')
 949         ) || return 1
 950     fi
 951
 952     cd "$WORKSPACE/services/api" \
 953         && rm -rf tmp/git \
 954         && mkdir -p tmp/git \
 955         && cd tmp/git \
 956         && tar xf ../../test/test.git.tar \
 957         && mkdir -p internal.git \
 958         && git --git-dir internal.git init \
 959             || return 1
 960
 961     (
 962         set -ex
 963         cd "$WORKSPACE/services/api"
 964         export RAILS_ENV=test
 965         if "$bundle" exec rails db:environment:set ; then
 966             "$bundle" exec rake db:drop
 967         fi
 968         "$bundle" exec rake db:setup
 969         "$bundle" exec rake db:fixtures:load
 970     ) || return 1
 971 }
 972
 973 declare -a pythonstuff
 974 pythonstuff=(
 975     sdk/python:py3
 976     sdk/cwl:py3
 977     services/dockercleaner:py3
 978     services/fuse:py3
 979     tools/crunchstat-summary:py3
 980 )
 981
 982 declare -a gostuff
 983 gostuff=($(cd "$WORKSPACE" && git grep -lw func | grep \\.go | sed -e 's/\/[^\/]*$//' | sort -u))
 984
 985 install_apps/workbench() {
 986     cd "$WORKSPACE/apps/workbench" \
 987         && mkdir -p tmp/cache \
 988         && RAILS_ENV=test bundle_install_trylocal \
 989         && RAILS_ENV=test RAILS_GROUPS=assets "$bundle" exec rake npm:install
 990 }
 991
 992 test_doc() {
 993     (
 994         set -e
 995         cd "$WORKSPACE/doc"
 996         ARVADOS_API_HOST=pirca.arvadosapi.com
 997         # Make sure python-epydoc is installed or the next line won't
 998         # do much good!
 999         PYTHONPATH=$WORKSPACE/sdk/python/ "$bundle" exec rake linkchecker baseurl=file://$WORKSPACE/doc/.site/ arvados_workbench_host=https://workbench.$ARVADOS_API_HOST arvados_api_host=$ARVADOS_API_HOST
1000     )
1001 }
1002
1003 test_gofmt() {
1004     cd "$WORKSPACE" || return 1
1005     dirs=$(ls -d */ | egrep -v 'vendor|tmp')
1006     [[ -z "$(gofmt -e -d $dirs | tee -a /dev/stderr)" ]]
1007 }
1008
1009 test_services/api() {
1010     rm -f "$WORKSPACE/services/api/git-commit.version"
1011     cd "$WORKSPACE/services/api" \
1012         && eval env RAILS_ENV=test ${short:+RAILS_TEST_SHORT=1} "$bundle" exec rake test TESTOPTS=\'-v -d\' ${testargs[services/api]}
1013 }
1014
1015 test_sdk/ruby() {
1016     cd "$WORKSPACE/sdk/ruby" \
1017         && "$bundle" exec rake test TESTOPTS=-v ${testargs[sdk/ruby]}
1018 }
1019
1020 test_sdk/R() {
1021   if [[ "$NEED_SDK_R" = true ]]; then
1022     cd "$WORKSPACE/sdk/R" \
1023         && Rscript --vanilla run_test.R
1024   fi
1025 }
1026
1027 test_sdk/cli() {
1028     cd "$WORKSPACE/sdk/cli" \
1029         && mkdir -p /tmp/keep \
1030         && KEEP_LOCAL_STORE=/tmp/keep "$bundle" exec rake test TESTOPTS=-v ${testargs[sdk/cli]}
1031 }
1032
1033 test_sdk/java-v2() {
1034     cd "$WORKSPACE/sdk/java-v2" && gradle test ${testargs[sdk/java-v2]}
1035 }
1036
1037 test_services/login-sync() {
1038     cd "$WORKSPACE/services/login-sync" \
1039         && "$bundle" exec rake test TESTOPTS=-v ${testargs[services/login-sync]}
1040 }
1041
1042 test_apps/workbench_units() {
1043     local TASK="test:units"
1044     cd "$WORKSPACE/apps/workbench" \
1045         && eval env RAILS_ENV=test ${short:+RAILS_TEST_SHORT=1} "$bundle" exec rake ${TASK} TESTOPTS=\'-v -d\' ${testargs[apps/workbench]} ${testargs[apps/workbench_units]}
1046 }
1047
1048 test_apps/workbench_functionals() {
1049     local TASK="test:functionals"
1050     cd "$WORKSPACE/apps/workbench" \
1051         && eval env RAILS_ENV=test ${short:+RAILS_TEST_SHORT=1} "$bundle" exec rake ${TASK} TESTOPTS=\'-v -d\' ${testargs[apps/workbench]} ${testargs[apps/workbench_functionals]}
1052 }
1053
1054 test_apps/workbench_integration() {
1055     local TASK="test:integration"
1056     cd "$WORKSPACE/apps/workbench" \
1057         && eval env RAILS_ENV=test ${short:+RAILS_TEST_SHORT=1} "$bundle" exec rake ${TASK} TESTOPTS=\'-v -d\' ${testargs[apps/workbench]} ${testargs[apps/workbench_integration]}
1058 }
1059
1060 test_apps/workbench_benchmark() {
1061     local TASK="test:benchmark"
1062     cd "$WORKSPACE/apps/workbench" \
1063         && eval env RAILS_ENV=test ${short:+RAILS_TEST_SHORT=1} "$bundle" exec rake ${TASK} ${testargs[apps/workbench_benchmark]}
1064 }
1065
1066 test_apps/workbench_profile() {
1067     local TASK="test:profile"
1068     cd "$WORKSPACE/apps/workbench" \
1069         && eval env RAILS_ENV=test ${short:+RAILS_TEST_SHORT=1} "$bundle" exec rake ${TASK} ${testargs[apps/workbench_profile]}
1070 }
1071
1072 install_deps() {
1073     # Install parts needed by test suites
1074     do_install env
1075     do_install cmd/arvados-server go
1076     do_install sdk/cli
1077     do_install sdk/perl
1078     do_install sdk/python pip "${VENV3DIR}/bin/"
1079     do_install sdk/ruby
1080     do_install services/api
1081     do_install services/arv-git-httpd go
1082     do_install services/keepproxy go
1083     do_install services/keep-web go
1084 }
1085
1086 install_all() {
1087     do_install env
1088     do_install doc
1089     do_install sdk/ruby
1090     do_install sdk/R
1091     do_install sdk/perl
1092     do_install sdk/cli
1093     do_install services/login-sync
1094     for p in "${pythonstuff[@]}"
1095     do
1096        dir=${p%:py3}
1097        if [[ -z ${skip[python3]} ]]; then
1098            do_install ${dir} pip "$VENV3DIR/bin/"
1099        fi
1100     done
1101     for g in "${gostuff[@]}"
1102     do
1103         do_install "$g" go
1104     done
1105     do_install services/api
1106     do_install apps/workbench
1107 }
1108
1109 test_all() {
1110     stop_services
1111     do_test services/api
1112
1113     # Shortcut for when we're only running apiserver tests. This saves a bit of time,
1114     # because we don't need to start up the api server for subsequent tests.
1115     if [ ! -z "$only" ] && [ "$only" == "services/api" ]; then
1116         rotate_logfile "$WORKSPACE/services/api/log/" "test.log"
1117         exit_cleanly
1118     fi
1119
1120     do_test gofmt
1121     do_test doc
1122     do_test sdk/ruby
1123     do_test sdk/R
1124     do_test sdk/cli
1125     do_test services/login-sync
1126     do_test sdk/java-v2
1127     for p in "${pythonstuff[@]}"
1128     do
1129         dir=${p%:py3}
1130         if [[ -z ${skip[python3]} ]]; then
1131             do_test ${dir} pip "$VENV3DIR/bin/"
1132         fi
1133     done
1134
1135     for g in "${gostuff[@]}"
1136     do
1137         do_test "$g" go
1138     done
1139     do_test apps/workbench_units
1140     do_test apps/workbench_functionals
1141     do_test apps/workbench_integration
1142     do_test apps/workbench_benchmark
1143     do_test apps/workbench_profile
1144 }
1145
1146 test_go() {
1147     do_test gofmt
1148     for g in "${gostuff[@]}"
1149     do
1150         do_test "$g" go
1151     done
1152 }
1153
1154 help_interactive() {
1155     echo "== Interactive commands:"
1156     echo "TARGET                 (short for 'test DIR')"
1157     echo "test TARGET"
1158     echo "10 test TARGET         (run test 10 times)"
1159     echo "test TARGET:py3        (test with python3)"
1160     echo "test TARGET -check.vv  (pass arguments to test)"
1161     echo "install TARGET"
1162     echo "install env            (go/python libs)"
1163     echo "install deps           (go/python libs + arvados components needed for integration tests)"
1164     echo "reset                  (...services used by integration tests)"
1165     echo "exit"
1166     echo "== Test targets:"
1167     echo "${!testfuncargs[@]}" | tr ' ' '\n' | sort | column
1168 }
1169
1170 initialize
1171
1172 declare -A testfuncargs=()
1173 for g in "${gostuff[@]}"; do
1174     testfuncargs[$g]="$g go"
1175 done
1176 for p in "${pythonstuff[@]}"; do
1177     dir=${p%:py3}
1178     testfuncargs[$dir:py3]="$dir pip $VENV3DIR/bin/"
1179 done
1180
1181 testfuncargs["sdk/cli"]="sdk/cli"
1182 testfuncargs["sdk/R"]="sdk/R"
1183 testfuncargs["sdk/java-v2"]="sdk/java-v2"
1184 testfuncargs["apps/workbench_units"]="apps/workbench_units"
1185 testfuncargs["apps/workbench_functionals"]="apps/workbench_functionals"
1186 testfuncargs["apps/workbench_integration"]="apps/workbench_integration"
1187 testfuncargs["apps/workbench_benchmark"]="apps/workbench_benchmark"
1188 testfuncargs["apps/workbench_profile"]="apps/workbench_profile"
1189
1190 if [[ -z ${interactive} ]]; then
1191     install_all
1192     test_all
1193 else
1194     skip=()
1195     only=()
1196     only_install=()
1197     if [[ -e "$VENV3DIR/bin/activate" ]]; then stop_services; fi
1198     setnextcmd() {
1199         if [[ "$TERM" = dumb ]]; then
1200             # assume emacs, or something, is offering a history buffer
1201             # and pre-populating the command will only cause trouble
1202             nextcmd=
1203         elif [[ ! -e "$VENV3DIR/bin/activate" ]]; then
1204             nextcmd="install deps"
1205         else
1206             nextcmd=""
1207         fi
1208     }
1209     echo
1210     help_interactive
1211     nextcmd="install deps"
1212     setnextcmd
1213     HISTFILE="$WORKSPACE/tmp/.history"
1214     history -r
1215     while read -p 'What next? ' -e -i "$nextcmd" nextcmd; do
1216         history -s "$nextcmd"
1217         history -w
1218         count=1
1219         if [[ "${nextcmd}" =~ ^[0-9] ]]; then
1220           read count nextcmd <<<"${nextcmd}"
1221         fi
1222         read verb target opts <<<"${nextcmd}"
1223         target="${target%/}"
1224         target="${target/\/:/:}"
1225         case "${verb}" in
1226             "exit" | "quit")
1227                 exit_cleanly
1228                 ;;
1229             "reset")
1230                 stop_services
1231                 ;;
1232             "test" | "install")
1233                 case "$target" in
1234                     "")
1235                         help_interactive
1236                         ;;
1237                     all | deps)
1238                         ${verb}_${target}
1239                         ;;
1240                     *)
1241                         argstarget=${target%:py3}
1242                         testargs["$argstarget"]="${opts}"
1243                         tt="${testfuncargs[${target}]}"
1244                         tt="${tt:-$target}"
1245                         while [ $count -gt 0 ]; do
1246                           do_$verb $tt
1247                           let "count=count-1"
1248                         done
1249                         ;;
1250                 esac
1251                 ;;
1252             "" | "help" | *)
1253                 help_interactive
1254                 ;;
1255         esac
1256         if [[ ${#successes[@]} -gt 0 || ${#failures[@]} -gt 0 ]]; then
1257             report_outcomes
1258             successes=()
1259             failures=()
1260         fi
1261         cd "$WORKSPACE"
1262         setnextcmd
1263     done
1264     echo
1265 fi
1266 exit_cleanly