headscale/integration/run.sh
Mike Poindexter 3101f895a7
Fix 764 (#2093)
* Fix KeyExpiration when a zero time value has a timezone

When a zero time value is loaded from JSON or a DB in a way that
assigns it the local timezone, it does not roudtrip in JSON as a
value for which IsZero returns true. This causes KeyExpiry to be
treated as a far past value instead of a nilish value.

See https://github.com/golang/go/issues/57040

* Fix whitespace

* Ensure that postgresql is used for all tests when env var is set

* Pass through value of HEADSCALE_INTEGRATION_POSTGRES env var

* Add option to set timezone on headscale container

* Add test for registration with auth key in alternate timezone
2024-09-03 09:22:17 +02:00

77 lines
1.7 KiB
Bash
Executable File

#!/usr/bin/env ksh
run_tests() {
test_name=$1
num_tests=$2
success_count=0
failure_count=0
runtimes=()
echo "-------------------"
echo "Running Tests for $test_name"
for ((i = 1; i <= num_tests; i++)); do
docker network prune -f >/dev/null 2>&1
docker rm headscale-test-suite >/dev/null 2>&1 || true
docker kill "$(docker ps -q)" >/dev/null 2>&1 || true
echo "Run $i"
start=$(date +%s)
docker run \
--tty --rm \
--volume ~/.cache/hs-integration-go:/go \
--name headscale-test-suite \
--volume "$PWD:$PWD" -w "$PWD"/integration \
--volume /var/run/docker.sock:/var/run/docker.sock \
--volume "$PWD"/control_logs:/tmp/control \
-e "HEADSCALE_INTEGRATION_POSTGRES" \
golang:1 \
go test ./... \
-failfast \
-timeout 120m \
-parallel 1 \
-run "^$test_name\$" >./control_logs/"$test_name"_"$i".log 2>&1
status=$?
end=$(date +%s)
runtime=$((end - start))
runtimes+=("$runtime")
if [ "$status" -eq 0 ]; then
((success_count++))
else
((failure_count++))
fi
done
echo "-------------------"
echo "Test Summary for $test_name"
echo "-------------------"
echo "Total Tests: $num_tests"
echo "Successful Tests: $success_count"
echo "Failed Tests: $failure_count"
echo "Runtimes in seconds: ${runtimes[*]}"
echo
}
# Check if both arguments are provided
if [ $# -ne 2 ]; then
echo "Usage: $0 <test_name> <num_tests>"
exit 1
fi
test_name=$1
num_tests=$2
docker network prune -f
if [ "$test_name" = "all" ]; then
rg --regexp "func (Test.+)\(.*" ./integration/ --replace '$1' --no-line-number --no-filename --no-heading | sort | while read -r test_name; do
run_tests "$test_name" "$num_tests"
done
else
run_tests "$test_name" "$num_tests"
fi