From d9e3adc1085d5f8ddc525088bf8532a3c080f971 Mon Sep 17 00:00:00 2001
From: Yamashita Yuu <yamashita@geishatokyo.com>
Date: Sun, 15 Jun 2014 16:32:25 +0900
Subject: [PATCH] Performe as same as before v20140614 if `pyenv
 virtualenv-init` is not configured (#26)

---
 bin/pyenv-sh-activate     |  6 ++++
 bin/pyenv-sh-deactivate   | 10 ++++--
 bin/pyenv-virtualenv-init |  3 ++
 test/activate.bats        | 64 +++++++++++++++++++++++++++++++++++++++
 test/deactivate.bats      | 32 ++++++++++++++++++--
 5 files changed, 111 insertions(+), 4 deletions(-)

diff --git a/bin/pyenv-sh-activate b/bin/pyenv-sh-activate
index 0aff9eba..547c6bba 100755
--- a/bin/pyenv-sh-activate
+++ b/bin/pyenv-sh-activate
@@ -33,6 +33,12 @@ if [ -z "$versions" ]; then
   IFS="$OLDIFS"
 fi
 
+if [ -z "${PYENV_VIRTUALENV_INIT}" ]; then
+  # Backward compatibility issue
+  # https://github.com/yyuu/pyenv-virtualenv/issues/26
+  no_shell=
+fi
+
 if [ "${#versions[@]}" -gt 1 ]; then
   echo "pyenv-virtualenv: cannot activate multiple versions at once: ${versions[@]}" 1>&2
   exit 1
diff --git a/bin/pyenv-sh-deactivate b/bin/pyenv-sh-deactivate
index 8cc75683..38b08fb9 100755
--- a/bin/pyenv-sh-deactivate
+++ b/bin/pyenv-sh-deactivate
@@ -11,6 +11,12 @@ set -e
 
 shell="$(basename "${PYENV_SHELL:-$SHELL}")"
 case "$shell" in
-fish ) echo "functions -q deactivate; and deactivate";;
-*    ) echo "declare -f deactivate 1>/dev/null 2>&1 && deactivate";;
+fish ) echo "functions -q deactivate; and deactivate;";;
+*    ) echo "declare -f deactivate 1>/dev/null 2>&1 && deactivate;";;
 esac
+
+if [ -z "${PYENV_VIRTUALENV_INIT}" ]; then
+  # Backward compatibility issue
+  # https://github.com/yyuu/pyenv-virtualenv/issues/26
+  echo "pyenv shell --unset"
+fi
diff --git a/bin/pyenv-virtualenv-init b/bin/pyenv-virtualenv-init
index 20f8744f..ae5b9f15 100755
--- a/bin/pyenv-virtualenv-init
+++ b/bin/pyenv-virtualenv-init
@@ -65,6 +65,7 @@ fi
 case "$shell" in
 bash )
   cat <<EOS
+export PYENV_VIRTUALENV_INIT=1
 _pyenv_virtualenv_hook() {
   if [[ "\$(pyenv version-name)" == "system" ]]; then
     pyenv deactivate || true;
@@ -80,6 +81,7 @@ EOS
   ;;
 fish )
   cat <<EOS
+setenv PYENV_VIRTUALENV_INIT=1;
 function _pyenv_virtualenv_hook --on-event fish_prompt;
   if [ (pyenv version-name) = "system" ]
     eval (pyenv sh-deactivate); or true
@@ -92,6 +94,7 @@ EOS
   ;;
 zsh )
   cat <<EOS
