From cbcdd66863756c48df97b446efed69a250a1938b Mon Sep 17 00:00:00 2001
From: Ivan Povalyukhin <ivanpoval@gmail.com>
Date: Sun, 29 Mar 2015 17:21:28 -0700
Subject: [PATCH] test base.plugin#pickfrom

---
 test/plugins/base.plugin.bats |  9 +++++++++
 test/test_helper.bash         | 14 ++++++++++++--
 2 files changed, 21 insertions(+), 2 deletions(-)

diff --git a/test/plugins/base.plugin.bats b/test/plugins/base.plugin.bats
index fe936e60..41c9ea5c 100755
--- a/test/plugins/base.plugin.bats
+++ b/test/plugins/base.plugin.bats
@@ -17,3 +17,12 @@ load ../../plugins/available/base.plugin
   declare -r mask_ip=$(echo $output | tr -s '[0-9]' '?')
   [[ $mask_ip == 'Your public IP is: ?.?.?.?' ]]
 }
+
+@test 'plugins base: pickfrom()' {
+  mkdir -p $BASH_IT_ROOT
+  stub_file="${BASH_IT_ROOT}/stub_file"
+  printf "l1\nl2\nl3" > $stub_file
+  run pickfrom $stub_file
+  assert_success
+  [[ $output == l? ]]
+}
diff --git a/test/test_helper.bash b/test/test_helper.bash
index da34c438..faf2b081 100644
--- a/test/test_helper.bash
+++ b/test/test_helper.bash
@@ -1,4 +1,3 @@
-unset BASH_IT
 unset BASH_IT_THEME
 unset GIT_HOSTING
 unset NGINX_PATH
@@ -6,7 +5,18 @@ unset IRC_CLIENT
 unset TODO
 unset SCM_CHECK
 
-BASH_IT_TEST_DIR="${BATS_TMPDIR}/bash_it"
+BASH_IT_TEST_DIR="${BATS_TMPDIR}/.bash_it"
+
+# guard against executing this block twice due to bats internals
+if [ "$BASH_IT_ROOT" != "${BASH_IT_TEST_DIR}/root" ]; then
+  export BASH_IT_ROOT="${BASH_IT_TEST_DIR}/root"
+  export HOME="${BASH_IT_TEST_DIR}/home"
+  export BASH_IT=$BASH_IT_TEST_DIR
+
+  PATH=/usr/bin:/bin:/usr/sbin:/sbin:/usr/local/bin
+  PATH="${BASH_IT_TEST_DIR}/bin:$PATH"
+  export PATH
+fi
 
 teardown() {
   rm -rf "$BASH_IT_TEST_DIR"