pyenv/plugins/pyenv-binary/test/save.bats

#!/usr/bin/env bats

load test_helper

create_version() {
  mkdir -p "${PYENV_ROOT}/versions/$1/bin"
}

platform() {
  echo "$(uname -s | tr '[:upper:]' '[:lower:]')-$(uname -m)"
}

@test "fails with no version given" {
  run pyenv-binary-save
  assert_failure "Usage: pyenv binary save <version> [<output-dir>] [--name <name>]"
}

@test "fails for a version that is not installed" {
  run pyenv-binary-save 9.9.9
  assert_failure "pyenv-binary: version \`9.9.9' is not installed"
}

@test "rejects a version name containing a slash" {
  run pyenv-binary-save "foo/bar"
  assert_failure "pyenv-binary: invalid version name \`foo/bar'"
}

@test "rejects the parent directory reference" {
  run pyenv-binary-save ".."
  assert_failure "pyenv-binary: invalid version name \`..'"
}

@test "packages an installed version" {
  create_version "3.12.7"
  local out="${BATS_TEST_TMPDIR}/dist"
  local archive="${out}/3.12.7-$(platform).tar.gz"

  run pyenv-binary-save "3.12.7" "$out"
  assert_success "Saved 3.12.7-$(platform).tar.gz and 3.12.7-$(platform).meta to $out"
  assert [ -f "$archive" ]
  assert [ -f "${out}/3.12.7-$(platform).meta" ]
  run tar -tzf "$archive"
  assert_success
  assert_line 0 "3.12.7/"
}

@test "uses an explicit package name" {
  create_version "3.12.7"
  local out="${BATS_TEST_TMPDIR}/dist"

  run pyenv-binary-save "3.12.7" "$out" --name "custom"
  assert_success "Saved custom.tar.gz and custom.meta to $out"
  assert [ -f "${out}/custom.tar.gz" ]
  run grep '^archive=' "${out}/custom.meta"
  assert_success "archive=custom.tar.gz"
}

@test "fails when --name has no value" {
  run pyenv-binary-save "3.12.7" --name
  assert_failure "pyenv-binary: --name needs a value"
}

@test "rejects an invalid package name" {
  create_version "3.12.7"

  run pyenv-binary-save "3.12.7" --name "foo/bar"
  assert_failure "pyenv-binary: invalid package name \`foo/bar'"
}

@test "records the platform in the metadata" {
  create_version "3.12.7"
  local out="${BATS_TEST_TMPDIR}/dist"
  pyenv-binary-save "3.12.7" "$out" >/dev/null

  run cat "${out}/3.12.7-$(platform).meta"
  assert_success
  assert_line "version=3.12.7"
  assert_line "platform=$(platform)"
  assert_line "archive=3.12.7-$(platform).tar.gz"
}

@test "records only the libraries ldd resolves outside the prefix" {
  create_version "3.12.7"
  touch "${PYENV_ROOT}/versions/3.12.7/bin/python3.12"
  create_path_executable uname <<'STUB'
case "$1" in
  -s) echo Linux ;;
  -m) echo x86_64 ;;
esac
STUB
  create_path_executable ldd <<'STUB'
prefix="${PYENV_ROOT}/versions/3.12.7"
cat <<EOF
	linux-vdso.so.1 (0x00007ffd1adfe000)
	libpython3.12.so.1.0 => ${prefix}/lib/libpython3.12.so.1.0 (0x00007f4a3c000000)
	libm.so.6 => /lib/x86_64-linux-gnu/libm.so.6 (0x00007f4a3bc00000)
	libc.so.6 => /lib/x86_64-linux-gnu/libc.so.6 (0x00007f4a3b800000)
	/lib64/ld-linux-x86-64.so.2 (0x00007f4a3c200000)
EOF
STUB

  run pyenv-binary-save "3.12.7" "${BATS_TEST_TMPDIR}/dist"
  assert_success
  run grep '^dep=' "${BATS_TEST_TMPDIR}/dist/"*.meta
  assert_output "dep=libc.so.6
dep=libm.so.6"
}

@test "records only the libraries otool resolves outside the prefix" {
  create_version "3.12.7"
  touch "${PYENV_ROOT}/versions/3.12.7/bin/python3.12"
  create_path_executable uname <<'STUB'
case "$1" in
  -s) echo Darwin ;;
  -m) echo arm64 ;;
esac
STUB
  create_path_executable otool <<'STUB'
prefix="${PYENV_ROOT}/versions/3.12.7"
cat <<EOF
${2}:
	@rpath/libpython3.12.dylib (compatibility version 3.12.0, current version 3.12.0)
	${prefix}/lib/libcrypto.3.dylib (compatibility version 3.0.0, current version 3.0.0)
	/usr/lib/libSystem.B.dylib (compatibility version 1.0.0, current version 1345.0.0)
EOF
STUB

  run pyenv-binary-save "3.12.7" "${BATS_TEST_TMPDIR}/dist"
  assert_success
  run grep '^dep=' "${BATS_TEST_TMPDIR}/dist/"*.meta
  assert_output "dep=/usr/lib/libSystem.B.dylib"
}