mirror of
https://github.com/kennethreitz/pipenv.git
synced 2026-06-05 06:46:15 +00:00
Dan Ryan
107 lines
3.4 KiB
Python
107 lines
3.4 KiB
Python
# -*- coding=utf-8 -*-
|
|
import io
|
|
import pytest
|
|
import os
|
|
from pipenv.project import Project
|
|
from pipenv.utils import temp_environ
|
|
from pipenv.patched import pipfile
|
|
|
|
|
|
@pytest.mark.project
|
|
@pytest.mark.sources
|
|
@pytest.mark.environ
|
|
def test_pipfile_envvar_expansion(PipenvInstance):
|
|
with PipenvInstance(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(p.pipfile_path))
|
|
|
|
|
|
@pytest.mark.project
|
|
@pytest.mark.sources
|
|
@pytest.mark.parametrize('lock_first', [True, False])
|
|
def test_get_source(PipenvInstance, pypi, lock_first):
|
|
with PipenvInstance(pypi=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.python.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.return_code == 0
|
|
project = Project()
|
|
sources = [
|
|
['pypi', 'https://pypi.python.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', [u'\n', u'\r\n'])
|
|
def test_maintain_file_line_endings(PipenvInstance, pypi, newlines):
|
|
with PipenvInstance(pypi=pypi, chdir=True) as p:
|
|
# Initial pipfile + lockfile generation
|
|
c = p.pipenv('install pytz')
|
|
assert c.return_code == 0
|
|
|
|
# Rewrite each file with parameterized newlines
|
|
for fn in [p.pipfile_path, p.lockfile_path]:
|
|
with io.open(fn) as f:
|
|
contents = f.read()
|
|
written_newlines = f.newlines
|
|
assert written_newlines == u'\n'
|
|
with io.open(fn, 'w', newline=newlines) as f:
|
|
f.write(contents)
|
|
|
|
# Run pipenv install to programatically rewrite
|
|
c = p.pipenv('install chardet')
|
|
assert c.return_code == 0
|
|
|
|
# Make sure we kept the right newlines
|
|
for fn in [p.pipfile_path, p.lockfile_path]:
|
|
with io.open(fn) as f:
|
|
contents = f.read()
|
|
actual_newlines = f.newlines
|
|
assert actual_newlines == newlines
|