Files
pipenv/tests/integration/test_project.py
T
Dan Ryan f93b9f2428 Add tests for project source changes
Signed-off-by: Dan Ryan <dan@danryan.co>
2018-04-16 23:15:04 -04:00

107 lines
3.2 KiB
Python

# -*- coding=utf-8 -*-
import pytest
import os
from pipenv.project import Project
@pytest.mark.project
@pytest.mark.sources
def test_get_cached_source(PipenvInstance, pypi):
with PipenvInstance(pypi=pypi, chdir=True) as p:
# Make sure unparseable packages don't wind up in the pipfile
# Escape $ for shell input
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.get('PIPENV_TEST_INDEX')).strip()
f.write(contents)
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.sources if s.get('name') == name]
assert source[0]['name'] == name
assert source[0]['url'] == url
source = project.get_source(name=name)
assert source['name'] == name
assert source['url'] == url
source = project.get_source(url=url)
assert source['name'] == name
assert source['url'] == url
source = project.find_source(name)
assert source['name'] == name
assert source['url'] == url
source = project.find_source(url)
assert source['name'] == name
assert source['url'] == url
@pytest.mark.project
@pytest.mark.sources
def test_get_uncached_source(PipenvInstance, pypi):
with PipenvInstance(pypi=pypi, chdir=True) as p:
# Make sure unparseable packages don't wind up in the pipfile
# Escape $ for shell input
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.get('PIPENV_TEST_INDEX')).strip()
f.write(contents)
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[0]['name'] == name
assert source[0]['url'] == url
source = project.get_source(name=name)
assert source['name'] == name
assert source['url'] == url
source = project.get_source(url=url)
assert source['name'] == name
assert source['url'] == url
source = project.find_source(name)
assert source['name'] == name
assert source['url'] == url
source = project.find_source(url)
assert source['name'] == name
assert source['url'] == url