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 . `dirname "$(readlink -f "$0")"`/libcloud-pin.sh
   7
   8 COLUMNS=80
   9 . `dirname "$(readlink -f "$0")"`/run-library.sh
  10
  11 read -rd "\000" helpmessage <<EOF
  12 $(basename $0): Install and test Arvados components.
  13
  14 Exit non-zero if any tests fail.
  15
  16 Syntax:
  17         $(basename $0) WORKSPACE=/path/to/arvados [options]
  18
  19 Options:
  20
  21 --skip FOO     Do not test the FOO component.
  22 --only FOO     Do not test anything except the FOO component.
  23 --temp DIR     Install components and dependencies under DIR instead of
  24                making a new temporary directory. Implies --leave-temp.
  25 --leave-temp   Do not remove GOPATH, virtualenv, and other temp dirs at exit.
  26                Instead, show the path to give as --temp to reuse them in
  27                subsequent invocations.
  28 --repeat N     Repeat each install/test step until it succeeds N times.
  29 --retry        Prompt to retry if an install or test suite fails.
  30 --skip-install Do not run any install steps. Just run tests.
  31                You should provide GOPATH, GEMHOME, and VENVDIR options
  32                from a previous invocation if you use this option.
  33 --only-install Run specific install step
  34 --short        Skip (or scale down) some slow tests.
  35 WORKSPACE=path Arvados source tree to test.
  36 CONFIGSRC=path Dir with api server config files to copy into source tree.
  37                (If none given, leave config files alone in source tree.)
  38 services/api_test="TEST=test/functional/arvados/v1/collections_controller_test.rb"
  39                Restrict apiserver tests to the given file
  40 sdk/python_test="--test-suite test.test_keep_locator"
  41                Restrict Python SDK tests to the given class
  42 apps/workbench_test="TEST=test/integration/pipeline_instances_test.rb"
  43                Restrict Workbench tests to the given file
  44 services/arv-git-httpd_test="-check.vv"
  45                Show all log messages, even when tests pass (also works
  46                with services/keepstore_test etc.)
  47 ARVADOS_DEBUG=1
  48                Print more debug messages
  49 envvar=value   Set \$envvar to value. Primarily useful for WORKSPACE,
  50                *_test, and other examples shown above.
  51
  52 Assuming --skip-install is not given, all components are installed
  53 into \$GOPATH, \$VENDIR, and \$GEMHOME before running any tests. Many
  54 test suites depend on other components being installed, and installing
  55 everything tends to be quicker than debugging dependencies.
  56
  57 As a special concession to the current CI server config, CONFIGSRC
  58 defaults to $HOME/arvados-api-server if that directory exists.
  59
  60 More information and background:
  61
  62 https://arvados.org/projects/arvados/wiki/Running_tests
  63
  64 Available tests:
  65
  66 apps/workbench (*)
  67 apps/workbench_units (*)
  68 apps/workbench_functionals (*)
  69 apps/workbench_integration (*)
  70 apps/workbench_benchmark
  71 apps/workbench_profile
  72 doc
  73 services/api
  74 services/arv-git-httpd
  75 services/crunchstat
  76 services/dockercleaner
  77 services/fuse
  78 services/health
  79 services/keep-web
  80 services/keepproxy
  81 services/keepstore
  82 services/keep-balance
  83 services/login-sync
  84 services/nodemanager
  85 services/nodemanager_integration
  86 services/crunch-run
  87 services/crunch-dispatch-local
  88 services/crunch-dispatch-slurm
  89 services/ws
  90 sdk/cli
  91 sdk/pam
  92 sdk/python
  93 sdk/python:py3
  94 sdk/ruby
  95 sdk/go/arvados
  96 sdk/go/arvadosclient
  97 sdk/go/dispatch
  98 sdk/go/keepclient
  99 sdk/go/health
 100 sdk/go/httpserver
 101 sdk/go/manifest
 102 sdk/go/blockdigest
 103 sdk/go/streamer
 104 sdk/go/stats
 105 sdk/go/crunchrunner
 106 sdk/cwl
 107 tools/arv-sync-groups
 108 tools/crunchstat-summary
 109 tools/keep-exercise
 110 tools/keep-rsync
 111 tools/keep-block-check
 112
 113 (*) apps/workbench is shorthand for apps/workbench_units +
 114     apps/workbench_functionals + apps/workbench_integration
 115
 116 EOF
 117
 118 # First make sure to remove any ARVADOS_ variables from the calling
 119 # environment that could interfere with the tests.
 120 unset $(env | cut -d= -f1 | grep \^ARVADOS_)
 121
 122 # Reset other variables that could affect our [tests'] behavior by
 123 # accident.
 124 GITDIR=
 125 GOPATH=
 126 VENVDIR=
 127 VENV3DIR=
 128 PYTHONPATH=
 129 GEMHOME=
 130 PERLINSTALLBASE=
 131
 132 short=
 133 only_install=
 134 temp=
 135 temp_preserve=
 136
 137 clear_temp() {
 138     if [[ -z "$temp" ]]; then
 139         # we didn't even get as far as making a temp dir
 140         :
 141     elif [[ -z "$temp_preserve" ]]; then
 142         rm -rf "$temp"
 143     else
 144         echo "Leaving behind temp dirs in $temp"
 145     fi
 146 }
 147
 148 fatal() {
 149     clear_temp
 150     echo >&2 "Fatal: $* (encountered in ${FUNCNAME[1]} at ${BASH_SOURCE[1]} line ${BASH_LINENO[0]})"
 151     exit 1
 152 }
 153
 154 exit_cleanly() {
 155     trap - INT
 156     create-plot-data-from-log.sh $BUILD_NUMBER "$WORKSPACE/apps/workbench/log/test.log" "$WORKSPACE/apps/workbench/log/"
 157     rotate_logfile "$WORKSPACE/apps/workbench/log/" "test.log"
 158     stop_services
 159     rotate_logfile "$WORKSPACE/services/api/log/" "test.log"
 160     report_outcomes
 161     clear_temp
 162     exit ${#failures}
 163 }
 164
 165 sanity_checks() {
 166     ( [[ -n "$WORKSPACE" ]] && [[ -d "$WORKSPACE/services" ]] ) \
 167         || fatal "WORKSPACE environment variable not set to a source directory (see: $0 --help)"
 168     echo Checking dependencies:
 169     echo -n 'virtualenv: '
 170     virtualenv --version \
 171         || fatal "No virtualenv. Try: apt-get install virtualenv (on ubuntu: python-virtualenv)"
 172     echo -n 'ruby: '
 173     ruby -v \
 174         || fatal "No ruby. Install >=2.1.9 (using rbenv, rvm, or source)"
 175     echo -n 'go: '
 176     go version \
 177         || fatal "No go binary. See http://golang.org/doc/install"
 178     [[ $(go version) =~ go1.([0-9]+) ]] && [[ ${BASH_REMATCH[1]} -ge 8 ]] \
 179         || fatal "Go >= 1.8 required. See http://golang.org/doc/install"
 180     echo -n 'gcc: '
 181     gcc --version | egrep ^gcc \
 182         || fatal "No gcc. Try: apt-get install build-essential"
 183     echo -n 'fuse.h: '
 184     find /usr/include -path '*fuse/fuse.h' | egrep --max-count=1 . \
 185         || fatal "No fuse/fuse.h. Try: apt-get install libfuse-dev"
 186     echo -n 'gnutls.h: '
 187     find /usr/include -path '*gnutls/gnutls.h' | egrep --max-count=1 . \
 188         || fatal "No gnutls/gnutls.h. Try: apt-get install libgnutls28-dev"
 189     echo -n 'Python2 pyconfig.h: '
 190     find /usr/include -path '*/python2*/pyconfig.h' | egrep --max-count=1 . \
 191         || fatal "No Python2 pyconfig.h. Try: apt-get install python2.7-dev"
 192     echo -n 'Python3 pyconfig.h: '
 193     find /usr/include -path '*/python3*/pyconfig.h' | egrep --max-count=1 . \
 194         || fatal "No Python3 pyconfig.h. Try: apt-get install python3-dev"
 195     echo -n 'nginx: '
 196     PATH="$PATH:/sbin:/usr/sbin:/usr/local/sbin" nginx -v \
 197         || fatal "No nginx. Try: apt-get install nginx"
 198     echo -n 'perl: '
 199     perl -v | grep version \
 200         || fatal "No perl. Try: apt-get install perl"
 201     for mod in ExtUtils::MakeMaker JSON LWP Net::SSL; do
 202         echo -n "perl $mod: "
 203         perl -e "use $mod; print \"\$$mod::VERSION\\n\"" \
 204             || fatal "No $mod. Try: apt-get install perl-modules libcrypt-ssleay-perl libjson-perl libwww-perl"
 205     done
 206     echo -n 'gitolite: '
 207     which gitolite \
 208         || fatal "No gitolite. Try: apt-get install gitolite3"
 209     echo -n 'npm: '
 210     npm --version \
 211         || fatal "No npm. Try: wget -O- https://nodejs.org/dist/v6.11.2/node-v6.11.2-linux-x64.tar.xz | sudo tar -C /usr/local -xJf - && sudo ln -s ../node-v6.11.2-linux-x64/bin/{node,npm} /usr/local/bin/"
 212     echo -n 'cadaver: '
 213     cadaver --version | grep -w cadaver \
 214           || fatal "No cadaver. Try: apt-get install cadaver"
 215     echo -n 'libattr1 xattr.h: '
 216     find /usr/include -path '*/attr/xattr.h' | egrep --max-count=1 . \
 217         || fatal "No libattr1 xattr.h. Try: apt-get install libattr1-dev"
 218     echo -n 'libcurl curl.h: '
 219     find /usr/include -path '*/curl/curl.h' | egrep --max-count=1 . \
 220         || fatal "No libcurl curl.h. Try: apt-get install libcurl4-gnutls-dev"
 221     echo -n 'libpq libpq-fe.h: '
 222     find /usr/include -path '*/postgresql/libpq-fe.h' | egrep --max-count=1 . \
 223         || fatal "No libpq libpq-fe.h. Try: apt-get install libpq-dev"
 224     echo -n 'services/api/config/database.yml: '
 225     if [[ ! -f "$WORKSPACE/services/api/config/database.yml" ]]; then
 226             fatal "Please provide a database.yml file for the test suite"
 227     else
 228             echo "OK"
 229     fi
 230     echo -n 'postgresql: '
 231     psql --version || fatal "No postgresql. Try: apt-get install postgresql postgresql-client-common"
 232     echo -n 'phantomjs: '
 233     phantomjs --version || fatal "No phantomjs. Try: apt-get install phantomjs"
 234     echo -n 'xvfb: '
 235     which Xvfb || fatal "No xvfb. Try: apt-get install xvfb"
 236 }
 237
 238 rotate_logfile() {
 239   # i.e.  rotate_logfile "$WORKSPACE/apps/workbench/log/" "test.log"
 240   # $BUILD_NUMBER is set by Jenkins if this script is being called as part of a Jenkins run
 241   if [[ -f "$1/$2" ]]; then
 242     THEDATE=`date +%Y%m%d%H%M%S`
 243     mv "$1/$2" "$1/$THEDATE-$BUILD_NUMBER-$2"
 244     gzip "$1/$THEDATE-$BUILD_NUMBER-$2"
 245   fi
 246 }
 247
 248 declare -a failures
 249 declare -A skip
 250 declare -A testargs
 251 skip[apps/workbench_profile]=1
 252
 253 while [[ -n "$1" ]]
 254 do
 255     arg="$1"; shift
 256     case "$arg" in
 257         --help)
 258             echo >&2 "$helpmessage"
 259             echo >&2
 260             exit 1
 261             ;;
 262         --skip)
 263             skip[$1]=1; shift
 264             ;;
 265         --only)
 266             only="$1"; skip[$1]=""; shift
 267             ;;
 268         --short)
 269             short=1
 270             ;;
 271         --skip-install)
 272             only_install=nothing
 273             ;;
 274         --only-install)
 275             only_install="$1"; shift
 276             ;;
 277         --temp)
 278             temp="$1"; shift
 279             temp_preserve=1
 280             ;;
 281         --leave-temp)
 282             temp_preserve=1
 283             ;;
 284         --repeat)
 285             repeat=$((${1}+0)); shift
 286             ;;
 287         --retry)
 288             retry=1
 289             ;;
 290         *_test=*)
 291             suite="${arg%%_test=*}"
 292             args="${arg#*=}"
 293             testargs["$suite"]="$args"
 294             ;;
 295         *=*)
 296             eval export $(echo $arg | cut -d= -f1)=\"$(echo $arg | cut -d= -f2-)\"
 297             ;;
 298         *)
 299             echo >&2 "$0: Unrecognized option: '$arg'. Try: $0 --help"
 300             exit 1
 301             ;;
 302     esac
 303 done
 304
 305 start_api() {
 306     echo 'Starting API server...'
 307     if [[ ! -d "$WORKSPACE/services/api/log" ]]; then
 308         mkdir -p "$WORKSPACE/services/api/log"
 309     fi
 310     # Remove empty api.pid file if it exists
 311     if [[ -f "$WORKSPACE/tmp/api.pid" && ! -s "$WORKSPACE/tmp/api.pid" ]]; then
 312         rm -f "$WORKSPACE/tmp/api.pid"
 313     fi
 314     cd "$WORKSPACE" \
 315         && eval $(python sdk/python/tests/run_test_server.py start --auth admin) \
 316         && export ARVADOS_TEST_API_HOST="$ARVADOS_API_HOST" \
 317         && export ARVADOS_TEST_API_INSTALLED="$$" \
 318         && python sdk/python/tests/run_test_server.py start_ws \
 319         && python sdk/python/tests/run_test_server.py start_nginx \
 320         && (env | egrep ^ARVADOS)
 321 }
 322
 323 start_nginx_proxy_services() {
 324     echo 'Starting keepproxy, keep-web, ws, arv-git-httpd, and nginx ssl proxy...'
 325     cd "$WORKSPACE" \
 326         && python sdk/python/tests/run_test_server.py start_keep_proxy \
 327         && python sdk/python/tests/run_test_server.py start_keep-web \
 328         && python sdk/python/tests/run_test_server.py start_arv-git-httpd \
 329         && python sdk/python/tests/run_test_server.py start_ws \
 330         && python sdk/python/tests/run_test_server.py start_nginx \
 331         && export ARVADOS_TEST_PROXY_SERVICES=1
 332 }
 333
 334 stop_services() {
 335     if [[ -n "$ARVADOS_TEST_PROXY_SERVICES" ]]; then
 336         unset ARVADOS_TEST_PROXY_SERVICES
 337         cd "$WORKSPACE" \
 338             && python sdk/python/tests/run_test_server.py stop_nginx \
 339             && python sdk/python/tests/run_test_server.py stop_arv-git-httpd \
 340             && python sdk/python/tests/run_test_server.py stop_ws \
 341             && python sdk/python/tests/run_test_server.py stop_keep-web \
 342             && python sdk/python/tests/run_test_server.py stop_keep_proxy
 343     fi
 344     if [[ -n "$ARVADOS_TEST_API_HOST" ]]; then
 345         unset ARVADOS_TEST_API_HOST
 346         cd "$WORKSPACE" \
 347             && python sdk/python/tests/run_test_server.py stop_nginx \
 348             && python sdk/python/tests/run_test_server.py stop_ws \
 349             && python sdk/python/tests/run_test_server.py stop
 350     fi
 351 }
 352
 353 interrupt() {
 354     failures+=("($(basename $0) interrupted)")
 355     exit_cleanly
 356 }
 357 trap interrupt INT
 358
 359 sanity_checks
 360
 361 echo "WORKSPACE=$WORKSPACE"
 362
 363 if [[ -z "$CONFIGSRC" ]] && [[ -d "$HOME/arvados-api-server" ]]; then
 364     # Jenkins expects us to use this by default.
 365     CONFIGSRC="$HOME/arvados-api-server"
 366 fi
 367
 368 # Clean up .pyc files that may exist in the workspace
 369 cd "$WORKSPACE"
 370 find -name '*.pyc' -delete
 371
 372 if [[ -z "$temp" ]]; then
 373     temp="$(mktemp -d)"
 374 fi
 375
 376 # Set up temporary install dirs (unless existing dirs were supplied)
 377 for tmpdir in VENVDIR VENV3DIR GOPATH GEMHOME PERLINSTALLBASE
 378 do
 379     if [[ -z "${!tmpdir}" ]]; then
 380         eval "$tmpdir"="$temp/$tmpdir"
 381     fi
 382     if ! [[ -d "${!tmpdir}" ]]; then
 383         mkdir "${!tmpdir}" || fatal "can't create ${!tmpdir} (does $temp exist?)"
 384     fi
 385 done
 386
 387 setup_ruby_environment() {
 388     if [[ -s "$HOME/.rvm/scripts/rvm" ]] ; then
 389         source "$HOME/.rvm/scripts/rvm"
 390         using_rvm=true
 391     elif [[ -s "/usr/local/rvm/scripts/rvm" ]] ; then
 392         source "/usr/local/rvm/scripts/rvm"
 393         using_rvm=true
 394     else
 395         using_rvm=false
 396     fi
 397
 398     if [[ "$using_rvm" == true ]]; then
 399         # If rvm is in use, we can't just put separate "dependencies"
 400         # and "gems-under-test" paths to GEM_PATH: passenger resets
 401         # the environment to the "current gemset", which would lose
 402         # our GEM_PATH and prevent our test suites from running ruby
 403         # programs (for example, the Workbench test suite could not
 404         # boot an API server or run arv). Instead, we have to make an
 405         # rvm gemset and use it for everything.
 406
 407         [[ `type rvm | head -n1` == "rvm is a function" ]] \
 408             || fatal 'rvm check'
 409
 410         # Put rvm's favorite path back in first place (overriding
 411         # virtualenv, which just put itself there). Ignore rvm's
 412         # complaint about not being in first place already.
 413         rvm use @default 2>/dev/null
 414
 415         # Create (if needed) and switch to an @arvados-tests-* gemset,
 416         # salting the gemset name so it doesn't interfere with
 417         # concurrent builds in other workspaces. Leave the choice of
 418         # ruby to the caller.
 419         gemset="arvados-tests-$(echo -n "${WORKSPACE}" | md5sum | head -c16)"
 420         rvm use "@${gemset}" --create \
 421             || fatal 'rvm gemset setup'
 422
 423         rvm env
 424     else
 425         # When our "bundle install"s need to install new gems to
 426         # satisfy dependencies, we want them to go where "gem install
 427         # --user-install" would put them. (However, if the caller has
 428         # already set GEM_HOME, we assume that's where dependencies
 429         # should be installed, and we should leave it alone.)
 430
 431         if [ -z "$GEM_HOME" ]; then
 432             user_gempath="$(gem env gempath)"
 433             export GEM_HOME="${user_gempath%%:*}"
 434         fi
 435         PATH="$(gem env gemdir)/bin:$PATH"
 436
 437         # When we build and install our own gems, we install them in our
 438         # $GEMHOME tmpdir, and we want them to be at the front of GEM_PATH and
 439         # PATH so integration tests prefer them over other versions that
 440         # happen to be installed in $user_gempath, system dirs, etc.
 441
 442         tmpdir_gem_home="$(env - PATH="$PATH" HOME="$GEMHOME" gem env gempath | cut -f1 -d:)"
 443         PATH="$tmpdir_gem_home/bin:$PATH"
 444         export GEM_PATH="$tmpdir_gem_home"
 445
 446         echo "Will install dependencies to $(gem env gemdir)"
 447         echo "Will install arvados gems to $tmpdir_gem_home"
 448         echo "Gem search path is GEM_PATH=$GEM_PATH"
 449     fi
 450     bundle config || gem install bundler \
 451         || fatal 'install bundler'
 452 }
 453
 454 with_test_gemset() {
 455     if [[ "$using_rvm" == true ]]; then
 456         "$@"
 457     else
 458         GEM_HOME="$tmpdir_gem_home" GEM_PATH="$tmpdir_gem_home" "$@"
 459     fi
 460 }
 461
 462 gem_uninstall_if_exists() {
 463     if gem list "$1\$" | egrep '^\w'; then
 464         gem uninstall --force --all --executables "$1"
 465     fi
 466 }
 467
 468 setup_virtualenv() {
 469     local venvdest="$1"; shift
 470     if ! [[ -e "$venvdest/bin/activate" ]] || ! [[ -e "$venvdest/bin/pip" ]]; then
 471         virtualenv --setuptools "$@" "$venvdest" || fatal "virtualenv $venvdest failed"
 472     fi
 473     if [[ $("$venvdest/bin/python" --version 2>&1) =~ \ 3\.[012]\. ]]; then
 474         # pip 8.0.0 dropped support for python 3.2, e.g., debian wheezy
 475         "$venvdest/bin/pip" install 'setuptools>=18.5' 'pip>=7,<8'
 476     else
 477         "$venvdest/bin/pip" install 'setuptools>=18.5' 'pip>=7'
 478     fi
 479     # ubuntu1404 can't seem to install mock via tests_require, but it can do this.
 480     "$venvdest/bin/pip" install 'mock>=1.0' 'pbr<1.7.0'
 481 }
 482
 483 export PERLINSTALLBASE
 484 export PERLLIB="$PERLINSTALLBASE/lib/perl5:${PERLLIB:+$PERLLIB}"
 485
 486 export GOPATH
 487 mkdir -p "$GOPATH/src/git.curoverse.com"
 488 ln -sfT "$WORKSPACE" "$GOPATH/src/git.curoverse.com/arvados.git" \
 489     || fatal "symlink failed"
 490 go get -v github.com/kardianos/govendor \
 491     || fatal "govendor install failed"
 492
 493 setup_virtualenv "$VENVDIR" --python python2.7
 494 . "$VENVDIR/bin/activate"
 495
 496 # Needed for run_test_server.py which is used by certain (non-Python) tests.
 497 pip freeze 2>/dev/null | egrep ^PyYAML= \
 498     || pip install PyYAML >/dev/null \
 499     || fatal "pip install PyYAML failed"
 500
 501 # Preinstall forked version of libcloud, because nodemanager "pip install"
 502 # won't pick it up by default.
 503 pip freeze 2>/dev/null | egrep ^apache-libcloud==$LIBCLOUD_PIN \
 504     || pip install --pre --ignore-installed https://github.com/curoverse/libcloud/archive/apache-libcloud-$LIBCLOUD_PIN.zip >/dev/null \
 505     || fatal "pip install apache-libcloud failed"
 506
 507 # We need an unreleased (as of 2017-08-17) llfuse bugfix, otherwise our fuse test suite deadlocks.
 508 pip freeze | grep -x llfuse==1.2.0 || (
 509     set -e
 510     yes | pip uninstall llfuse || true
 511     cython --version || fatal "no cython; try sudo apt-get install cython"
 512     cd "$temp"
 513     (cd python-llfuse 2>/dev/null || git clone https://github.com/curoverse/python-llfuse)
 514     cd python-llfuse
 515     git checkout 620722fd990ea642ddb8e7412676af482c090c0c
 516     git checkout setup.py
 517     sed -i -e "s:'1\\.2':'1.2.0':" setup.py
 518     python setup.py build_cython
 519     python setup.py install --force
 520 ) || fatal "llfuse fork failed"
 521 pip freeze | grep -x llfuse==1.2.0 || fatal "error: installed llfuse 1.2.0 but '$(pip freeze | grep llfuse)' ???"
 522
 523 # Deactivate Python 2 virtualenv
 524 deactivate
 525
 526 declare -a pythonstuff
 527 pythonstuff=(
 528     sdk/pam
 529     sdk/python
 530     sdk/python:py3
 531     sdk/cwl
 532     services/dockercleaner:py3
 533     services/fuse
 534     services/nodemanager
 535     tools/crunchstat-summary
 536     )
 537
 538 # If Python 3 is available, set up its virtualenv in $VENV3DIR.
 539 # Otherwise, skip dependent tests.
 540 PYTHON3=$(which python3)
 541 if [[ ${?} = 0 ]]; then
 542     setup_virtualenv "$VENV3DIR" --python python3
 543 else
 544     PYTHON3=
 545     cat >&2 <<EOF
 546
 547 Warning: python3 could not be found. Python 3 tests will be skipped.
 548
 549 EOF
 550 fi
 551
 552 # Reactivate Python 2 virtualenv
 553 . "$VENVDIR/bin/activate"
 554
 555 # Note: this must be the last time we change PATH, otherwise rvm will
 556 # whine a lot.
 557 setup_ruby_environment
 558
 559 echo "PATH is $PATH"
 560
 561 if ! which bundler >/dev/null
 562 then
 563     gem install --user-install bundler || fatal 'Could not install bundler'
 564 fi
 565
 566 retry() {
 567     remain="${repeat}"
 568     while :
 569     do
 570         if ${@}; then
 571             if [[ "$remain" -gt 1 ]]; then
 572                 remain=$((${remain}-1))
 573                 title "Repeating ${remain} more times"
 574             else
 575                 break
 576             fi
 577         elif [[ "$retry" == 1 ]]; then
 578             read -p 'Try again? [Y/n] ' x
 579             if [[ "$x" != "y" ]] && [[ "$x" != "" ]]
 580             then
 581                 break
 582             fi
 583         else
 584             break
 585         fi
 586     done
 587 }
 588
 589 do_test() {
 590     case "${1}" in
 591         apps/workbench_units | apps/workbench_functionals | apps/workbench_integration)
 592             suite=apps/workbench
 593             ;;
 594         services/nodemanager | services/nodemanager_integration)
 595             suite=services/nodemanager_suite
 596             ;;
 597         *)
 598             suite="${1}"
 599             ;;
 600     esac
 601     if [[ -z "${skip[$suite]}" && -z "${skip[$1]}" && \
 602                 (-z "${only}" || "${only}" == "${suite}" || \
 603                  "${only}" == "${1}") ]]; then
 604         retry do_test_once ${@}
 605     else
 606         title "Skipping ${1} tests"
 607     fi
 608 }
 609
 610 do_test_once() {
 611     unset result
 612
 613     title "Running $1 tests"
 614     timer_reset
 615     if [[ "$2" == "go" ]]
 616     then
 617         covername="coverage-$(echo "$1" | sed -e 's/\//_/g')"
 618         coverflags=("-covermode=count" "-coverprofile=$WORKSPACE/tmp/.$covername.tmp")
 619         # We do "go get -t" here to catch compilation errors
 620         # before trying "go test". Otherwise, coverage-reporting
 621         # mode makes Go show the wrong line numbers when reporting
 622         # compilation errors.
 623         go get -t "git.curoverse.com/arvados.git/$1" && \
 624             cd "$GOPATH/src/git.curoverse.com/arvados.git/$1" && \
 625             [[ -z "$(gofmt -e -d . | tee /dev/stderr)" ]] && \
 626             if [[ -n "${testargs[$1]}" ]]
 627         then
 628             # "go test -check.vv giturl" doesn't work, but this
 629             # does:
 630             go test ${short:+-short} ${testargs[$1]}
 631         else
 632             # The above form gets verbose even when testargs is
 633             # empty, so use this form in such cases:
 634             go test ${short:+-short} ${coverflags[@]} "git.curoverse.com/arvados.git/$1"
 635         fi
 636         result=${result:-$?}
 637         if [[ -f "$WORKSPACE/tmp/.$covername.tmp" ]]
 638         then
 639             go tool cover -html="$WORKSPACE/tmp/.$covername.tmp" -o "$WORKSPACE/tmp/$covername.html"
 640             rm "$WORKSPACE/tmp/.$covername.tmp"
 641         fi
 642     elif [[ "$2" == "pip" ]]
 643     then
 644         tries=0
 645         cd "$WORKSPACE/$1" && while :
 646         do
 647             tries=$((${tries}+1))
 648             # $3 can name a path directory for us to use, including trailing
 649             # slash; e.g., the bin/ subdirectory of a virtualenv.
 650             "${3}python" setup.py ${short:+--short-tests-only} test ${testargs[$1]}
 651             result=$?
 652             if [[ ${tries} < 3 && ${result} == 137 ]]
 653             then
 654                 printf '\n*****\n%s tests killed -- retrying\n*****\n\n' "$1"
 655                 continue
 656             else
 657                 break
 658             fi
 659         done
 660     elif [[ "$2" != "" ]]
 661     then
 662         "test_$2"
 663     else
 664         "test_$1"
 665     fi
 666     result=${result:-$?}
 667     checkexit $result "$1 tests"
 668     title "End of $1 tests (`timer`)"
 669     return $result
 670 }
 671
 672 do_install() {
 673     if [[ -z "${only_install}" || "${only_install}" == "${1}" ]]; then
 674         retry do_install_once ${@}
 675     else
 676         title "Skipping $1 install"
 677     fi
 678 }
 679
 680 do_install_once() {
 681     title "Running $1 install"
 682     timer_reset
 683     if [[ "$2" == "go" ]]
 684     then
 685         go get -t "git.curoverse.com/arvados.git/$1"
 686     elif [[ "$2" == "pip" ]]
 687     then
 688         # $3 can name a path directory for us to use, including trailing
 689         # slash; e.g., the bin/ subdirectory of a virtualenv.
 690
 691         # Need to change to a different directory after creating
 692         # the source dist package to avoid a pip bug.
 693         # see https://arvados.org/issues/5766 for details.
 694
 695         # Also need to install twice, because if it believes the package is
 696         # already installed, pip it won't install it.  So the first "pip
 697         # install" ensures that the dependencies are met, the second "pip
 698         # install" ensures that we've actually installed the local package
 699         # we just built.
 700         cd "$WORKSPACE/$1" \
 701             && "${3}python" setup.py sdist rotate --keep=1 --match .tar.gz \
 702             && cd "$WORKSPACE" \
 703             && "${3}pip" install --quiet "$WORKSPACE/$1/dist"/*.tar.gz \
 704             && "${3}pip" install --quiet --no-deps --ignore-installed "$WORKSPACE/$1/dist"/*.tar.gz
 705     elif [[ "$2" != "" ]]
 706     then
 707         "install_$2"
 708     else
 709         "install_$1"
 710     fi
 711     result=$?
 712     checkexit $result "$1 install"
 713     title "End of $1 install (`timer`)"
 714     return $result
 715 }
 716
 717 bundle_install_trylocal() {
 718     (
 719         set -e
 720         echo "(Running bundle install --local. 'could not find package' messages are OK.)"
 721         if ! bundle install --local --no-deployment; then
 722             echo "(Running bundle install again, without --local.)"
 723             bundle install --no-deployment
 724         fi
 725         bundle package --all
 726     )
 727 }
 728
 729 install_doc() {
 730     cd "$WORKSPACE/doc" \
 731         && bundle_install_trylocal \
 732         && rm -rf .site
 733 }
 734 do_install doc
 735
 736 install_gem() {
 737     gemname=$1
 738     srcpath=$2
 739     with_test_gemset gem_uninstall_if_exists "$gemname" \
 740         && cd "$WORKSPACE/$srcpath" \
 741         && bundle_install_trylocal \
 742         && gem build "$gemname.gemspec" \
 743         && with_test_gemset gem install --no-ri --no-rdoc $(ls -t "$gemname"-*.gem|head -n1)
 744 }
 745
 746 install_ruby_sdk() {
 747     install_gem arvados sdk/ruby
 748 }
 749 do_install sdk/ruby ruby_sdk
 750
 751 install_perl_sdk() {
 752     cd "$WORKSPACE/sdk/perl" \
 753         && perl Makefile.PL INSTALL_BASE="$PERLINSTALLBASE" \
 754         && make install INSTALLDIRS=perl
 755 }
 756 do_install sdk/perl perl_sdk
 757
 758 install_cli() {
 759     install_gem arvados-cli sdk/cli
 760 }
 761 do_install sdk/cli cli
 762
 763 install_login-sync() {
 764     install_gem arvados-login-sync services/login-sync
 765 }
 766 do_install services/login-sync login-sync
 767
 768 # Install the Python SDK early. Various other test suites (like
 769 # keepproxy) bring up run_test_server.py, which imports the arvados
 770 # module. We can't actually *test* the Python SDK yet though, because
 771 # its own test suite brings up some of those other programs (like
 772 # keepproxy).
 773 for p in "${pythonstuff[@]}"
 774 do
 775     dir=${p%:py3}
 776     if [[ ${dir} = ${p} ]]; then
 777         if [[ -z ${skip[python2]} ]]; then
 778             do_install ${dir} pip
 779         fi
 780     elif [[ -n ${PYTHON3} ]]; then
 781         if [[ -z ${skip[python3]} ]]; then
 782             do_install ${dir} pip "$VENV3DIR/bin/"
 783         fi
 784     fi
 785 done
 786
 787 install_apiserver() {
 788     cd "$WORKSPACE/services/api" \
 789         && RAILS_ENV=test bundle_install_trylocal
 790
 791     rm -f config/environments/test.rb
 792     cp config/environments/test.rb.example config/environments/test.rb
 793
 794     if [ -n "$CONFIGSRC" ]
 795     then
 796         for f in database.yml
 797         do
 798             cp "$CONFIGSRC/$f" config/ || fatal "$f"
 799         done
 800     fi
 801
 802     # Clear out any lingering postgresql connections to the test
 803     # database, so that we can drop it. This assumes the current user
 804     # is a postgresql superuser.
 805     cd "$WORKSPACE/services/api" \
 806         && test_database=$(python -c "import yaml; print yaml.load(file('config/database.yml'))['test']['database']") \
 807         && psql "$test_database" -c "SELECT pg_terminate_backend (pg_stat_activity.procpid::int) FROM pg_stat_activity WHERE pg_stat_activity.datname = '$test_database';" 2>/dev/null
 808
 809     mkdir -p "$WORKSPACE/services/api/tmp/pids"
 810
 811     cd "$WORKSPACE/services/api" \
 812         && RAILS_ENV=test bundle exec rake db:drop \
 813         && RAILS_ENV=test bundle exec rake db:setup \
 814         && RAILS_ENV=test bundle exec rake db:fixtures:load
 815 }
 816 do_install services/api apiserver
 817
 818 cd "$GOPATH/src/git.curoverse.com/arvados.git" && \
 819     "$GOPATH/bin/govendor" sync -v || \
 820         fatal "govendor sync failed"
 821 declare -a gostuff
 822 gostuff=(
 823     sdk/go/arvados
 824     sdk/go/arvadosclient
 825     sdk/go/blockdigest
 826     sdk/go/dispatch
 827     sdk/go/health
 828     sdk/go/httpserver
 829     sdk/go/manifest
 830     sdk/go/streamer
 831     sdk/go/crunchrunner
 832     sdk/go/stats
 833     lib/crunchstat
 834     services/arv-git-httpd
 835     services/crunchstat
 836     services/health
 837     services/keep-web
 838     services/keepstore
 839     sdk/go/keepclient
 840     services/keep-balance
 841     services/keepproxy
 842     services/crunch-dispatch-local
 843     services/crunch-dispatch-slurm
 844     services/crunch-run
 845     services/ws
 846     tools/keep-block-check
 847     tools/keep-exercise
 848     tools/keep-rsync
 849     tools/arv-sync-groups
 850 )
 851 for g in "${gostuff[@]}"
 852 do
 853     do_install "$g" go
 854 done
 855
 856 install_workbench() {
 857     cd "$WORKSPACE/apps/workbench" \
 858         && mkdir -p tmp/cache \
 859         && RAILS_ENV=test bundle_install_trylocal \
 860         && RAILS_ENV=test RAILS_GROUPS=assets bundle exec rake npm:install
 861 }
 862 do_install apps/workbench workbench
 863
 864 unset http_proxy https_proxy no_proxy
 865
 866 test_doclinkchecker() {
 867     (
 868         set -e
 869         cd "$WORKSPACE/doc"
 870         ARVADOS_API_HOST=qr1hi.arvadosapi.com
 871         # Make sure python-epydoc is installed or the next line won't
 872         # do much good!
 873         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
 874     )
 875 }
 876 do_test doc doclinkchecker
 877
 878 stop_services
 879
 880 test_apiserver() {
 881     rm -f "$WORKSPACE/services/api/git-commit.version"
 882     cd "$WORKSPACE/services/api" \
 883         && env RAILS_ENV=test ${short:+RAILS_TEST_SHORT=1} bundle exec rake test TESTOPTS=-v ${testargs[services/api]}
 884 }
 885 do_test services/api apiserver
 886
 887 # Shortcut for when we're only running apiserver tests. This saves a bit of time,
 888 # because we don't need to start up the api server for subsequent tests.
 889 if [ ! -z "$only" ] && [ "$only" == "services/api" ]; then
 890   rotate_logfile "$WORKSPACE/services/api/log/" "test.log"
 891   exit_cleanly
 892 fi
 893
 894 start_api || { stop_services; fatal "start_api"; }
 895
 896 test_ruby_sdk() {
 897     cd "$WORKSPACE/sdk/ruby" \
 898         && bundle exec rake test TESTOPTS=-v ${testargs[sdk/ruby]}
 899 }
 900 do_test sdk/ruby ruby_sdk
 901
 902 test_cli() {
 903     cd "$WORKSPACE/sdk/cli" \
 904         && mkdir -p /tmp/keep \
 905         && KEEP_LOCAL_STORE=/tmp/keep bundle exec rake test TESTOPTS=-v ${testargs[sdk/cli]}
 906 }
 907 do_test sdk/cli cli
 908
 909 test_login-sync() {
 910     cd "$WORKSPACE/services/login-sync" \
 911         && bundle exec rake test TESTOPTS=-v ${testargs[services/login-sync]}
 912 }
 913 do_test services/login-sync login-sync
 914
 915 test_nodemanager_integration() {
 916     cd "$WORKSPACE/services/nodemanager" \
 917         && tests/integration_test.py ${testargs[services/nodemanager_integration]}
 918 }
 919 do_test services/nodemanager_integration nodemanager_integration
 920
 921 for p in "${pythonstuff[@]}"
 922 do
 923     dir=${p%:py3}
 924     if [[ ${dir} = ${p} ]]; then
 925         if [[ -z ${skip[python2]} ]]; then
 926             do_test ${dir} pip
 927         fi
 928     elif [[ -n ${PYTHON3} ]]; then
 929         if [[ -z ${skip[python3]} ]]; then
 930             do_test ${dir} pip "$VENV3DIR/bin/"
 931         fi
 932     fi
 933 done
 934
 935 for g in "${gostuff[@]}"
 936 do
 937     do_test "$g" go
 938 done
 939
 940 test_workbench_units() {
 941     start_nginx_proxy_services \
 942         && cd "$WORKSPACE/apps/workbench" \
 943         && env RAILS_ENV=test ${short:+RAILS_TEST_SHORT=1} bundle exec rake test:units TESTOPTS=-v ${testargs[apps/workbench]}
 944 }
 945 do_test apps/workbench_units workbench_units
 946
 947 test_workbench_functionals() {
 948     start_nginx_proxy_services \
 949         && cd "$WORKSPACE/apps/workbench" \
 950         && env RAILS_ENV=test ${short:+RAILS_TEST_SHORT=1} bundle exec rake test:functionals TESTOPTS=-v ${testargs[apps/workbench]}
 951 }
 952 do_test apps/workbench_functionals workbench_functionals
 953
 954 test_workbench_integration() {
 955     start_nginx_proxy_services \
 956         && cd "$WORKSPACE/apps/workbench" \
 957         && env RAILS_ENV=test ${short:+RAILS_TEST_SHORT=1} bundle exec rake test:integration TESTOPTS=-v ${testargs[apps/workbench]}
 958 }
 959 do_test apps/workbench_integration workbench_integration
 960
 961
 962 test_workbench_benchmark() {
 963     start_nginx_proxy_services \
 964         && cd "$WORKSPACE/apps/workbench" \
 965         && env RAILS_ENV=test ${short:+RAILS_TEST_SHORT=1} bundle exec rake test:benchmark ${testargs[apps/workbench_benchmark]}
 966 }
 967 do_test apps/workbench_benchmark workbench_benchmark
 968
 969 test_workbench_profile() {
 970     start_nginx_proxy_services \
 971         && cd "$WORKSPACE/apps/workbench" \
 972         && env RAILS_ENV=test ${short:+RAILS_TEST_SHORT=1} bundle exec rake test:profile ${testargs[apps/workbench_profile]}
 973 }
 974 do_test apps/workbench_profile workbench_profile
 975
 976 exit_cleanly