pyenv verification source

test/rehash.bats

@@ -0,0 +1,178 @@
+#!/usr/bin/env bats
+
+load test_helper
+
+@test "empty rehash" {
+  assert [ ! -d "${PYENV_ROOT}/shims" ]
+  run pyenv-rehash
+  assert_success ""
+  assert [ -d "${PYENV_ROOT}/shims" ]
+  rmdir "${PYENV_ROOT}/shims"
+}
+
+@test "non-writable shims directory" {
+  mkdir -p "${PYENV_ROOT}/shims"
+  chmod -w "${PYENV_ROOT}/shims"
+  run pyenv-rehash
+  assert_failure "pyenv: cannot rehash: ${PYENV_ROOT}/shims isn't writable"
+}
+
+@test "rehash in progress" {
+  export PYENV_REHASH_TIMEOUT=1
+  mkdir -p "${PYENV_ROOT}/shims"
+  touch "${PYENV_ROOT}/shims/.pyenv-shim"
+  #avoid failure due to a localized error message
+  LANG=C run pyenv-rehash
+  assert_failure <<!
+pyenv: cannot rehash: couldn't acquire lock ${PYENV_ROOT}/shims/.pyenv-shim for 1 seconds. Last error message:
+*/pyenv-rehash: line *: ${PYENV_ROOT}/shims/.pyenv-shim: cannot overwrite existing file
+!
+}
+
+@test "wait until lock acquisition" {
+  export PYENV_REHASH_TIMEOUT=5
+  mkdir -p "${PYENV_ROOT}/shims"
+  touch "${PYENV_ROOT}/shims/.pyenv-shim"
+  bash -c "sleep 1 && rm -f ${PYENV_ROOT}/shims/.pyenv-shim" &
+  run pyenv-rehash
+  assert_success
+}
+
+@test "creates shims" {
+  create_alt_executable_in_version "2.7" "python"
+  create_alt_executable_in_version "2.7" "fab"
+  create_alt_executable_in_version "3.4" "python"
+  create_alt_executable_in_version "3.4" "py.test"
+
+  assert [ ! -e "${PYENV_ROOT}/shims/fab" ]
+  assert [ ! -e "${PYENV_ROOT}/shims/python" ]
+  assert [ ! -e "${PYENV_ROOT}/shims/py.test" ]
+
+  run pyenv-rehash
+  assert_success ""
+
+  run ls "${PYENV_ROOT}/shims"
+  assert_success
+  assert_output <<OUT
+fab
+py.test
+python
+OUT
+}
+
+@test "removes stale shims" {
+  mkdir -p "${PYENV_ROOT}/shims"
+  touch "${PYENV_ROOT}/shims/oldshim1"
+  chmod +x "${PYENV_ROOT}/shims/oldshim1"
+
+  create_alt_executable_in_version "3.4" "fab"
+  create_alt_executable_in_version "3.4" "python"
+
+  run pyenv-rehash
+  assert_success ""
+
+  assert [ ! -e "${PYENV_ROOT}/shims/oldshim1" ]
+}
+
+@test "binary install locations containing spaces" {
+  create_alt_executable_in_version "dirname1 p247" "python"
+  create_alt_executable_in_version "dirname2 preview1" "py.test"
+
+  assert [ ! -e "${PYENV_ROOT}/shims/python" ]
+  assert [ ! -e "${PYENV_ROOT}/shims/py.test" ]
+
+  run pyenv-rehash
+  assert_success ""
+
+  run ls "${PYENV_ROOT}/shims"
+  assert_success
+  assert_output <<OUT
+py.test
+python
+OUT
+}
+
+@test "carries original IFS within hooks" {
+  create_hook rehash hello.bash <<SH
+hellos=(\$(printf "hello\\tugly world\\nagain"))
+echo HELLO="\$(printf ":%s" "\${hellos[@]}")"
+exit
+SH
+
+  IFS=$' \t\n' run pyenv-rehash
+  assert_success
+  assert_output "HELLO=:hello:ugly:world:again"
+}
+
+@test "sh-rehash in bash" {
+  create_alt_executable_in_version "3.4" "python"
+  PYENV_SHELL=bash run pyenv-sh-rehash
+  assert_success "command pyenv rehash
+hash -r 2>/dev/null || true"
+}
+
+@test "sh-rehash in bash (integration)" {
+  create_alt_executable_in_version "3.4" "python"
+  PYENV_SHELL=bash run eval "$(pyenv-sh-rehash)"
+  assert_success
+  assert [ -x "${PYENV_ROOT}/shims/python" ]
+}
+
+@test "sh-rehash in fish" {
+  create_alt_executable_in_version "3.4" "python"
+  PYENV_SHELL=fish run pyenv-sh-rehash
+  assert_success "command pyenv rehash"
+}
+
+@test "sh-rehash in fish (integration)" {
+  command -v fish >/dev/null || skip "-- fish not installed" 
+  create_alt_executable_in_version "3.4" "python"
+  PYENV_SHELL=fish fish -Nc "source (pyenv-sh-rehash | psub)"
+  assert_success
+  assert [ -x "${PYENV_ROOT}/shims/python" ]
+}
+
+@test "sh-rehash in pwsh" {
+  create_alt_executable_in_version "3.4" "python"
+  PYENV_SHELL=pwsh run pyenv-sh-rehash
+  assert_success "& (get-command pyenv -commandtype application) rehash"
+}
+
+@test "sh-rehash in pwsh (integration)" {
+  command -v pwsh >/dev/null || skip "-- pwsh not installed" 
+  assert [ ! -x "${PYENV_ROOT}/shims/python" ]
+  create_alt_executable_in_version "3.4" "python"
+  run pwsh -nop -c -<<'!'
+$ErrorActionPreference = "Stop"
+$PSNativeCommandUseErrorActionPreference = $true
+iex ((& pyenv init - pwsh) -join "`n")
+& pyenv rehash
+!
+  assert_success ""
+  assert [ -x "${PYENV_ROOT}/shims/python" ]
+}
+
+@test "shim sets _PYENV_SHIM_PATH when linked from elsewhere" {
+  export PYENV_VERSION="custom"
+  create_alt_executable python3
+  #must stub pyenv before rehash 'cuz the path is hardcoded into shims
+  create_stub pyenv <<!
+[[ \$1 == 'exec' ]] && \
+echo _PYENV_SHIM_PATH="\$_PYENV_SHIM_PATH"
+!
+  pyenv-rehash
+  mkdir -p "${PYENV_TEST_DIR}/alt-shim"
+  ln -s "${PYENV_ROOT}/shims/python3" "${PYENV_TEST_DIR}/alt-shim/python3"
+  
+  run "${PYENV_TEST_DIR}/alt-shim/python3"
+  assert_success
+  assert_output <<!
+_PYENV_SHIM_PATH=${PYENV_TEST_DIR}/alt-shim
+!
+
+  run python3
+  assert_success
+  assert_output <<!
+_PYENV_SHIM_PATH=
+!
+}