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     echo -n 'graphviz: '
 237     dot -V || fatal "No graphviz. Try: apt-get install graphviz"
 238 }
 239
 240 rotate_logfile() {
 241   # i.e.  rotate_logfile "$WORKSPACE/apps/workbench/log/" "test.log"
 242   # $BUILD_NUMBER is set by Jenkins if this script is being called as part of a Jenkins run
 243   if [[ -f "$1/$2" ]]; then
 244     THEDATE=`date +%Y%m%d%H%M%S`
 245     mv "$1/$2" "$1/$THEDATE-$BUILD_NUMBER-$2"
 246     gzip "$1/$THEDATE-$BUILD_NUMBER-$2"
 247   fi
 248 }
 249
 250 declare -a failures
 251 declare -A skip
 252 declare -A testargs
 253 skip[apps/workbench_profile]=1
 254
 255 while [[ -n "$1" ]]
 256 do
 257     arg="$1"; shift
 258     case "$arg" in
 259         --help)
 260             echo >&2 "$helpmessage"
 261             echo >&2
 262             exit 1
 263             ;;
 264         --skip)
 265             skip[$1]=1; shift
 266             ;;
 267         --only)
 268             only="$1"; skip[$1]=""; shift
 269             ;;
 270         --short)
 271             short=1
 272             ;;
 273         --skip-install)
 274             only_install=nothing
 275             ;;
 276         --only-install)
 277             only_install="$1"; shift
 278             ;;
 279         --temp)
 280             temp="$1"; shift
 281             temp_preserve=1
 282             ;;
 283         --leave-temp)
 284             temp_preserve=1
 285             ;;
 286         --repeat)
 287             repeat=$((${1}+0)); shift
 288             ;;
 289         --retry)
 290             retry=1
 291             ;;
 292         *_test=*)
 293             suite="${arg%%_test=*}"
 294             args="${arg#*=}"
 295             testargs["$suite"]="$args"
 296             ;;
 297         *=*)
 298             eval export $(echo $arg | cut -d= -f1)=\"$(echo $arg | cut -d= -f2-)\"
 299             ;;
 300         *)
 301             echo >&2 "$0: Unrecognized option: '$arg'. Try: $0 --help"
 302             exit 1
 303             ;;
 304     esac
 305 done
 306
 307 start_api() {
 308     echo 'Starting API server...'
 309     if [[ ! -d "$WORKSPACE/services/api/log" ]]; then
 310         mkdir -p "$WORKSPACE/services/api/log"
 311     fi
 312     # Remove empty api.pid file if it exists
 313     if [[ -f "$WORKSPACE/tmp/api.pid" && ! -s "$WORKSPACE/tmp/api.pid" ]]; then
 314         rm -f "$WORKSPACE/tmp/api.pid"
 315     fi
 316     cd "$WORKSPACE" \
 317         && eval $(python sdk/python/tests/run_test_server.py start --auth admin) \
 318         && export ARVADOS_TEST_API_HOST="$ARVADOS_API_HOST" \
 319         && export ARVADOS_TEST_API_INSTALLED="$$" \
 320         && python sdk/python/tests/run_test_server.py start_ws \
 321         && python sdk/python/tests/run_test_server.py start_nginx \
 322         && (env | egrep ^ARVADOS)
 323 }
 324
 325 start_nginx_proxy_services() {
 326     echo 'Starting keepproxy, keep-web, ws, arv-git-httpd, and nginx ssl proxy...'
 327     cd "$WORKSPACE" \
 328         && python sdk/python/tests/run_test_server.py start_keep_proxy \
 329         && python sdk/python/tests/run_test_server.py start_keep-web \
 330         && python sdk/python/tests/run_test_server.py start_arv-git-httpd \
 331         && python sdk/python/tests/run_test_server.py start_ws \
 332         && python sdk/python/tests/run_test_server.py start_nginx \
 333         && export ARVADOS_TEST_PROXY_SERVICES=1
 334 }
 335
 336 stop_services() {
 337     if [[ -n "$ARVADOS_TEST_PROXY_SERVICES" ]]; then
 338         unset ARVADOS_TEST_PROXY_SERVICES
 339         cd "$WORKSPACE" \
 340             && python sdk/python/tests/run_test_server.py stop_nginx \
 341             && python sdk/python/tests/run_test_server.py stop_arv-git-httpd \
 342             && python sdk/python/tests/run_test_server.py stop_ws \
 343             && python sdk/python/tests/run_test_server.py stop_keep-web \
 344             && python sdk/python/tests/run_test_server.py stop_keep_proxy
 345     fi
 346     if [[ -n "$ARVADOS_TEST_API_HOST" ]]; then
 347         unset ARVADOS_TEST_API_HOST
 348         cd "$WORKSPACE" \
 349             && python sdk/python/tests/run_test_server.py stop_nginx \
 350             && python sdk/python/tests/run_test_server.py stop_ws \
 351             && python sdk/python/tests/run_test_server.py stop
 352     fi
 353 }
 354
 355 interrupt() {
 356     failures+=("($(basename $0) interrupted)")
 357     exit_cleanly
 358 }
 359 trap interrupt INT
 360
 361 sanity_checks
 362
 363 echo "WORKSPACE=$WORKSPACE"
 364
 365 if [[ -z "$CONFIGSRC" ]] && [[ -d "$HOME/arvados-api-server" ]]; then
 366     # Jenkins expects us to use this by default.
 367     CONFIGSRC="$HOME/arvados-api-server"
 368 fi
 369
 370 # Clean up .pyc files that may exist in the workspace
 371 cd "$WORKSPACE"
 372 find -name '*.pyc' -delete
 373
 374 if [[ -z "$temp" ]]; then
 375     temp="$(mktemp -d)"
 376 fi
 377
 378 # Set up temporary install dirs (unless existing dirs were supplied)
 379 for tmpdir in VENVDIR VENV3DIR GOPATH GEMHOME PERLINSTALLBASE
 380 do
 381     if [[ -z "${!tmpdir}" ]]; then
 382         eval "$tmpdir"="$temp/$tmpdir"
 383     fi
 384     if ! [[ -d "${!tmpdir}" ]]; then
 385         mkdir "${!tmpdir}" || fatal "can't create ${!tmpdir} (does $temp exist?)"
 386     fi
 387 done
 388
 389 setup_ruby_environment() {
 390     if [[ -s "$HOME/.rvm/scripts/rvm" ]] ; then
 391         source "$HOME/.rvm/scripts/rvm"
 392         using_rvm=true
 393     elif [[ -s "/usr/local/rvm/scripts/rvm" ]] ; then
 394         source "/usr/local/rvm/scripts/rvm"
 395         using_rvm=true
 396     else
 397         using_rvm=false
 398     fi
 399
 400     if [[ "$using_rvm" == true ]]; then
 401         # If rvm is in use, we can't just put separate "dependencies"
 402         # and "gems-under-test" paths to GEM_PATH: passenger resets
 403         # the environment to the "current gemset", which would lose
 404         # our GEM_PATH and prevent our test suites from running ruby
 405         # programs (for example, the Workbench test suite could not
 406         # boot an API server or run arv). Instead, we have to make an
 407         # rvm gemset and use it for everything.
 408
 409         [[ `type rvm | head -n1` == "rvm is a function" ]] \
 410             || fatal 'rvm check'
 411
 412         # Put rvm's favorite path back in first place (overriding
 413         # virtualenv, which just put itself there). Ignore rvm's
 414         # complaint about not being in first place already.
 415         rvm use @default 2>/dev/null
 416
 417         # Create (if needed) and switch to an @arvados-tests-* gemset,
 418         # salting the gemset name so it doesn't interfere with
 419         # concurrent builds in other workspaces. Leave the choice of
 420         # ruby to the caller.
 421         gemset="arvados-tests-$(echo -n "${WORKSPACE}" | md5sum | head -c16)"
 422         rvm use "@${gemset}" --create \
 423             || fatal 'rvm gemset setup'
 424
 425         rvm env
 426     else
 427         # When our "bundle install"s need to install new gems to
 428         # satisfy dependencies, we want them to go where "gem install
 429         # --user-install" would put them. (However, if the caller has
 430         # already set GEM_HOME, we assume that's where dependencies
 431         # should be installed, and we should leave it alone.)
 432
 433         if [ -z "$GEM_HOME" ]; then
 434             user_gempath="$(gem env gempath)"
 435             export GEM_HOME="${user_gempath%%:*}"
 436         fi
 437         PATH="$(gem env gemdir)/bin:$PATH"
 438
 439         # When we build and install our own gems, we install them in our
 440         # $GEMHOME tmpdir, and we want them to be at the front of GEM_PATH and
 441         # PATH so integration tests prefer them over other versions that
 442         # happen to be installed in $user_gempath, system dirs, etc.
 443
 444         tmpdir_gem_home="$(env - PATH="$PATH" HOME="$GEMHOME" gem env gempath | cut -f1 -d:)"
 445         PATH="$tmpdir_gem_home/bin:$PATH"
 446         export GEM_PATH="$tmpdir_gem_home"
 447
 448         echo "Will install dependencies to $(gem env gemdir)"
 449         echo "Will install arvados gems to $tmpdir_gem_home"
 450         echo "Gem search path is GEM_PATH=$GEM_PATH"
 451     fi
 452     bundle config || gem install bundler \
 453         || fatal 'install bundler'
 454 }
 455
 456 with_test_gemset() {
 457     if [[ "$using_rvm" == true ]]; then
 458         "$@"
 459     else
 460         GEM_HOME="$tmpdir_gem_home" GEM_PATH="$tmpdir_gem_home" "$@"
 461     fi
 462 }
 463
 464 gem_uninstall_if_exists() {
 465     if gem list "$1\$" | egrep '^\w'; then
 466         gem uninstall --force --all --executables "$1"
 467     fi
 468 }
 469
 470 setup_virtualenv() {
 471     local venvdest="$1"; shift
 472     if ! [[ -e "$venvdest/bin/activate" ]] || ! [[ -e "$venvdest/bin/pip" ]]; then
 473         virtualenv --setuptools "$@" "$venvdest" || fatal "virtualenv $venvdest failed"
 474     fi
 475     if [[ $("$venvdest/bin/python" --version 2>&1) =~ \ 3\.[012]\. ]]; then
 476         # pip 8.0.0 dropped support for python 3.2, e.g., debian wheezy
 477         "$venvdest/bin/pip" install 'setuptools>=18.5' 'pip>=7,<8'
 478     else
 479         "$venvdest/bin/pip" install 'setuptools>=18.5' 'pip>=7'
 480     fi
 481     # ubuntu1404 can't seem to install mock via tests_require, but it can do this.
 482     "$venvdest/bin/pip" install 'mock>=1.0' 'pbr<1.7.0'
 483 }
 484
 485 export PERLINSTALLBASE
 486 export PERLLIB="$PERLINSTALLBASE/lib/perl5:${PERLLIB:+$PERLLIB}"
 487
 488 export GOPATH
 489 mkdir -p "$GOPATH/src/git.curoverse.com"
 490 ln -sfT "$WORKSPACE" "$GOPATH/src/git.curoverse.com/arvados.git" \
 491     || fatal "symlink failed"
 492 go get -v github.com/kardianos/govendor \
 493     || fatal "govendor install failed"
 494
 495 setup_virtualenv "$VENVDIR" --python python2.7
 496 . "$VENVDIR/bin/activate"
 497
 498 # Needed for run_test_server.py which is used by certain (non-Python) tests.
 499 pip freeze 2>/dev/null | egrep ^PyYAML= \
 500     || pip install PyYAML >/dev/null \
 501     || fatal "pip install PyYAML failed"
 502
 503 # Preinstall forked version of libcloud, because nodemanager "pip install"
 504 # won't pick it up by default.
 505 pip freeze 2>/dev/null | egrep ^apache-libcloud==$LIBCLOUD_PIN \
 506     || pip install --pre --ignore-installed https://github.com/curoverse/libcloud/archive/apache-libcloud-$LIBCLOUD_PIN.zip >/dev/null \
 507     || fatal "pip install apache-libcloud failed"
 508
 509 # We need an unreleased (as of 2017-08-17) llfuse bugfix, otherwise our fuse test suite deadlocks.
 510 pip freeze | grep -x llfuse==1.2.0 || (
 511     set -e
 512     yes | pip uninstall llfuse || true
 513     cython --version || fatal "no cython; try sudo apt-get install cython"
 514     cd "$temp"
 515     (cd python-llfuse 2>/dev/null || git clone https://github.com/curoverse/python-llfuse)
 516     cd python-llfuse
 517     git checkout 620722fd990ea642ddb8e7412676af482c090c0c
 518     git checkout setup.py
 519     sed -i -e "s:'1\\.2':'1.2.0':" setup.py
 520     python setup.py build_cython
 521     python setup.py install --force
 522 ) || fatal "llfuse fork failed"
 523 pip freeze | grep -x llfuse==1.2.0 || fatal "error: installed llfuse 1.2.0 but '$(pip freeze | grep llfuse)' ???"
 524
 525 # Deactivate Python 2 virtualenv
 526 deactivate
 527
 528 declare -a pythonstuff
 529 pythonstuff=(
 530     sdk/pam
 531     sdk/python
 532     sdk/python:py3
 533     sdk/cwl
 534     services/dockercleaner:py3
 535     services/fuse
 536     services/nodemanager
 537     tools/crunchstat-summary
 538     )
 539
 540 # If Python 3 is available, set up its virtualenv in $VENV3DIR.
 541 # Otherwise, skip dependent tests.
 542 PYTHON3=$(which python3)
 543 if [[ ${?} = 0 ]]; then
 544     setup_virtualenv "$VENV3DIR" --python python3
 545 else
 546     PYTHON3=
 547     cat >&2 <<EOF
 548
 549 Warning: python3 could not be found. Python 3 tests will be skipped.
 550
 551 EOF
 552 fi
 553
 554 # Reactivate Python 2 virtualenv
 555 . "$VENVDIR/bin/activate"
 556
 557 # Note: this must be the last time we change PATH, otherwise rvm will
 558 # whine a lot.
 559 setup_ruby_environment
 560
 561 echo "PATH is $PATH"
 562
 563 if ! which bundler >/dev/null
 564 then
 565     gem install --user-install bundler || fatal 'Could not install bundler'
 566 fi
 567
 568 retry() {
 569     remain="${repeat}"
 570     while :
 571     do
 572         if ${@}; then
 573             if [[ "$remain" -gt 1 ]]; then
 574                 remain=$((${remain}-1))
 575                 title "Repeating ${remain} more times"
 576             else
 577                 break
 578             fi
 579         elif [[ "$retry" == 1 ]]; then
 580             read -p 'Try again? [Y/n] ' x
 581             if [[ "$x" != "y" ]] && [[ "$x" != "" ]]
 582             then
 583                 break
 584             fi
 585         else
 586             break
 587         fi
 588     done
 589 }
 590
 591 do_test() {
 592     case "${1}" in
 593         apps/workbench_units | apps/workbench_functionals | apps/workbench_integration)
 594             suite=apps/workbench
 595             ;;
 596         services/nodemanager | services/nodemanager_integration)
 597             suite=services/nodemanager_suite
 598             ;;
 599         *)
 600             suite="${1}"
 601             ;;
 602     esac
 603     if [[ -z "${skip[$suite]}" && -z "${skip[$1]}" && \
 604                 (-z "${only}" || "${only}" == "${suite}" || \
 605                  "${only}" == "${1}") ]]; then
 606         retry do_test_once ${@}
 607     else
 608         title "Skipping ${1} tests"
 609     fi
 610 }
 611
 612 do_test_once() {
 613     unset result
 614
 615     title "Running $1 tests"
 616     timer_reset
 617     if [[ "$2" == "go" ]]
 618     then
 619         covername="coverage-$(echo "$1" | sed -e 's/\//_/g')"
 620         coverflags=("-covermode=count" "-coverprofile=$WORKSPACE/tmp/.$covername.tmp")
 621         # We do "go get -t" here to catch compilation errors
 622         # before trying "go test". Otherwise, coverage-reporting
 623         # mode makes Go show the wrong line numbers when reporting
 624         # compilation errors.
 625         go get -t "git.curoverse.com/arvados.git/$1" && \
 626             cd "$GOPATH/src/git.curoverse.com/arvados.git/$1" && \
 627             [[ -z "$(gofmt -e -d . | tee /dev/stderr)" ]] && \
 628             if [[ -n "${testargs[$1]}" ]]
 629         then
 630             # "go test -check.vv giturl" doesn't work, but this
 631             # does:
 632             go test ${short:+-short} ${testargs[$1]}
 633         else
 634             # The above form gets verbose even when testargs is
 635             # empty, so use this form in such cases:
 636             go test ${short:+-short} ${coverflags[@]} "git.curoverse.com/arvados.git/$1"
 637         fi
 638         result=${result:-$?}
 639         if [[ -f "$WORKSPACE/tmp/.$covername.tmp" ]]
 640         then
 641             go tool cover -html="$WORKSPACE/tmp/.$covername.tmp" -o "$WORKSPACE/tmp/$covername.html"
 642             rm "$WORKSPACE/tmp/.$covername.tmp"
 643         fi
 644     elif [[ "$2" == "pip" ]]
 645     then
 646         tries=0
 647         cd "$WORKSPACE/$1" && while :
 648         do
 649             tries=$((${tries}+1))
 650             # $3 can name a path directory for us to use, including trailing
 651             # slash; e.g., the bin/ subdirectory of a virtualenv.
 652             "${3}python" setup.py ${short:+--short-tests-only} test ${testargs[$1]}
 653             result=$?
 654             if [[ ${tries} < 3 && ${result} == 137 ]]
 655             then
 656                 printf '\n*****\n%s tests killed -- retrying\n*****\n\n' "$1"
 657                 continue
 658             else
 659                 break
 660             fi
 661         done
 662     elif [[ "$2" != "" ]]
 663     then
 664         "test_$2"
 665     else
 666         "test_$1"
 667     fi
 668     result=${result:-$?}
 669     checkexit $result "$1 tests"
 670     title "End of $1 tests (`timer`)"
 671     return $result
 672 }
 673
 674 do_install() {
 675     if [[ -z "${only_install}" || "${only_install}" == "${1}" ]]; then
 676         retry do_install_once ${@}
 677     else
 678         title "Skipping $1 install"
 679     fi
 680 }
 681
 682 do_install_once() {
 683     title "Running $1 install"
 684     timer_reset
 685     if [[ "$2" == "go" ]]
 686     then
 687         go get -t "git.curoverse.com/arvados.git/$1"
 688     elif [[ "$2" == "pip" ]]
 689     then
 690         # $3 can name a path directory for us to use, including trailing
 691         # slash; e.g., the bin/ subdirectory of a virtualenv.
 692
 693         # Need to change to a different directory after creating
 694         # the source dist package to avoid a pip bug.
 695         # see https://arvados.org/issues/5766 for details.
 696
 697         # Also need to install twice, because if it believes the package is
 698         # already installed, pip it won't install it.  So the first "pip
 699         # install" ensures that the dependencies are met, the second "pip
 700         # install" ensures that we've actually installed the local package
 701         # we just built.
 702         cd "$WORKSPACE/$1" \
 703             && "${3}python" setup.py sdist rotate --keep=1 --match .tar.gz \
 704             && cd "$WORKSPACE" \
 705             && "${3}pip" install --quiet "$WORKSPACE/$1/dist"/*.tar.gz \
 706             && "${3}pip" install --quiet --no-deps --ignore-installed "$WORKSPACE/$1/dist"/*.tar.gz
 707     elif [[ "$2" != "" ]]
 708     then
 709         "install_$2"
 710     else
 711         "install_$1"
 712     fi
 713     result=$?
 714     checkexit $result "$1 install"
 715     title "End of $1 install (`timer`)"
 716     return $result
 717 }
 718
 719 bundle_install_trylocal() {
 720     (
 721         set -e
 722         echo "(Running bundle install --local. 'could not find package' messages are OK.)"
 723         if ! bundle install --local --no-deployment; then
 724             echo "(Running bundle install again, without --local.)"
 725             bundle install --no-deployment
 726         fi
 727         bundle package --all
 728     )
 729 }
 730
 731 install_doc() {
 732     cd "$WORKSPACE/doc" \
 733         && bundle_install_trylocal \
 734         && rm -rf .site
 735 }
 736 do_install doc
 737
 738 install_gem() {
 739     gemname=$1
 740     srcpath=$2
 741     with_test_gemset gem_uninstall_if_exists "$gemname" \
 742         && cd "$WORKSPACE/$srcpath" \
 743         && bundle_install_trylocal \
 744         && gem build "$gemname.gemspec" \
 745         && with_test_gemset gem install --no-ri --no-rdoc $(ls -t "$gemname"-*.gem|head -n1)
 746 }
 747
 748 install_ruby_sdk() {
 749     install_gem arvados sdk/ruby
 750 }
 751 do_install sdk/ruby ruby_sdk
 752
 753 install_perl_sdk() {
 754     cd "$WORKSPACE/sdk/perl" \
 755         && perl Makefile.PL INSTALL_BASE="$PERLINSTALLBASE" \
 756         && make install INSTALLDIRS=perl
 757 }
 758 do_install sdk/perl perl_sdk
 759
 760 install_cli() {
 761     install_gem arvados-cli sdk/cli
 762 }
 763 do_install sdk/cli cli
 764
 765 install_login-sync() {
 766     install_gem arvados-login-sync services/login-sync
 767 }
 768 do_install services/login-sync login-sync
 769
 770 # Install the Python SDK early. Various other test suites (like
 771 # keepproxy) bring up run_test_server.py, which imports the arvados
 772 # module. We can't actually *test* the Python SDK yet though, because
 773 # its own test suite brings up some of those other programs (like
 774 # keepproxy).
 775 for p in "${pythonstuff[@]}"
 776 do
 777     dir=${p%:py3}
 778     if [[ ${dir} = ${p} ]]; then
 779         if [[ -z ${skip[python2]} ]]; then
 780             do_install ${dir} pip
 781         fi
 782     elif [[ -n ${PYTHON3} ]]; then
 783         if [[ -z ${skip[python3]} ]]; then
 784             do_install ${dir} pip "$VENV3DIR/bin/"
 785         fi
 786     fi
 787 done
 788
 789 install_apiserver() {
 790     cd "$WORKSPACE/services/api" \
 791         && RAILS_ENV=test bundle_install_trylocal
 792
 793     rm -f config/environments/test.rb
 794     cp config/environments/test.rb.example config/environments/test.rb
 795
 796     if [ -n "$CONFIGSRC" ]
 797     then
 798         for f in database.yml
 799         do
 800             cp "$CONFIGSRC/$f" config/ || fatal "$f"
 801         done
 802     fi
 803
 804     # Clear out any lingering postgresql connections to the test
 805     # database, so that we can drop it. This assumes the current user
 806     # is a postgresql superuser.
 807     cd "$WORKSPACE/services/api" \
 808         && test_database=$(python -c "import yaml; print yaml.load(file('config/database.yml'))['test']['database']") \
 809         && 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
 810
 811     mkdir -p "$WORKSPACE/services/api/tmp/pids"
 812
 813     cd "$WORKSPACE/services/api" \
 814         && RAILS_ENV=test bundle exec rake db:drop \
 815         && RAILS_ENV=test bundle exec rake db:setup \
 816         && RAILS_ENV=test bundle exec rake db:fixtures:load
 817 }
 818 do_install services/api apiserver
 819
 820 cd "$GOPATH/src/git.curoverse.com/arvados.git" && \
 821     "$GOPATH/bin/govendor" sync -v || \
 822         fatal "govendor sync failed"
 823 declare -a gostuff
 824 gostuff=(
 825     sdk/go/arvados
 826     sdk/go/arvadosclient
 827     sdk/go/blockdigest
 828     sdk/go/dispatch
 829     sdk/go/health
 830     sdk/go/httpserver
 831     sdk/go/manifest
 832     sdk/go/streamer
 833     sdk/go/crunchrunner
 834     sdk/go/stats
 835     lib/crunchstat
 836     services/arv-git-httpd
 837     services/crunchstat
 838     services/health
 839     services/keep-web
 840     services/keepstore
 841     sdk/go/keepclient
 842     services/keep-balance
 843     services/keepproxy
 844     services/crunch-dispatch-local
 845     services/crunch-dispatch-slurm
 846     services/crunch-run
 847     services/ws
 848     tools/keep-block-check
 849     tools/keep-exercise
 850     tools/keep-rsync
 851     tools/arv-sync-groups
 852 )
 853 for g in "${gostuff[@]}"
 854 do
 855     do_install "$g" go
 856 done
 857
 858 install_workbench() {
 859     cd "$WORKSPACE/apps/workbench" \
 860         && mkdir -p tmp/cache \
 861         && RAILS_ENV=test bundle_install_trylocal \
 862         && RAILS_ENV=test RAILS_GROUPS=assets bundle exec rake npm:install
 863 }
 864 do_install apps/workbench workbench
 865
 866 unset http_proxy https_proxy no_proxy
 867
 868 test_doclinkchecker() {
 869     (
 870         set -e
 871         cd "$WORKSPACE/doc"
 872         ARVADOS_API_HOST=qr1hi.arvadosapi.com
 873         # Make sure python-epydoc is installed or the next line won't
 874         # do much good!
 875         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
 876     )
 877 }
 878 do_test doc doclinkchecker
 879
 880 stop_services
 881
 882 test_apiserver() {
 883     rm -f "$WORKSPACE/services/api/git-commit.version"
 884     cd "$WORKSPACE/services/api" \
 885         && env RAILS_ENV=test ${short:+RAILS_TEST_SHORT=1} bundle exec rake test TESTOPTS=-v ${testargs[services/api]}
 886 }
 887 do_test services/api apiserver
 888
 889 # Shortcut for when we're only running apiserver tests. This saves a bit of time,
 890 # because we don't need to start up the api server for subsequent tests.
 891 if [ ! -z "$only" ] && [ "$only" == "services/api" ]; then
 892   rotate_logfile "$WORKSPACE/services/api/log/" "test.log"
 893   exit_cleanly
 894 fi
 895
 896 start_api || { stop_services; fatal "start_api"; }
 897
 898 test_ruby_sdk() {
 899     cd "$WORKSPACE/sdk/ruby" \
 900         && bundle exec rake test TESTOPTS=-v ${testargs[sdk/ruby]}
 901 }
 902 do_test sdk/ruby ruby_sdk
 903
 904 test_cli() {
 905     cd "$WORKSPACE/sdk/cli" \
 906         && mkdir -p /tmp/keep \
 907         && KEEP_LOCAL_STORE=/tmp/keep bundle exec rake test TESTOPTS=-v ${testargs[sdk/cli]}
 908 }
 909 do_test sdk/cli cli
 910
 911 test_login-sync() {
 912     cd "$WORKSPACE/services/login-sync" \
 913         && bundle exec rake test TESTOPTS=-v ${testargs[services/login-sync]}
 914 }
 915 do_test services/login-sync login-sync
 916
 917 test_nodemanager_integration() {
 918     cd "$WORKSPACE/services/nodemanager" \
 919         && tests/integration_test.py ${testargs[services/nodemanager_integration]}
 920 }
 921 do_test services/nodemanager_integration nodemanager_integration
 922
 923 for p in "${pythonstuff[@]}"
 924 do
 925     dir=${p%:py3}
 926     if [[ ${dir} = ${p} ]]; then
 927         if [[ -z ${skip[python2]} ]]; then
 928             do_test ${dir} pip
 929         fi
 930     elif [[ -n ${PYTHON3} ]]; then
 931         if [[ -z ${skip[python3]} ]]; then
 932             do_test ${dir} pip "$VENV3DIR/bin/"
 933         fi
 934     fi
 935 done
 936
 937 for g in "${gostuff[@]}"
 938 do
 939     do_test "$g" go
 940 done
 941
 942 test_workbench_units() {
 943     start_nginx_proxy_services \
 944         && cd "$WORKSPACE/apps/workbench" \
 945         && env RAILS_ENV=test ${short:+RAILS_TEST_SHORT=1} bundle exec rake test:units TESTOPTS=-v ${testargs[apps/workbench]}
 946 }
 947 do_test apps/workbench_units workbench_units
 948
 949 test_workbench_functionals() {
 950     start_nginx_proxy_services \
 951         && cd "$WORKSPACE/apps/workbench" \
 952         && env RAILS_ENV=test ${short:+RAILS_TEST_SHORT=1} bundle exec rake test:functionals TESTOPTS=-v ${testargs[apps/workbench]}
 953 }
 954 do_test apps/workbench_functionals workbench_functionals
 955
 956 test_workbench_integration() {
 957     start_nginx_proxy_services \
 958         && cd "$WORKSPACE/apps/workbench" \
 959         && env RAILS_ENV=test ${short:+RAILS_TEST_SHORT=1} bundle exec rake test:integration TESTOPTS=-v ${testargs[apps/workbench]}
 960 }
 961 do_test apps/workbench_integration workbench_integration
 962
 963
 964 test_workbench_benchmark() {
 965     start_nginx_proxy_services \
 966         && cd "$WORKSPACE/apps/workbench" \
 967         && env RAILS_ENV=test ${short:+RAILS_TEST_SHORT=1} bundle exec rake test:benchmark ${testargs[apps/workbench_benchmark]}
 968 }
 969 do_test apps/workbench_benchmark workbench_benchmark
 970
 971 test_workbench_profile() {
 972     start_nginx_proxy_services \
 973         && cd "$WORKSPACE/apps/workbench" \
 974         && env RAILS_ENV=test ${short:+RAILS_TEST_SHORT=1} bundle exec rake test:profile ${testargs[apps/workbench_profile]}
 975 }
 976 do_test apps/workbench_profile workbench_profile
 977
 978 exit_cleanly