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