Nicer local testing (pwncollege#383)

* helper script for local testing

* fixes

* minor updates

* fixes

* document

* move into test dir

README.md

@@ -148,3 +148,5 @@ Have a small update?
 Send a PR so everyone can benefit.
 For more substantial changes, open an issue to ensure we're on the same page.
 Together, we make this project better for all! 🚀
+
+You can run the dojo CI testcases locally using `test/local-tester.sh`.

test/local-tester.sh

@@ -0,0 +1,70 @@
+#!/bin/bash -ex
+
+cd $(dirname "${BASH_SOURCE[0]}")/..
+
+function usage {
+	set +x
+	echo "Usage: $0 [-r DB_BACKUP ] [ -c CONTAINER_NAME ] [ -T ]"
+	echo ""
+	echo "	-r	db backup to restore (relative to dojo/data/backups)"
+	echo "	-c	the name of the dojo container (default: dojo-test)"
+	echo "	-D	use a blank data volume (builds everything from scratch)"
+	echo "	-T	don't run tests"
+	exit
+}
+
+VOLUME_ARGS=("-v" "$PWD:/opt/pwn.college:shared")
+ENV_ARGS=( )
+DB_RESTORE=""
+CONTAINER_NAME=dojo-test
+TEST=yes
+while getopts "r:c:he:TD" OPT
+do
+	case $OPT in
+		r) DB_RESTORE="$OPTARG" ;;
+		c) CONTAINER_NAME="$OPTARG" ;;
+		T) TEST=no ;;
+		D)
+			DATA_DIR=$(mktemp -d)
+			VOLUME_ARGS+=("-v" "$DATA_DIR:/opt/pwn.college/data:shared")
+			;;
+		e) ENV_ARGS+=("-e" "$OPTARG") ;;
+		h) usage ;;
+		?)
+			OPTIND=$(($OPTIND-1))
+			break
+			;;
+	esac
+done
+shift $((OPTIND-1))
+
+[ "${#VOLUME_ARGS[@]}" -eq 2 ] && VOLUME_ARGS+=(
+	"-v" "/opt/pwn.college/data/dojos"
+	"-v" "/opt/pwn.college/data/mysql"
+)
+
+export CONTAINER_NAME
+docker kill "$CONTAINER_NAME" 2>/dev/null || echo "No $CONTAINER_NAME container to kill."
+docker rm "$CONTAINER_NAME" 2>/dev/null || echo "No $CONTAINER_NAME container to remove."
+while docker ps -a | grep "$CONTAINER_NAME"; do sleep 1; done
+
+# freaking bad unmount
+sleep 1
+mount | grep $PWD | while read -a ENTRY
+do
+	sudo umount "${ENTRY[2]}"
+done
+
+docker run --rm --privileged -d "${VOLUME_ARGS[@]}" "${ENV_ARGS[@]}" -p 2222:22 -p 80:80 -p 443:443 --name "$CONTAINER_NAME" dojo || exit 1
+
+# fix the insane routing thing
+read -a GW <<<$(ip route show default)
+read -a NS <<<$(docker exec "$CONTAINER_NAME" cat /etc/resolv.conf | grep nameserver)
+docker exec "$CONTAINER_NAME" ip route add "${GW[2]}" via 172.17.0.1
+docker exec "$CONTAINER_NAME" ip route add "${NS[1]}" via 172.17.0.1 || echo "Failed to add nameserver route"
+
+docker exec "$CONTAINER_NAME" dojo wait
+[ -n "$DB_RESTORE" ] && until docker exec "$CONTAINER_NAME" dojo restore $DB_RESTORE; do sleep 1; done
+
+until curl -s localhost.pwn.college | grep -q pwn; do sleep 1; done
+[ "$TEST" == "yes" ] && MOZ_HEADLESS=1 pytest -v test/test_running.py test/test_welcome.py