From 7079cbe979623786bee9985cc8a82ba5893dc325 Mon Sep 17 00:00:00 2001
From: Yamashita Yuu <yamashita@geishatokyo.com>
Date: Mon, 30 Jun 2014 21:35:43 +0900
Subject: [PATCH] Fix broken `setenv` in `pyenv-virtualenv-init`

---
 bin/pyenv-virtualenv-init |  2 +-
 test/init.bats            | 43 ++++++++++++++++++++++++++++++++++++---
 2 files changed, 41 insertions(+), 4 deletions(-)

diff --git a/bin/pyenv-virtualenv-init b/bin/pyenv-virtualenv-init
index ae5b9f1..cdf6f8d 100755
--- a/bin/pyenv-virtualenv-init
+++ b/bin/pyenv-virtualenv-init
@@ -81,7 +81,7 @@ EOS
   ;;
 fish )
   cat <<EOS
-setenv PYENV_VIRTUALENV_INIT=1;
+setenv PYENV_VIRTUALENV_INIT 1;
 function _pyenv_virtualenv_hook --on-event fish_prompt;
   if [ (pyenv version-name) = "system" ]
     eval (pyenv sh-deactivate); or true
diff --git a/test/init.bats b/test/init.bats
index 6edcc47..5071396 100644
--- a/test/init.bats
+++ b/test/init.bats
@@ -28,17 +28,54 @@ load test_helper
 @test "outputs bash-specific syntax" {
   run pyenv-virtualenv-init - bash
   assert_success
-  assert_output_contains '  PROMPT_COMMAND="_pyenv_virtualenv_hook;$PROMPT_COMMAND";'
+  assert_output <<EOS
+export PYENV_VIRTUALENV_INIT=1
+_pyenv_virtualenv_hook() {
+  if [[ "\$(pyenv version-name)" == "system" ]]; then
+    pyenv deactivate || true;
+  elif [[ "\$VIRTUAL_ENV" != "\$(pyenv prefix)" ]]; then
+    pyenv deactivate || true;
+    pyenv activate 2>/dev/null || true
+  fi
+};
+if ! [[ "\$PROMPT_COMMAND" =~ _pyenv_virtualenv_hook ]]; then
+  PROMPT_COMMAND="_pyenv_virtualenv_hook;\$PROMPT_COMMAND";
+fi
+EOS
 }
 
 @test "outputs fish-specific syntax" {
   run pyenv-virtualenv-init - fish
   assert_success
-  assert_output_contains 'function _pyenv_virtualenv_hook --on-event fish_prompt;'
+  assert_output <<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
+  else if [ "\$VIRTUAL_ENV" != (pyenv prefix) ]
+    eval (pyenv sh-deactivate); or true
+    eval (pyenv sh-activate 2>/dev/null); or true
+  end
+end
+EOS
 }
 
 @test "outputs zsh-specific syntax" {
   run pyenv-virtualenv-init - zsh
   assert_success
-  assert_output_contains '  precmd_functions+=_pyenv_virtualenv_hook;'
+  assert_output <<EOS
+export PYENV_VIRTUALENV_INIT=1
+_pyenv_virtualenv_hook() {
+  if [[ "\$(pyenv version-name)" == "system" ]]; then
+    pyenv deactivate || true
+  elif [[ "\$VIRTUAL_ENV" != "\$(pyenv prefix)" ]]; then
+    pyenv deactivate || true
+    pyenv activate 2>/dev/null || true
+  fi
+}
+typeset -a precmd_functions
+if [[ -z \$precmd_functions[(r)_pyenv_virtualenv_hook] ]]; then
+  precmd_functions+=_pyenv_virtualenv_hook;
+fi
+EOS
 }