109 lines
4.0 KiB
Python
109 lines
4.0 KiB
Python
from __future__ import annotations
|
|
|
|
import sys
|
|
from platform import python_implementation
|
|
from typing import TYPE_CHECKING
|
|
|
|
import pytest
|
|
import virtualenv
|
|
|
|
from pipdeptree.__main__ import main
|
|
|
|
if TYPE_CHECKING:
|
|
from pathlib import Path
|
|
|
|
from pytest_mock import MockerFixture
|
|
|
|
|
|
@pytest.mark.parametrize("args_joined", [True, False])
|
|
def test_custom_interpreter(
|
|
tmp_path: Path,
|
|
mocker: MockerFixture,
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
capfd: pytest.CaptureFixture[str],
|
|
args_joined: bool,
|
|
) -> None:
|
|
# Delete $PYTHONPATH so that it cannot be passed to the custom interpreter process (since we don't know what
|
|
# distribution metadata to expect when it's used).
|
|
monkeypatch.delenv("PYTHONPATH", False)
|
|
|
|
monkeypatch.chdir(tmp_path)
|
|
result = virtualenv.cli_run([str(tmp_path / "venv"), "--activators", ""])
|
|
py = str(result.creator.exe.relative_to(tmp_path))
|
|
cmd = ["", f"--python={result.creator.exe}"] if args_joined else ["", "--python", py]
|
|
mocker.patch("pipdeptree._discovery.sys.argv", cmd)
|
|
main()
|
|
out, _ = capfd.readouterr()
|
|
found = {i.split("==")[0] for i in out.splitlines()}
|
|
implementation = python_implementation()
|
|
if implementation == "CPython":
|
|
expected = {"pip", "setuptools", "wheel"}
|
|
elif implementation == "PyPy":
|
|
expected = {"cffi", "greenlet", "pip", "readline", "setuptools", "wheel"} # pragma: no cover
|
|
else:
|
|
raise ValueError(implementation)
|
|
if sys.version_info >= (3, 12):
|
|
expected -= {"setuptools", "wheel"}
|
|
assert found == expected, out
|
|
|
|
|
|
def test_custom_interpreter_with_local_only(
|
|
tmp_path: Path,
|
|
mocker: MockerFixture,
|
|
capfd: pytest.CaptureFixture[str],
|
|
) -> None:
|
|
venv_path = str(tmp_path / "venv")
|
|
|
|
result = virtualenv.cli_run([venv_path, "--system-site-packages", "--activators", ""])
|
|
|
|
cmd = ["", f"--python={result.creator.exe}", "--local-only"]
|
|
mocker.patch("pipdeptree._discovery.sys.prefix", venv_path)
|
|
mocker.patch("pipdeptree._discovery.sys.argv", cmd)
|
|
main()
|
|
out, _ = capfd.readouterr()
|
|
found = {i.split("==")[0] for i in out.splitlines()}
|
|
implementation = python_implementation()
|
|
if implementation == "CPython":
|
|
expected = {"pip", "setuptools", "wheel"}
|
|
elif implementation == "PyPy": # pragma: no cover
|
|
expected = {"cffi", "greenlet", "pip", "readline", "setuptools", "wheel"} # pragma: no cover
|
|
else:
|
|
raise ValueError(implementation) # pragma: no cover
|
|
if sys.version_info >= (3, 12):
|
|
expected -= {"setuptools", "wheel"} # pragma: no cover
|
|
assert found == expected, out
|
|
|
|
|
|
def test_custom_interpreter_ensure_pythonpath_envar_is_honored(
|
|
tmp_path: Path,
|
|
mocker: MockerFixture,
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
capfd: pytest.CaptureFixture[str],
|
|
) -> None:
|
|
# ensures that we honor $PYTHONPATH when passing it to the custom interpreter process
|
|
venv_path = str(tmp_path / "venv")
|
|
result = virtualenv.cli_run([venv_path, "--activators", ""])
|
|
|
|
another_path = tmp_path / "another-path"
|
|
fake_dist = another_path / "foo-1.2.3.dist-info"
|
|
fake_dist.mkdir(parents=True)
|
|
fake_metadata = fake_dist / "METADATA"
|
|
with fake_metadata.open("w") as f:
|
|
f.write("Metadata-Version: 2.3\n" "Name: foo\n" "Version: 1.2.3\n")
|
|
cmd = ["", f"--python={result.creator.exe}"]
|
|
mocker.patch("pipdeptree._discovery.sys.argv", cmd)
|
|
monkeypatch.setenv("PYTHONPATH", str(another_path))
|
|
main()
|
|
out, _ = capfd.readouterr()
|
|
found = {i.split("==")[0] for i in out.splitlines()}
|
|
implementation = python_implementation()
|
|
if implementation == "CPython":
|
|
expected = {"foo", "pip", "setuptools", "wheel"}
|
|
elif implementation == "PyPy": # pragma: cpython no cover
|
|
expected = {"foo", "cffi", "greenlet", "pip", "readline", "setuptools", "wheel"}
|
|
else: # pragma: no cover
|
|
raise ValueError(implementation)
|
|
if sys.version_info >= (3, 12): # pragma: >=3.12 cover
|
|
expected -= {"setuptools", "wheel"}
|
|
assert found == expected, out
|