mirror of https://github.com/pypa/pip
821 lines
25 KiB
Python
821 lines
25 KiB
Python
import os.path
|
|
import shutil
|
|
import textwrap
|
|
from hashlib import sha256
|
|
|
|
import pytest
|
|
from pip._vendor.six import PY2
|
|
|
|
from pip._internal.cli.status_codes import ERROR
|
|
from pip._internal.utils.urls import path_to_url
|
|
from tests.lib import create_really_basic_wheel
|
|
from tests.lib.path import Path
|
|
from tests.lib.server import file_response
|
|
|
|
|
|
def fake_wheel(data, wheel_path):
|
|
wheel_name = os.path.basename(wheel_path)
|
|
name, version, rest = wheel_name.split("-", 2)
|
|
wheel_data = create_really_basic_wheel(name, version)
|
|
data.packages.joinpath(wheel_path).write_bytes(wheel_data)
|
|
|
|
|
|
@pytest.mark.network
|
|
def test_download_if_requested(script):
|
|
"""
|
|
It should download (in the scratch path) and not install if requested.
|
|
"""
|
|
result = script.pip(
|
|
'download', '-d', 'pip_downloads', 'INITools==0.1'
|
|
)
|
|
result.did_create(
|
|
Path('scratch') / 'pip_downloads' / 'INITools-0.1.tar.gz'
|
|
)
|
|
result.did_not_create(script.site_packages / 'initools')
|
|
|
|
|
|
@pytest.mark.network
|
|
def test_basic_download_setuptools(script):
|
|
"""
|
|
It should download (in the scratch path) and not install if requested.
|
|
"""
|
|
result = script.pip('download', 'setuptools')
|
|
setuptools_prefix = str(Path('scratch') / 'setuptools')
|
|
assert any(
|
|
path.startswith(setuptools_prefix) for path in result.files_created
|
|
)
|
|
|
|
|
|
def test_download_wheel(script, data):
|
|
"""
|
|
Test using "pip download" to download a *.whl archive.
|
|
"""
|
|
result = script.pip(
|
|
'download',
|
|
'--no-index',
|
|
'-f', data.packages,
|
|
'-d', '.', 'meta'
|
|
)
|
|
result.did_create(Path('scratch') / 'meta-1.0-py2.py3-none-any.whl')
|
|
result.did_not_create(script.site_packages / 'piptestpackage')
|
|
|
|
|
|
@pytest.mark.network
|
|
def test_single_download_from_requirements_file(script):
|
|
"""
|
|
It should support download (in the scratch path) from PyPI from a
|
|
requirements file
|
|
"""
|
|
script.scratch_path.joinpath("test-req.txt").write_text(textwrap.dedent("""
|
|
INITools==0.1
|
|
"""))
|
|
result = script.pip(
|
|
'download', '-r', script.scratch_path / 'test-req.txt', '-d', '.',
|
|
)
|
|
result.did_create(Path('scratch') / 'INITools-0.1.tar.gz')
|
|
result.did_not_create(script.site_packages / 'initools')
|
|
|
|
|
|
@pytest.mark.network
|
|
def test_basic_download_should_download_dependencies(script):
|
|
"""
|
|
It should download dependencies (in the scratch path)
|
|
"""
|
|
result = script.pip(
|
|
'download', 'Paste[openid]==1.7.5.1', '-d', '.'
|
|
)
|
|
result.did_create(Path('scratch') / 'Paste-1.7.5.1.tar.gz')
|
|
openid_tarball_prefix = str(Path('scratch') / 'python-openid-')
|
|
assert any(
|
|
path.startswith(openid_tarball_prefix) for path in result.files_created
|
|
)
|
|
result.did_not_create(script.site_packages / 'openid')
|
|
|
|
|
|
def test_download_wheel_archive(script, data):
|
|
"""
|
|
It should download a wheel archive path
|
|
"""
|
|
wheel_filename = 'colander-0.9.9-py2.py3-none-any.whl'
|
|
wheel_path = '/'.join((data.find_links, wheel_filename))
|
|
result = script.pip(
|
|
'download', wheel_path,
|
|
'-d', '.', '--no-deps'
|
|
)
|
|
result.did_create(Path('scratch') / wheel_filename)
|
|
|
|
|
|
def test_download_should_download_wheel_deps(script, data):
|
|
"""
|
|
It should download dependencies for wheels(in the scratch path)
|
|
"""
|
|
wheel_filename = 'colander-0.9.9-py2.py3-none-any.whl'
|
|
dep_filename = 'translationstring-1.1.tar.gz'
|
|
wheel_path = '/'.join((data.find_links, wheel_filename))
|
|
result = script.pip(
|
|
'download', wheel_path,
|
|
'-d', '.', '--find-links', data.find_links, '--no-index'
|
|
)
|
|
result.did_create(Path('scratch') / wheel_filename)
|
|
result.did_create(Path('scratch') / dep_filename)
|
|
|
|
|
|
@pytest.mark.network
|
|
def test_download_should_skip_existing_files(script):
|
|
"""
|
|
It should not download files already existing in the scratch dir
|
|
"""
|
|
script.scratch_path.joinpath("test-req.txt").write_text(textwrap.dedent("""
|
|
INITools==0.1
|
|
"""))
|
|
|
|
result = script.pip(
|
|
'download', '-r', script.scratch_path / 'test-req.txt', '-d', '.',
|
|
)
|
|
result.did_create(Path('scratch') / 'INITools-0.1.tar.gz')
|
|
result.did_not_create(script.site_packages / 'initools')
|
|
|
|
# adding second package to test-req.txt
|
|
script.scratch_path.joinpath("test-req.txt").write_text(textwrap.dedent("""
|
|
INITools==0.1
|
|
python-openid==2.2.5
|
|
"""))
|
|
|
|
# only the second package should be downloaded
|
|
result = script.pip(
|
|
'download', '-r', script.scratch_path / 'test-req.txt', '-d', '.',
|
|
)
|
|
openid_tarball_prefix = str(Path('scratch') / 'python-openid-')
|
|
assert any(
|
|
path.startswith(openid_tarball_prefix) for path in result.files_created
|
|
)
|
|
result.did_not_create(Path('scratch') / 'INITools-0.1.tar.gz')
|
|
result.did_not_create(script.site_packages / 'initools')
|
|
result.did_not_create(script.site_packages / 'openid')
|
|
|
|
|
|
@pytest.mark.network
|
|
def test_download_vcs_link(script):
|
|
"""
|
|
It should allow -d flag for vcs links, regression test for issue #798.
|
|
"""
|
|
result = script.pip(
|
|
'download', '-d', '.', 'git+git://github.com/pypa/pip-test-package.git'
|
|
)
|
|
result.did_create(Path('scratch') / 'pip-test-package-0.1.1.zip')
|
|
result.did_not_create(script.site_packages / 'piptestpackage')
|
|
|
|
|
|
def test_only_binary_set_then_download_specific_platform(script, data):
|
|
"""
|
|
Confirm that specifying an interpreter/platform constraint
|
|
is allowed when ``--only-binary=:all:`` is set.
|
|
"""
|
|
fake_wheel(data, 'fake-1.0-py2.py3-none-any.whl')
|
|
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--platform', 'linux_x86_64',
|
|
'fake'
|
|
)
|
|
result.did_create(Path('scratch') / 'fake-1.0-py2.py3-none-any.whl')
|
|
|
|
|
|
def test_no_deps_set_then_download_specific_platform(script, data):
|
|
"""
|
|
Confirm that specifying an interpreter/platform constraint
|
|
is allowed when ``--no-deps`` is set.
|
|
"""
|
|
fake_wheel(data, 'fake-1.0-py2.py3-none-any.whl')
|
|
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--no-deps',
|
|
'--dest', '.',
|
|
'--platform', 'linux_x86_64',
|
|
'fake'
|
|
)
|
|
result.did_create(Path('scratch') / 'fake-1.0-py2.py3-none-any.whl')
|
|
|
|
|
|
def test_download_specific_platform_fails(script, data):
|
|
"""
|
|
Confirm that specifying an interpreter/platform constraint
|
|
enforces that ``--no-deps`` or ``--only-binary=:all:`` is set.
|
|
"""
|
|
fake_wheel(data, 'fake-1.0-py2.py3-none-any.whl')
|
|
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--dest', '.',
|
|
'--platform', 'linux_x86_64',
|
|
'fake',
|
|
expect_error=True,
|
|
)
|
|
assert '--only-binary=:all:' in result.stderr
|
|
|
|
|
|
def test_no_binary_set_then_download_specific_platform_fails(script, data):
|
|
"""
|
|
Confirm that specifying an interpreter/platform constraint
|
|
enforces that ``--only-binary=:all:`` is set without ``--no-binary``.
|
|
"""
|
|
fake_wheel(data, 'fake-1.0-py2.py3-none-any.whl')
|
|
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--no-binary=fake',
|
|
'--dest', '.',
|
|
'--platform', 'linux_x86_64',
|
|
'fake',
|
|
expect_error=True,
|
|
)
|
|
assert '--only-binary=:all:' in result.stderr
|
|
|
|
|
|
def test_download_specify_platform(script, data):
|
|
"""
|
|
Test using "pip download --platform" to download a .whl archive
|
|
supported for a specific platform
|
|
"""
|
|
fake_wheel(data, 'fake-1.0-py2.py3-none-any.whl')
|
|
|
|
# Confirm that universal wheels are returned even for specific
|
|
# platforms.
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--platform', 'linux_x86_64',
|
|
'fake'
|
|
)
|
|
result.did_create(Path('scratch') / 'fake-1.0-py2.py3-none-any.whl')
|
|
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--platform', 'macosx_10_9_x86_64',
|
|
'fake'
|
|
)
|
|
|
|
data.reset()
|
|
fake_wheel(data, 'fake-1.0-py2.py3-none-macosx_10_9_x86_64.whl')
|
|
fake_wheel(data, 'fake-2.0-py2.py3-none-linux_x86_64.whl')
|
|
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--platform', 'macosx_10_10_x86_64',
|
|
'fake'
|
|
)
|
|
result.did_create(
|
|
Path('scratch') /
|
|
'fake-1.0-py2.py3-none-macosx_10_9_x86_64.whl'
|
|
)
|
|
|
|
# OSX platform wheels are not backward-compatible.
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--platform', 'macosx_10_8_x86_64',
|
|
'fake',
|
|
expect_error=True,
|
|
)
|
|
|
|
# No linux wheel provided for this version.
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--platform', 'linux_x86_64',
|
|
'fake==1',
|
|
expect_error=True,
|
|
)
|
|
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--platform', 'linux_x86_64',
|
|
'fake==2'
|
|
)
|
|
result.did_create(
|
|
Path('scratch') / 'fake-2.0-py2.py3-none-linux_x86_64.whl'
|
|
)
|
|
|
|
|
|
class TestDownloadPlatformManylinuxes(object):
|
|
"""
|
|
"pip download --platform" downloads a .whl archive supported for
|
|
manylinux platforms.
|
|
"""
|
|
|
|
@pytest.mark.parametrize("platform", [
|
|
"linux_x86_64",
|
|
"manylinux1_x86_64",
|
|
"manylinux2010_x86_64",
|
|
"manylinux2014_x86_64",
|
|
])
|
|
def test_download_universal(self, platform, script, data):
|
|
"""
|
|
Universal wheels are returned even for specific platforms.
|
|
"""
|
|
fake_wheel(data, 'fake-1.0-py2.py3-none-any.whl')
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--platform', platform,
|
|
'fake',
|
|
)
|
|
result.did_create(Path('scratch') / 'fake-1.0-py2.py3-none-any.whl')
|
|
|
|
@pytest.mark.parametrize("wheel_abi,platform", [
|
|
("manylinux1_x86_64", "manylinux1_x86_64"),
|
|
("manylinux1_x86_64", "manylinux2010_x86_64"),
|
|
("manylinux2010_x86_64", "manylinux2010_x86_64"),
|
|
("manylinux1_x86_64", "manylinux2014_x86_64"),
|
|
("manylinux2010_x86_64", "manylinux2014_x86_64"),
|
|
("manylinux2014_x86_64", "manylinux2014_x86_64"),
|
|
])
|
|
def test_download_compatible_manylinuxes(
|
|
self, wheel_abi, platform, script, data,
|
|
):
|
|
"""
|
|
Earlier manylinuxes are compatible with later manylinuxes.
|
|
"""
|
|
wheel = 'fake-1.0-py2.py3-none-{}.whl'.format(wheel_abi)
|
|
fake_wheel(data, wheel)
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--platform', platform,
|
|
'fake',
|
|
)
|
|
result.did_create(Path('scratch') / wheel)
|
|
|
|
def test_explicit_platform_only(self, data, script):
|
|
"""
|
|
When specifying the platform, manylinux1 needs to be the
|
|
explicit platform--it won't ever be added to the compatible
|
|
tags.
|
|
"""
|
|
fake_wheel(data, 'fake-1.0-py2.py3-none-linux_x86_64.whl')
|
|
script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--platform', 'linux_x86_64',
|
|
'fake',
|
|
)
|
|
|
|
|
|
def test_download__python_version(script, data):
|
|
"""
|
|
Test using "pip download --python-version" to download a .whl archive
|
|
supported for a specific interpreter
|
|
"""
|
|
fake_wheel(data, 'fake-1.0-py2.py3-none-any.whl')
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--python-version', '2',
|
|
'fake'
|
|
)
|
|
result.did_create(Path('scratch') / 'fake-1.0-py2.py3-none-any.whl')
|
|
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--python-version', '3',
|
|
'fake'
|
|
)
|
|
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--python-version', '27',
|
|
'fake'
|
|
)
|
|
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--python-version', '33',
|
|
'fake'
|
|
)
|
|
|
|
data.reset()
|
|
fake_wheel(data, 'fake-1.0-py2-none-any.whl')
|
|
fake_wheel(data, 'fake-2.0-py3-none-any.whl')
|
|
|
|
# No py3 provided for version 1.
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--python-version', '3',
|
|
'fake==1.0',
|
|
expect_error=True,
|
|
)
|
|
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--python-version', '2',
|
|
'fake'
|
|
)
|
|
result.did_create(Path('scratch') / 'fake-1.0-py2-none-any.whl')
|
|
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--python-version', '26',
|
|
'fake'
|
|
)
|
|
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--python-version', '3',
|
|
'fake'
|
|
)
|
|
result.did_create(Path('scratch') / 'fake-2.0-py3-none-any.whl')
|
|
|
|
|
|
def make_wheel_with_python_requires(script, package_name, python_requires):
|
|
"""
|
|
Create a wheel using the given python_requires.
|
|
|
|
:return: the path to the wheel file.
|
|
"""
|
|
package_dir = script.scratch_path / package_name
|
|
package_dir.mkdir()
|
|
|
|
text = textwrap.dedent("""\
|
|
from setuptools import setup
|
|
setup(name='{}',
|
|
python_requires='{}',
|
|
version='1.0')
|
|
""").format(package_name, python_requires)
|
|
package_dir.joinpath('setup.py').write_text(text)
|
|
script.run(
|
|
'python', 'setup.py', 'bdist_wheel', '--universal', cwd=package_dir,
|
|
allow_stderr_warning=PY2,
|
|
)
|
|
|
|
file_name = '{}-1.0-py2.py3-none-any.whl'.format(package_name)
|
|
return package_dir / 'dist' / file_name
|
|
|
|
|
|
def test_download__python_version_used_for_python_requires(
|
|
script, data, with_wheel,
|
|
):
|
|
"""
|
|
Test that --python-version is used for the Requires-Python check.
|
|
"""
|
|
wheel_path = make_wheel_with_python_requires(
|
|
script, 'mypackage', python_requires='==3.2',
|
|
)
|
|
wheel_dir = os.path.dirname(wheel_path)
|
|
|
|
def make_args(python_version):
|
|
return [
|
|
'download', '--no-index', '--find-links', wheel_dir,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--python-version', python_version,
|
|
'mypackage==1.0',
|
|
]
|
|
|
|
args = make_args('33')
|
|
result = script.pip(*args, expect_error=True)
|
|
expected_err = (
|
|
"ERROR: Package 'mypackage' requires a different Python: "
|
|
"3.3.0 not in '==3.2'"
|
|
)
|
|
assert expected_err in result.stderr, 'stderr: {}'.format(result.stderr)
|
|
|
|
# Now try with a --python-version that satisfies the Requires-Python.
|
|
args = make_args('32')
|
|
script.pip(*args) # no exception
|
|
|
|
|
|
def test_download_specify_abi(script, data):
|
|
"""
|
|
Test using "pip download --abi" to download a .whl archive
|
|
supported for a specific abi
|
|
"""
|
|
fake_wheel(data, 'fake-1.0-py2.py3-none-any.whl')
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--implementation', 'fk',
|
|
'--abi', 'fake_abi',
|
|
'fake'
|
|
)
|
|
result.did_create(Path('scratch') / 'fake-1.0-py2.py3-none-any.whl')
|
|
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--implementation', 'fk',
|
|
'--abi', 'none',
|
|
'fake'
|
|
)
|
|
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--abi', 'cp27m',
|
|
'fake',
|
|
)
|
|
|
|
data.reset()
|
|
fake_wheel(data, 'fake-1.0-fk2-fakeabi-fake_platform.whl')
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--python-version', '2',
|
|
'--implementation', 'fk',
|
|
'--platform', 'fake_platform',
|
|
'--abi', 'fakeabi',
|
|
'fake'
|
|
)
|
|
result.did_create(
|
|
Path('scratch') / 'fake-1.0-fk2-fakeabi-fake_platform.whl'
|
|
)
|
|
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--implementation', 'fk',
|
|
'--platform', 'fake_platform',
|
|
'--abi', 'none',
|
|
'fake',
|
|
expect_error=True,
|
|
)
|
|
|
|
|
|
def test_download_specify_implementation(script, data):
|
|
"""
|
|
Test using "pip download --abi" to download a .whl archive
|
|
supported for a specific abi
|
|
"""
|
|
fake_wheel(data, 'fake-1.0-py2.py3-none-any.whl')
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--implementation', 'fk',
|
|
'fake'
|
|
)
|
|
result.did_create(Path('scratch') / 'fake-1.0-py2.py3-none-any.whl')
|
|
|
|
data.reset()
|
|
fake_wheel(data, 'fake-1.0-fk3-none-any.whl')
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--implementation', 'fk',
|
|
'--python-version', '3',
|
|
'fake'
|
|
)
|
|
result.did_create(Path('scratch') / 'fake-1.0-fk3-none-any.whl')
|
|
|
|
result = script.pip(
|
|
'download', '--no-index', '--find-links', data.find_links,
|
|
'--only-binary=:all:',
|
|
'--dest', '.',
|
|
'--implementation', 'fk',
|
|
'--python-version', '2',
|
|
'fake',
|
|
expect_error=True,
|
|
)
|
|
|
|
|
|
def test_download_exit_status_code_when_no_requirements(script):
|
|
"""
|
|
Test download exit status code when no requirements specified
|
|
"""
|
|
result = script.pip('download', expect_error=True)
|
|
assert (
|
|
"You must give at least one requirement to download" in result.stderr
|
|
)
|
|
assert result.returncode == ERROR
|
|
|
|
|
|
def test_download_exit_status_code_when_blank_requirements_file(script):
|
|
"""
|
|
Test download exit status code when blank requirements file specified
|
|
"""
|
|
script.scratch_path.joinpath("blank.txt").write_text("\n")
|
|
script.pip('download', '-r', 'blank.txt')
|
|
|
|
|
|
def test_download_prefer_binary_when_tarball_higher_than_wheel(script, data):
|
|
fake_wheel(data, 'source-0.8-py2.py3-none-any.whl')
|
|
result = script.pip(
|
|
'download',
|
|
'--prefer-binary',
|
|
'--no-index',
|
|
'-f', data.packages,
|
|
'-d', '.', 'source'
|
|
)
|
|
result.did_create(Path('scratch') / 'source-0.8-py2.py3-none-any.whl')
|
|
result.did_not_create(Path('scratch') / 'source-1.0.tar.gz')
|
|
|
|
|
|
def test_prefer_binary_tarball_higher_than_wheel_req_file(script, data):
|
|
fake_wheel(data, 'source-0.8-py2.py3-none-any.whl')
|
|
script.scratch_path.joinpath("test-req.txt").write_text(textwrap.dedent("""
|
|
--prefer-binary
|
|
source
|
|
"""))
|
|
result = script.pip(
|
|
'download',
|
|
'-r', script.scratch_path / 'test-req.txt',
|
|
'--no-index',
|
|
'-f', data.packages,
|
|
'-d', '.'
|
|
)
|
|
|
|
result.did_create(Path('scratch') / 'source-0.8-py2.py3-none-any.whl')
|
|
result.did_not_create(Path('scratch') / 'source-1.0.tar.gz')
|
|
|
|
|
|
def test_download_prefer_binary_when_wheel_doesnt_satisfy_req(script, data):
|
|
fake_wheel(data, 'source-0.8-py2.py3-none-any.whl')
|
|
script.scratch_path.joinpath("test-req.txt").write_text(textwrap.dedent("""
|
|
source>0.9
|
|
"""))
|
|
|
|
result = script.pip(
|
|
'download',
|
|
'--prefer-binary',
|
|
'--no-index',
|
|
'-f', data.packages,
|
|
'-d', '.',
|
|
'-r', script.scratch_path / 'test-req.txt'
|
|
)
|
|
result.did_create(Path('scratch') / 'source-1.0.tar.gz')
|
|
result.did_not_create(Path('scratch') / 'source-0.8-py2.py3-none-any.whl')
|
|
|
|
|
|
def test_prefer_binary_when_wheel_doesnt_satisfy_req_req_file(script, data):
|
|
fake_wheel(data, 'source-0.8-py2.py3-none-any.whl')
|
|
script.scratch_path.joinpath("test-req.txt").write_text(textwrap.dedent("""
|
|
--prefer-binary
|
|
source>0.9
|
|
"""))
|
|
|
|
result = script.pip(
|
|
'download',
|
|
'--no-index',
|
|
'-f', data.packages,
|
|
'-d', '.',
|
|
'-r', script.scratch_path / 'test-req.txt'
|
|
)
|
|
result.did_create(Path('scratch') / 'source-1.0.tar.gz')
|
|
result.did_not_create(Path('scratch') / 'source-0.8-py2.py3-none-any.whl')
|
|
|
|
|
|
def test_download_prefer_binary_when_only_tarball_exists(script, data):
|
|
result = script.pip(
|
|
'download',
|
|
'--prefer-binary',
|
|
'--no-index',
|
|
'-f', data.packages,
|
|
'-d', '.', 'source'
|
|
)
|
|
result.did_create(Path('scratch') / 'source-1.0.tar.gz')
|
|
|
|
|
|
def test_prefer_binary_when_only_tarball_exists_req_file(script, data):
|
|
script.scratch_path.joinpath("test-req.txt").write_text(textwrap.dedent("""
|
|
--prefer-binary
|
|
source
|
|
"""))
|
|
result = script.pip(
|
|
'download',
|
|
'--no-index',
|
|
'-f', data.packages,
|
|
'-d', '.',
|
|
'-r', script.scratch_path / 'test-req.txt'
|
|
)
|
|
result.did_create(Path('scratch') / 'source-1.0.tar.gz')
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def shared_script(tmpdir_factory, script_factory):
|
|
tmpdir = Path(str(tmpdir_factory.mktemp("download_shared_script")))
|
|
script = script_factory(tmpdir.joinpath("workspace"))
|
|
return script
|
|
|
|
|
|
def test_download_file_url(shared_script, shared_data, tmpdir):
|
|
download_dir = tmpdir / 'download'
|
|
download_dir.mkdir()
|
|
downloaded_path = download_dir / 'simple-1.0.tar.gz'
|
|
|
|
simple_pkg = shared_data.packages / 'simple-1.0.tar.gz'
|
|
|
|
shared_script.pip(
|
|
'download',
|
|
'-d',
|
|
str(download_dir),
|
|
'--no-index',
|
|
path_to_url(str(simple_pkg)),
|
|
)
|
|
|
|
assert downloaded_path.exists()
|
|
assert simple_pkg.read_bytes() == downloaded_path.read_bytes()
|
|
|
|
|
|
def test_download_file_url_existing_ok_download(
|
|
shared_script, shared_data, tmpdir
|
|
):
|
|
download_dir = tmpdir / 'download'
|
|
download_dir.mkdir()
|
|
downloaded_path = download_dir / 'simple-1.0.tar.gz'
|
|
fake_existing_package = shared_data.packages / 'simple-2.0.tar.gz'
|
|
shutil.copy(str(fake_existing_package), str(downloaded_path))
|
|
downloaded_path_bytes = downloaded_path.read_bytes()
|
|
digest = sha256(downloaded_path_bytes).hexdigest()
|
|
|
|
simple_pkg = shared_data.packages / 'simple-1.0.tar.gz'
|
|
url = "{}#sha256={}".format(path_to_url(simple_pkg), digest)
|
|
|
|
shared_script.pip('download', '-d', str(download_dir), url)
|
|
|
|
assert downloaded_path_bytes == downloaded_path.read_bytes()
|
|
|
|
|
|
def test_download_file_url_existing_bad_download(
|
|
shared_script, shared_data, tmpdir
|
|
):
|
|
download_dir = tmpdir / 'download'
|
|
download_dir.mkdir()
|
|
downloaded_path = download_dir / 'simple-1.0.tar.gz'
|
|
fake_existing_package = shared_data.packages / 'simple-2.0.tar.gz'
|
|
shutil.copy(str(fake_existing_package), str(downloaded_path))
|
|
|
|
simple_pkg = shared_data.packages / 'simple-1.0.tar.gz'
|
|
simple_pkg_bytes = simple_pkg.read_bytes()
|
|
digest = sha256(simple_pkg_bytes).hexdigest()
|
|
url = "{}#sha256={}".format(path_to_url(simple_pkg), digest)
|
|
|
|
shared_script.pip('download', '-d', str(download_dir), url)
|
|
|
|
assert simple_pkg_bytes == downloaded_path.read_bytes()
|
|
|
|
|
|
def test_download_http_url_bad_hash(
|
|
shared_script, shared_data, tmpdir, mock_server
|
|
):
|
|
download_dir = tmpdir / 'download'
|
|
download_dir.mkdir()
|
|
downloaded_path = download_dir / 'simple-1.0.tar.gz'
|
|
fake_existing_package = shared_data.packages / 'simple-2.0.tar.gz'
|
|
shutil.copy(str(fake_existing_package), str(downloaded_path))
|
|
|
|
simple_pkg = shared_data.packages / 'simple-1.0.tar.gz'
|
|
simple_pkg_bytes = simple_pkg.read_bytes()
|
|
digest = sha256(simple_pkg_bytes).hexdigest()
|
|
mock_server.set_responses([
|
|
file_response(simple_pkg)
|
|
])
|
|
mock_server.start()
|
|
base_address = 'http://{}:{}'.format(mock_server.host, mock_server.port)
|
|
url = "{}/simple-1.0.tar.gz#sha256={}".format(base_address, digest)
|
|
|
|
shared_script.pip('download', '-d', str(download_dir), url)
|
|
|
|
assert simple_pkg_bytes == downloaded_path.read_bytes()
|
|
|
|
mock_server.stop()
|
|
requests = mock_server.get_requests()
|
|
assert len(requests) == 1
|
|
assert requests[0]['PATH_INFO'] == '/simple-1.0.tar.gz'
|
|
assert requests[0]['HTTP_ACCEPT_ENCODING'] == 'identity'
|