aboutsummaryrefslogtreecommitdiff
path: root/tests/vm-check.sh
blob: 695e4d75dffbd43e287cc9edb099d847925a0348 (about) (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
#!/bin/sh
set -eu

usage() {
	cat <<-'EOF'
		Usage:
		  vm-check.sh [-x] -t TYPE -H HOST -r REMOTE [-- ARGS]
		  vm-check.sh -h
	EOF
}

help() {
	cat <<-'EOF'


		Options:
		  -x            shutdown the VM before running checks
		  -t TYPE       the type of VM being tested
		  -H HOST       the name of the host to SSH into
		  -r REMOTE     the base address of the remote package-repository
		  -h, --help    show this message



		Run end-to-end tests of the package repository.
	EOF
}


for flag in "$@"; do
	case "$flag" in
		--)
			break
			;;
		--help)
			usage
			help
			exit
			;;
		*)
			;;
	esac
done

SHUTDOWN=false
while getopts 'xt:H:r:h' flag; do
	case "$flag" in
		x)
			SHUTDOWN=true
			;;
		t)
			TYPE="$OPTARG"
			;;
		H)
			HOST="$OPTARG"
			;;
		r)
			REMOTE="$OPTARG"
			;;
		h)
			usage
			help
			exit
			;;
		*)
			usage >&2
			exit 2
			;;
	esac
done
shift $((OPTIND - 1))

eval "$(assert-arg -- "${TYPE:-}"   '-t TYPE')"
eval "$(assert-arg -- "${HOST:-}"   '-H HOST')"
eval "$(assert-arg -- "${REMOTE:-}" '-r REMOTE')"


SEP="$(color -c blacki ':')"

pre() {
        # Same as:
        # sed -u "s|^|[$CMD]: |"
        # but the "-u" option is not POSIX
        IFS=''
        while read -r line; do
                printf '%s%s\n' "$1" "$line"
        done
}

pre_guest() {
	pre "$(color -c green "$HOST")$SEP$(color -c yellow "$TYPE")$SEP "
}

pre_host() {
	pre "$(color -c blueb "$0")$SEP "
}

STATUS_F="$(mkstemp)"
shutdown() {
	STATUS="$(cat "$STATUS_F" || echo 1)"
	rm -f "$STATUS_F"
	if [ "$SHUTDOWN" = true ]; then
		vm down "$HOST" 2>&1 | pre_host
	fi
	exit "$STATUS"
}
trap shutdown EXIT

{

	{
		if [ "$SHUTDOWN" = true ]; then
			vm down "$HOST"
		fi
		vm up "$HOST"
		untill -m5 -n5 ssh "$HOST" uptime
		scp tests/checks/"$TYPE".sh "$HOST":check.sh
	} 2>&1 | pre_host

	{
		STATUS=0
		# shellcheck disable=2029
		ssh "$HOST" sh check.sh "$@" "$REMOTE" || STATUS=$?
		printf '%s' "$STATUS" > "$STATUS_F"
	} 2>&1 | pre_guest

} | ts -s '%.s'