1brc/test_all.sh

80 lines
2.3 KiB
Bash
Raw Normal View History

#!/bin/bash
#
# Copyright 2023 The original authors
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
set -euo pipefail
INPUT=${1:-""}
if [ "$INPUT" = "-h" ] || [ "$#" -gt 1 ]; then
echo "Usage: ./test_all.sh [input file pattern]"
echo
echo "For each available fork run ./test.sh <fork name> [input file pattern]."
echo "Note that optional <input file pattern> should be quoted if contains wild cards."
echo
echo "Examples:"
echo "./test_all.sh"
echo "./test_all.sh 2>/dev/null"
echo "./test_all.sh src/test/resources/samples/measurements-1.txt"
echo "./test_all.sh 'src/test/resources/samples/measurements-*.txt'"
exit 1
fi
if [ -t 1 ]; then
GREEN='\033[0;32m'
RED='\033[0;31m'
RESET='\033[0m'
else
GREEN=""
RED=""
RESET=""
fi
WITH_TIMEOUT=""
if [ -x "$(command -v timeout)" ]; then
WITH_TIMEOUT="timeout -s KILL 5s"
elif [ -x "$(command -v gtimeout)" ]; then # MacOS from `brew install coreutils`
WITH_TIMEOUT="gtimeout -s KILL 5s"
else
echo "$0: timeout command not available, tests may run indefinitely long." 1>&2
fi
for impl in $(ls calculate_average_*.sh | sort); do
noext="${impl%%.sh}"
fork=${noext##calculate_average_}
# ./test.sh calls ./prepare_$fork.sh e.g. to build native image
# which may take some time.
# Here we run it upfront, assuming that prepare result is cached
# to avoid timeout due to long preparation.
if [ -f "./prepare_$fork.sh" ]; then
if ! output=$("./prepare_$fork.sh" 2>&1); then
echo "$output" 1>&2
echo "FAIL $fork"
continue
fi
fi
if output=$($WITH_TIMEOUT ./test.sh "$fork" "$INPUT" 2>&1); then
echo -e "${GREEN}PASS${RESET} $fork"
elif [ $? -eq 137 ]; then
echo -e "${RED}TIME${RESET} $fork"
else
echo "$output" 1>&2
echo -e "${RED}FAIL${RESET} $fork"
fi
done