add script to run integration tests

Signed-off-by: Kristoffer Dalby <[email protected]>

flake.nix

@@ -96,6 +96,7 @@
           nfpm
           gotestsum
           gotests
+          ksh
 
           # 'dot' is needed for pprof graphs
           # go tool pprof -http=: <source>

hscontrol/poll.go

@@ -64,8 +64,6 @@ func (h *Headscale) handlePoll(
 ) {
 	logInfo, logErr := logPollFunc(mapRequest, machine, isNoise)
 
-	// TODO(kradalby): This is a stepping stone, mapper should be initiated once
-	// per client or something similar
 	mapp := mapper.NewMapper(
 		h.db,
 		h.privateKey2019,

integration/run.sh

@@ -0,0 +1,72 @@
+#!/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
+		ocker network prune -f >/dev/null 2>&1
+
+		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 \
+			golang:1 \
+			go test ./... \
+			-tags ts2019 \
+			-failfast \
+			-timeout 120m \
+			-parallel 1 \
+			-run "^$test_name\$" >/dev/null 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