+export PYENV_VIRTUALENV_INIT=1
 _pyenv_virtualenv_hook() {
   if [[ "\$(pyenv version-name)" == "system" ]]; then
     pyenv deactivate || true
diff --git a/test/activate.bats b/test/activate.bats
index d7fde1fe..f1e98eaf 100644
--- a/test/activate.bats
+++ b/test/activate.bats
@@ -8,6 +8,8 @@ setup() {
 }
 
 @test "activate virtualenv from current version" {
+  export PYENV_VIRTUALENV_INIT=1
+
   stub pyenv-version-name "echo venv"
   stub pyenv-virtualenv-prefix "venv : echo \"${PYENV_ROOT}/versions/venv\""
   stub pyenv-prefix "venv : echo \"${PYENV_ROOT}/versions/venv\""
@@ -24,7 +26,29 @@ source "${PYENV_ROOT}/versions/venv/bin/activate"
 EOS
 }
 
+@test "activate virtualenv from current version (without pyenv-virtualenv-init)" {
+  export PYENV_VIRTUALENV_INIT=
+
+  stub pyenv-version-name "echo venv"
+  stub pyenv-virtualenv-prefix "venv : echo \"${PYENV_ROOT}/versions/venv\""
+  stub pyenv-prefix "venv : echo \"${PYENV_ROOT}/versions/venv\""
+
+  PYENV_SHELL="bash" PYENV_VERSION="venv" run pyenv-sh-activate
+
+  unstub pyenv-version-name
+  unstub pyenv-virtualenv-prefix
+  unstub pyenv-prefix
+
+  assert_success
+  assert_output <<EOS
+pyenv shell "venv";
+source "${PYENV_ROOT}/versions/venv/bin/activate"
+EOS
+}
+
 @test "activate virtualenv from current version (fish)" {
+  export PYENV_VIRTUALENV_INIT=1
+
   stub pyenv-version-name "echo venv"
   stub pyenv-virtualenv-prefix "venv : echo \"${PYENV_ROOT}/versions/venv\""
   stub pyenv-prefix "venv : echo \"${PYENV_ROOT}/versions/venv\""
@@ -41,7 +65,47 @@ EOS
 EOS
 }
 
+@test "activate virtualenv from current version (fish) (without pyenv-virtualenv-init)" {
+  export PYENV_VIRTUALENV_INIT=
+
+  stub pyenv-version-name "echo venv"
+  stub pyenv-virtualenv-prefix "venv : echo \"${PYENV_ROOT}/versions/venv\""
+  stub pyenv-prefix "venv : echo \"${PYENV_ROOT}/versions/venv\""
+
+  PYENV_SHELL="fish" PYENV_VERSION="venv" run pyenv-sh-activate
+
+  unstub pyenv-version-name
+  unstub pyenv-virtualenv-prefix
+  unstub pyenv-prefix
+
+  assert_success
+  assert_output <<EOS
+pyenv shell "venv";
+. "${PYENV_ROOT}/versions/venv/bin/activate.fish"
+EOS
+}
+
 @test "activate virtualenv from command-line argument" {
+  export PYENV_VIRTUALENV_INIT=1
+
+  stub pyenv-virtualenv-prefix "venv27 : echo \"${PYENV_ROOT}/versions/venv27\""
+  stub pyenv-prefix "venv27 : echo \"${PYENV_ROOT}/versions/venv27\""
+
+  run pyenv-sh-activate "venv27"
+
+  unstub pyenv-virtualenv-prefix
+  unstub pyenv-prefix
+
+  assert_success
+  assert_output <<EOS
+pyenv shell "venv27";
+source "${PYENV_ROOT}/versions/venv27/bin/activate"
+EOS
+}
+
+@test "activate virtualenv from command-line argument (without pyenv-virtualenv-init)" {
+  export PYENV_VIRTUALENV_INIT=
+
   stub pyenv-virtualenv-prefix "venv27 : echo \"${PYENV_ROOT}/versions/venv27\""
   stub pyenv-prefix "venv27 : echo \"${PYENV_ROOT}/versions/venv27\""
 
diff --git a/test/deactivate.bats b/test/deactivate.bats
index e8dfbd67..79427eac 100644
--- a/test/deactivate.bats
+++ b/test/deactivate.bats
@@ -7,20 +7,48 @@ setup() {
 }
 
 @test "deactivate virtualenv" {
+  export PYENV_VIRTUALENV_INIT=1
+
+  PYENV_SHELL="bash" run pyenv-sh-deactivate
+
+  assert_success
+  assert_output <<EOS
+declare -f deactivate 1>/dev/null 2>&1 && deactivate;
+EOS
+}
+
+@test "deactivate virtualenv (without pyenv-virtualenv-init)" {
+  export PYENV_VIRTUALENV_INIT=
+
   PYENV_SHELL="bash" run pyenv-sh-deactivate
 
   assert_success
   assert_output <<EOS
-declare -f deactivate 1>/dev/null 2>&1 && deactivate
+declare -f deactivate 1>/dev/null 2>&1 && deactivate;
+pyenv shell --unset
 EOS
 }
 
 @test "deactivate virtualenv (fish)" {
+  export PYENV_VIRTUALENV_INIT=1
+
+  PYENV_SHELL="fish" run pyenv-sh-deactivate
+
+  assert_success
+  assert_output <<EOS
+functions -q deactivate; and deactivate;
+EOS
+}
+
+@test "deactivate virtualenv (fish) (without pyenv-virtualenv-init)" {
+  export PYENV_VIRTUALENV_INIT=
+
   PYENV_SHELL="fish" run pyenv-sh-deactivate
 
   assert_success
   assert_output <<EOS
-functions -q deactivate; and deactivate
+functions -q deactivate; and deactivate;
+pyenv shell --unset
 EOS
 }