X-Git-Url: https://git.notmuchmail.org/git?p=notmuch;a=blobdiff_plain;f=test%2Ftest-lib.sh;h=37f8ddfa4f94ff02cfe294392604b0c08545aa32;hp=988b00afde86b7ff306c2b0812d9b7e6fa0a0428;hb=eb157f8841b93a0df241b815e30ac54ca81cc83d;hpb=c0c5645538d9dda84657602ee4189f4ff3ca1aeb

diff --git a/test/test-lib.sh b/test/test-lib.sh
index 988b00af..37f8ddfa 100644
--- a/test/test-lib.sh
+++ b/test/test-lib.sh
@@ -219,10 +219,21 @@ test_fixed=0
 test_broken=0
 test_success=0
 
+declare -a _exit_functions=()
+
+at_exit_function () {
+	_exit_functions=($1 ${_exit_functions[@]/$1})
+}
+
+rm_exit_function () {
+	_exit_functions=(${_exit_functions[@]/$1})
+}
+
 _exit_common () {
 	code=$?
 	trap - EXIT
 	set +ex
+	for _fn in ${_exit_functions[@]}; do $_fn; done
 	rm -rf "$TEST_TMPDIR"
 }