mirror of
https://github.com/kennethreitz/pipenv.git
synced 2026-06-05 22:50:18 +00:00
Matt Davis
216 lines
6.9 KiB
Python
216 lines
6.9 KiB
Python
import os
|
|
import tarfile
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from pipenv.project import Project
|
|
from pipenv.utils.shell import temp_environ
|
|
from pipenv.vendor.vistir.path import is_in_path, normalize_path
|
|
from pipenv.vendor.plette import Pipfile
|
|
|
|
|
|
@pytest.mark.project
|
|
@pytest.mark.sources
|
|
@pytest.mark.environ
|
|
def test_pipfile_envvar_expansion(pipenv_instance_pypi):
|
|
with pipenv_instance_pypi(chdir=True) as p:
|
|
with temp_environ():
|
|
with open(p.pipfile_path, 'w') as f:
|
|
f.write("""
|
|
[[source]]
|
|
url = 'https://${TEST_HOST}/simple'
|
|
verify_ssl = false
|
|
name = "pypi"
|
|
|
|
[packages]
|
|
pytz = "*"
|
|
""".strip())
|
|
os.environ['TEST_HOST'] = 'localhost:5000'
|
|
project = Project()
|
|
assert project.sources[0]['url'] == 'https://localhost:5000/simple'
|
|
assert 'localhost:5000' not in str(Pipfile.load(open(p.pipfile_path)))
|
|
print(str(Pipfile.load(open(p.pipfile_path))))
|
|
|
|
|
|
@pytest.mark.project
|
|
@pytest.mark.sources
|
|
@pytest.mark.parametrize('lock_first', [True, False])
|
|
def test_get_source(pipenv_instance_private_pypi, lock_first):
|
|
with pipenv_instance_private_pypi(chdir=True) as p:
|
|
with open(p.pipfile_path, 'w') as f:
|
|
contents = """
|
|
[[source]]
|
|
url = "{0}"
|
|
verify_ssl = false
|
|
name = "testindex"
|
|
|
|
[[source]]
|
|
url = "https://pypi.org/simple"
|
|
verify_ssl = true
|
|
name = "pypi"
|
|
|
|
[packages]
|
|
pytz = "*"
|
|
six = {{version = "*", index = "pypi"}}
|
|
|
|
[dev-packages]
|
|
""".format(os.environ['PIPENV_TEST_INDEX']).strip()
|
|
f.write(contents)
|
|
|
|
if lock_first:
|
|
# force source to be cached
|
|
c = p.pipenv('lock')
|
|
assert c.returncode == 0
|
|
project = Project()
|
|
sources = [
|
|
['pypi', 'https://pypi.org/simple'],
|
|
['testindex', os.environ.get('PIPENV_TEST_INDEX')]
|
|
]
|
|
for src in sources:
|
|
name, url = src
|
|
source = [s for s in project.pipfile_sources() if s.get('name') == name]
|
|
assert source
|
|
source = source[0]
|
|
assert source['name'] == name
|
|
assert source['url'] == url
|
|
assert sorted(source.items()) == sorted(project.get_source(name=name).items())
|
|
assert sorted(source.items()) == sorted(project.get_source(url=url).items())
|
|
assert sorted(source.items()) == sorted(project.find_source(name).items())
|
|
assert sorted(source.items()) == sorted(project.find_source(url).items())
|
|
|
|
|
|
@pytest.mark.install
|
|
@pytest.mark.project
|
|
@pytest.mark.parametrize('newlines', ['\n', '\r\n'])
|
|
def test_maintain_file_line_endings(pipenv_instance_pypi, newlines):
|
|
with pipenv_instance_pypi(chdir=True) as p:
|
|
# Initial pipfile + lockfile generation
|
|
c = p.pipenv('install pytz')
|
|
assert c.returncode == 0
|
|
|
|
# Rewrite each file with parameterized newlines
|
|
for fn in [p.pipfile_path, p.lockfile_path]:
|
|
with open(fn) as f:
|
|
contents = f.read()
|
|
written_newlines = f.newlines
|
|
|
|
assert written_newlines == '\n', '{!r} != {!r} for {}'.format(
|
|
written_newlines, '\n', fn,
|
|
)
|
|
# message because of https://github.com/pytest-dev/pytest/issues/3443
|
|
with open(fn, 'w', newline=newlines) as f:
|
|
f.write(contents)
|
|
|
|
# Run pipenv install to programatically rewrite
|
|
c = p.pipenv('install chardet')
|
|
assert c.returncode == 0
|
|
|
|
# Make sure we kept the right newlines
|
|
for fn in [p.pipfile_path, p.lockfile_path]:
|
|
with open(fn) as f:
|
|
f.read() # Consumes the content to detect newlines.
|
|
actual_newlines = f.newlines
|
|
assert actual_newlines == newlines, '{!r} != {!r} for {}'.format(
|
|
actual_newlines, newlines, fn,
|
|
)
|
|
# message because of https://github.com/pytest-dev/pytest/issues/3443
|
|
|
|
|
|
@pytest.mark.project
|
|
@pytest.mark.sources
|
|
@pytest.mark.needs_internet
|
|
def test_many_indexes(pipenv_instance_pypi):
|
|
with pipenv_instance_pypi(chdir=True) as p:
|
|
with open(p.pipfile_path, 'w') as f:
|
|
contents = """
|
|
[[source]]
|
|
url = "{0}"
|
|
verify_ssl = false
|
|
name = "testindex"
|
|
|
|
[[source]]
|
|
url = "https://pypi.org/simple"
|
|
verify_ssl = true
|
|
name = "pypi"
|
|
|
|
[[source]]
|
|
url = "https://pypi.python.org/simple"
|
|
verify_ssl = true
|
|
name = "legacy"
|
|
|
|
[packages]
|
|
pytz = "*"
|
|
six = {{version = "*", index = "pypi"}}
|
|
|
|
[dev-packages]
|
|
""".format(os.environ['PIPENV_TEST_INDEX']).strip()
|
|
f.write(contents)
|
|
c = p.pipenv('install')
|
|
assert c.returncode == 0
|
|
|
|
|
|
@pytest.mark.project
|
|
@pytest.mark.virtualenv
|
|
def test_run_in_virtualenv_with_global_context(pipenv_instance_pypi, virtualenv):
|
|
with pipenv_instance_pypi(chdir=True, venv_root=virtualenv.as_posix(), ignore_virtualenvs=False, venv_in_project=False) as p:
|
|
c = p.pipenv("run pip freeze")
|
|
assert c.returncode == 0, (c.stdout, c.stderr)
|
|
assert 'Creating a virtualenv' not in c.stderr, c.stderr
|
|
project = Project()
|
|
assert Path(project.virtualenv_location).resolve() == Path(virtualenv), (
|
|
project.virtualenv_location, str(virtualenv)
|
|
)
|
|
|
|
c = p.pipenv(f"run pip install -i {p.index_url} click")
|
|
assert c.returncode == 0, (c.stdout, c.stderr)
|
|
assert "Courtesy Notice" in c.stderr, (c.stdout, c.stderr)
|
|
|
|
c = p.pipenv("install six")
|
|
assert c.returncode == 0, (c.stdout, c.stderr)
|
|
|
|
c = p.pipenv("run python -c 'import click;print(click.__file__)'")
|
|
assert c.returncode == 0, (c.stdout, c.stderr)
|
|
assert is_in_path(c.stdout.strip(), str(virtualenv)), (c.stdout.strip(), str(virtualenv))
|
|
|
|
c = p.pipenv("clean --dry-run")
|
|
assert c.returncode == 0, (c.stdout, c.stderr)
|
|
assert "click" in c.stdout, c.stdout
|
|
|
|
|
|
@pytest.mark.project
|
|
@pytest.mark.virtualenv
|
|
def test_run_in_virtualenv(pipenv_instance_pypi):
|
|
with pipenv_instance_pypi(chdir=True) as p:
|
|
c = p.pipenv('run pip freeze')
|
|
assert c.returncode == 0
|
|
assert 'Creating a virtualenv' in c.stderr
|
|
project = Project()
|
|
c = p.pipenv("run pip install click")
|
|
assert c.returncode == 0
|
|
c = p.pipenv("install six")
|
|
assert c.returncode == 0
|
|
c = p.pipenv('run python -c "import click;print(click.__file__)"')
|
|
assert c.returncode == 0
|
|
assert normalize_path(c.stdout.strip()).startswith(
|
|
normalize_path(str(project.virtualenv_location))
|
|
)
|
|
c = p.pipenv("clean --dry-run")
|
|
assert c.returncode == 0
|
|
assert "click" in c.stdout
|
|
|
|
|
|
@pytest.mark.project
|
|
@pytest.mark.sources
|
|
def test_no_sources_in_pipfile(pipenv_instance_pypi):
|
|
with pipenv_instance_pypi(chdir=True) as p:
|
|
with open(p.pipfile_path, 'w') as f:
|
|
contents = """
|
|
[packages]
|
|
pytest = "*"
|
|
""".strip()
|
|
f.write(contents)
|
|
c = p.pipenv('install --skip-lock')
|
|
assert c.returncode == 0
|