tests: define/use $TEST_SOURCE_AUTOENV

[zsh-autoenv.git] / tests / setup.sh

diff --git a/tests/setup.sh b/tests/setup.sh
index 9e5cf7b9f84427f3e718412c40b63b49d07eeb4b..27ba4239c446bdab4a71530af6210d5029e4edd4 100644 (file)
--- a/tests/setup.sh
+++ b/tests/setup.sh
@@ -18,8 +18,12 @@ fi
 _save_errexit=${options[errexit]}
 set -e
 
-export TEST_AUTOENV_PLUGIN_FILE="$TESTDIR/../autoenv.plugin.zsh"
-source $TEST_AUTOENV_PLUGIN_FILE
+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