blob: 86084cf8922095fca198f0092c18524c57f609dc (
plain)
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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
|
#!/bin/bash
WHITE="$(tput setaf 9 2>/dev/null || echo -n '')"
YELLOW="$(tput setaf 3 2>/dev/null || echo -n '')"
GREEN="$(tput setaf 2 2>/dev/null || echo -n '')"
RED="$(tput setaf 1 2>/dev/null || echo -n '')"
OFFSET=( )
STEP=" "
function with_program () {
local program="${1:?}"
hash "$program" &>/dev/null || {
function expect_run () {
echo 1>&2 "${WHITE} - skipped (missing program)"
}
function expect_run_sh () {
echo 1>&2 "${WHITE} - skipped (missing program)"
}
}
}
function title () {
echo "$WHITE-----------------------------------------------------"
echo "${GREEN}$*"
echo "$WHITE-----------------------------------------------------"
}
function _context () {
local name="${1:?}"
shift
echo 1>&2 "${YELLOW}${OFFSET[*]:-}[$name] $*"
OFFSET+=("$STEP")
}
function step () {
_note step "${WHITE}" "$*"
}
function stepn () {
step "$*" $'\n'
}
function with () {
_context with "$*"
}
function when () {
_context when "$*"
}
function _note () {
local name="${1:?}"
local color="${2:-}"
shift 2
echo 1>&2 -n "${OFFSET[*]:-}${color}[$name] ${*// /}"
}
function it () {
_note it "${GREEN}" "$*"
}
function precondition () {
_note precondition "${WHITE}" "$*"
}
function shortcoming () {
_note shortcoming "${RED}" "$*"
}
function step () {
_note step "${WHITE}" "$*"
}
function stepn () {
step "$*" $'\n'
}
function fail () {
echo 1>&2 "${RED} $*"
exit 1
}
function sandbox () {
sandbox_tempdir="$(mktemp -t sandbox.XXXXXX -d)"
# shellcheck disable=2064
trap "popd >/dev/null" EXIT
pushd "$sandbox_tempdir" >/dev/null \
|| fail "Could not change directory into temporary directory."
local custom_init="${1:-}"
if [ -n "$custom_init" ]; then
eval "$custom_init"
fi
}
function expect_equals () {
expect_run 0 test "${1:?}" = "${2:?}"
}
function expect_exists () {
expect_run 0 test -e "${1:?}"
}
function expect_run_sh () {
expect_run "${1:?}" bash -c -eu -o pipefail "${2:?}"
}
function expect_snapshot () {
local expected=${1:?}
local actual=${2:?}
if ! [ -e "$expected" ]; then
mkdir -p "${expected%/*}"
cp -R "$actual" "$expected"
fi
expect_run 0 diff -r -N "$expected" "$actual"
}
function expect_run () {
local expected_exit_code=$1
shift
local output=
set +e
if [[ -n "${SNAPSHOT_FILTER-}" ]]; then
output="$("$@" 2>&1 | "$SNAPSHOT_FILTER")"
else
output="$("$@" 2>&1)"
fi
local actual_exit_code=$?
if [[ "$actual_exit_code" == "$expected_exit_code" ]]; then
if [[ -n "${WITH_SNAPSHOT-}" ]]; then
local expected="$WITH_SNAPSHOT"
if ! [ -f "$expected" ]; then
mkdir -p "${expected%/*}"
echo -n "$output" > "$expected" || exit 1
fi
if ! diff "$expected" <(echo -n "$output"); then
echo 1>&2 "${RED} - FAIL"
echo 1>&2 "${WHITE}\$ $*"
echo 1>&2 "Output snapshot did not match snapshot at '$expected'"
echo 1>&2 "$output"
if [ -n "${ON_ERROR:-}" ]; then
eval "$ON_ERROR"
fi
exit 1
fi
fi
echo 1>&2
else
echo 1>&2 "${RED} - FAIL"
echo 1>&2 "${WHITE}\$ $*"
echo 1>&2 "${RED}Expected actual status $actual_exit_code to be $expected_exit_code"
echo 1>&2 "$output"
if [ -n "${ON_ERROR:-}" ]; then
eval "$ON_ERROR"
fi
exit 1
fi
set -e
}
|
