-# Ensure we have our mocked out ENV_AUTHORIZATION_FILE
+# Setup for tests.
+#
+# It returns 1 in case of errors, and no tests should be run then!
+#
+# Ensure we have our mocked out AUTOENV_ENV_FILENAME
+# (via .zshenv).
-[[ $ENV_AUTHORIZATION_FILE[0,4] == '/tmp' ]] || return 1
+# Treat unset variables as errors.
+# Not handled in varstash yet.
+# setopt nounset
-# Inject timeout for `read` while running tests.
-_AUTOENV_TEST_READ_ARGS='-t 1'
+if [[ $AUTOENV_ENV_FILENAME[0,4] != '/tmp' ]]; then
+ echo "AUTOENV_ENV_FILENAME is not in /tmp. Aborting."
+ return 1
+fi
+
+# Abort this setup script on any error.
+_save_errexit=${options[errexit]}
+set -e
+
+if (( ${+functions[antigen-like-loader-function]} )); then
+ TEST_SOURCE_AUTOENV=antigen-like-loader-function
+else
+ TEST_SOURCE_AUTOENV=(source $TESTDIR/../autoenv.plugin.zsh)
+fi
+$TEST_SOURCE_AUTOENV
+
+# Reset any authentication.
+echo -n >| $AUTOENV_ENV_FILENAME
+
+# Add file $1 (with optional hash $2) to authentication file.
+test_autoenv_add_to_env() {
+ _autoenv_hash_pair $1 ${2:-} >>| $AUTOENV_ENV_FILENAME
+}
+
+# Add enter and leave env files to authentication file.
+test_autoenv_auth_env_files() {
+ test_autoenv_add_to_env $PWD/$AUTOENV_FILE_ENTER
+ test_autoenv_add_to_env $PWD/$AUTOENV_FILE_LEAVE
+}
+
+# Now keep on going on errors again.
+options[errexit]=$_save_errexit
+unset _save_errexit