tests: define/use $TEST_SOURCE_AUTOENV

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

diff --git a/tests/setup.sh b/tests/setup.sh
index afa5c958777e5d95f767b0fb391ebc772bf45080..27ba4239c446bdab4a71530af6210d5029e4edd4 100644 (file)
--- a/tests/setup.sh
+++ b/tests/setup.sh
@@ -18,8 +18,12 @@ fi
 _save_errexit=${options[errexit]}
 set -e
 
-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