mirror of
https://github.com/weaveworks/scope.git
synced 2026-03-03 18:20:27 +00:00
108 lines
2.9 KiB
Bash
Executable File
108 lines
2.9 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
set -e
|
|
|
|
DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
GO_TEST_ARGS=(-tags netgo -cpu 4 -timeout 8m)
|
|
SLOW=
|
|
NO_GO_GET=
|
|
|
|
usage() {
|
|
echo "$0 [-slow] [-in-container foo]"
|
|
}
|
|
|
|
while [ $# -gt 0 ]; do
|
|
case "$1" in
|
|
"-slow")
|
|
SLOW=true
|
|
shift 1
|
|
;;
|
|
"-no-go-get")
|
|
NO_GO_GET=true
|
|
shift 1
|
|
;;
|
|
*)
|
|
usage
|
|
exit 2
|
|
;;
|
|
esac
|
|
done
|
|
|
|
if [ -n "$SLOW" ] || [ -n "$CIRCLECI" ]; then
|
|
SLOW=true
|
|
fi
|
|
|
|
if [ -n "$SLOW" ]; then
|
|
GO_TEST_ARGS=("${GO_TEST_ARGS[@]}" -race -covermode=atomic)
|
|
|
|
# shellcheck disable=SC2153
|
|
if [ -n "$COVERDIR" ]; then
|
|
coverdir="$COVERDIR"
|
|
else
|
|
coverdir=$(mktemp -d coverage.XXXXXXXXXX)
|
|
fi
|
|
|
|
mkdir -p "$coverdir"
|
|
fi
|
|
|
|
fail=0
|
|
|
|
if [ -z "$TESTDIRS" ]; then
|
|
# NB: Relies on paths being prefixed with './'.
|
|
TESTDIRS=($(git ls-files -- '*_test.go' | grep -vE '^(vendor|prog|experimental)/' | xargs -n1 dirname | sort -u | sed -e 's|^|./|'))
|
|
else
|
|
# TESTDIRS on the right side is not really an array variable, it
|
|
# is just a string with spaces, but it is written like that to
|
|
# shut up the shellcheck tool.
|
|
TESTDIRS=($(for d in ${TESTDIRS[*]}; do echo "$d"; done))
|
|
fi
|
|
|
|
# If running on circle, use the scheduler to work out what tests to run on what shard
|
|
if [ -n "$CIRCLECI" ] && [ -z "$NO_SCHEDULER" ] && [ -x "$DIR/sched" ]; then
|
|
PREFIX=$(go list -e ./ | sed -e 's/\//-/g')
|
|
TESTDIRS=($(echo "${TESTDIRS[@]}" | "$DIR/sched" sched "$PREFIX-$CIRCLE_BUILD_NUM" "$CIRCLE_NODE_TOTAL" "$CIRCLE_NODE_INDEX"))
|
|
echo "${TESTDIRS[@]}"
|
|
fi
|
|
|
|
PACKAGE_BASE=$(go list -e ./)
|
|
|
|
# Speed up the tests by compiling and installing their dependencies first.
|
|
go test -i "${GO_TEST_ARGS[@]}" "${TESTDIRS[@]}"
|
|
|
|
for dir in "${TESTDIRS[@]}"; do
|
|
if [ -z "$NO_GO_GET" ]; then
|
|
go get -t -tags netgo "$dir"
|
|
fi
|
|
|
|
GO_TEST_ARGS_RUN=("${GO_TEST_ARGS[@]}")
|
|
if [ -n "$SLOW" ]; then
|
|
COVERPKGS=$( (
|
|
go list "$dir"
|
|
go list -f '{{join .Deps "\n"}}' "$dir" | grep -v "vendor" | grep "^$PACKAGE_BASE/"
|
|
) | paste -s -d, -)
|
|
output=$(mktemp "$coverdir/unit.XXXXXXXXXX")
|
|
GO_TEST_ARGS_RUN=("${GO_TEST_ARGS[@]}" -coverprofile=$output -coverpkg=$COVERPKGS)
|
|
fi
|
|
|
|
START=$(date +%s)
|
|
if ! go test "${GO_TEST_ARGS_RUN[@]}" "$dir"; then
|
|
fail=1
|
|
fi
|
|
RUNTIME=$(($(date +%s) - START))
|
|
|
|
# Report test runtime when running on circle, to help scheduler
|
|
if [ -n "$CIRCLECI" ] && [ -z "$NO_SCHEDULER" ] && [ -x "$DIR/sched" ]; then
|
|
"$DIR/sched" time "$dir" "$RUNTIME"
|
|
fi
|
|
done
|
|
|
|
if [ -n "$SLOW" ] && [ -z "$COVERDIR" ]; then
|
|
go get github.com/weaveworks/tools/cover
|
|
cover "$coverdir"/* >profile.cov
|
|
rm -rf "$coverdir"
|
|
go tool cover -html=profile.cov -o=coverage.html
|
|
go tool cover -func=profile.cov | tail -n1
|
|
fi
|
|
|
|
exit $fail
|