2018-03-21 20:22:57 +01:00

115 lines
2.9 KiB
Bash
Executable File

#!/bin/bash
# Colorizing Go test output:
# This is meant to be used in place of `go test`. Provided this script is in
# your PATH, calling `color-go-test` will call through to `go test` and then
# colorize and reformat the output.
RED=$(tput setaf 1)
GREEN=$(tput setaf 2)
YELLOW=$(tput setaf 3)
COLOR_RESET=$(tput sgr0)
BOLD=$(tput bold)
previous_line_fail=false
verbose_output=false
verbose_flag_prefix="-v "
pass_count=0
fail_count=0
errors=()
echo_last_line() {
local time_string=$1
local color=$GREEN
if [ $verbose_output = false ]; then
echo -e "\n"
if [ ${#errors[@]} -gt 0 ]; then
for error in "${errors[@]}"; do
echo -e "$error"
done
fi
fi
if [ $fail_count -gt 0 ]; then
color=$RED
fi
local num_tests=$((pass_count + fail_count))
echo -e "\n${color}${BOLD}$num_tests tests, $fail_count failure, run time ($time_string)${COLOR_RESET}"
}
colorize_output() {
while read line; do
if echo $line | grep --quiet '^FAIL$'; then
continue
elif echo $line | grep --quiet '^PASS$'; then
continue
elif echo $line | grep --quiet '^=== RUN'; then
continue
elif echo $line | grep --quiet '^exit status 1$'; then
continue
elif echo $line | grep --quiet 'FAIL'; then
if echo $line | grep --quiet "\-\-\- FAIL:"; then
fail_count=$((fail_count + 1))
error_message="${RED}$(echo $line | sed 's/--- FAIL:/✗/')${COLOR_RESET}"
if [ $verbose_output = true ]; then
echo $error_message
else
errors+=("$error_message")
printf "${RED}.${COLOR_RESET}"
fi
previous_line_fail=true
else
local test_run_time=$(echo $line | grep -o '[0-9]*\.[0-9]*s$')
echo_last_line $test_run_time
previous_line_fail=false
fi
elif [ $previous_line_fail = true ]; then
error_message=" ${YELLOW}$line${COLOR_RESET}"
if [ $verbose_output = true ]; then
echo -e "$error_message"
else
errors+=("$error_message")
fi
previous_line_fail=false
elif echo $line | grep --quiet 'PASS'; then
if echo $line | grep --quiet "\-\-\- PASS:"; then
if [ $verbose_output = true ]; then
echo "${GREEN}$(echo $line | sed 's/--- PASS:/✔/')${COLOR_RESET}"
else
printf "${GREEN}.${COLOR_RESET}"
fi
pass_count=$((pass_count + 1))
else
local test_run_time=$(echo $line | grep -o '[0-9]*\.[0-9]*s$')
echo_last_line $test_run_time
fi
previous_line_fail=false
elif echo $line | grep --quiet '^ok '; then
local test_run_time=$(echo $line | grep -o '[0-9]*\.[0-9]*s$')
echo_last_line $test_run_time
previous_line_fail=false
else
echo $line
previous_line_fail=false
fi
done
}
for flag in $@; do
if [ "$flag" = "-v" ]; then
verbose_output=true
verbose_flag_prefix=""
fi
done
go test ${verbose_flag_prefix}$@ | colorize_output