mirror of
https://github.com/pypa/pip
synced 2023-12-13 21:30:23 +01:00
139 lines
4.9 KiB
Python
139 lines
4.9 KiB
Python
import errno
|
|
|
|
import pytest
|
|
from mock import Mock, call, patch
|
|
|
|
from pip._internal.commands.install import (
|
|
build_wheels,
|
|
create_env_error_message,
|
|
decide_user_install,
|
|
)
|
|
|
|
|
|
class TestWheelCache:
|
|
|
|
def check_build_wheels(
|
|
self,
|
|
pep517_requirements,
|
|
legacy_requirements,
|
|
):
|
|
"""
|
|
Return: (mock_calls, return_value).
|
|
"""
|
|
def build(reqs, **kwargs):
|
|
# Fail the first requirement.
|
|
return [reqs[0]]
|
|
|
|
builder = Mock()
|
|
builder.build.side_effect = build
|
|
|
|
build_failures = build_wheels(
|
|
builder=builder,
|
|
pep517_requirements=pep517_requirements,
|
|
legacy_requirements=legacy_requirements,
|
|
)
|
|
|
|
return (builder.build.mock_calls, build_failures)
|
|
|
|
@patch('pip._internal.commands.install.is_wheel_installed')
|
|
def test_build_wheels__wheel_installed(self, is_wheel_installed):
|
|
is_wheel_installed.return_value = True
|
|
|
|
mock_calls, build_failures = self.check_build_wheels(
|
|
pep517_requirements=['a', 'b'],
|
|
legacy_requirements=['c', 'd'],
|
|
)
|
|
|
|
# Legacy requirements were built.
|
|
assert mock_calls == [
|
|
call(['a', 'b'], should_unpack=True),
|
|
call(['c', 'd'], should_unpack=True),
|
|
]
|
|
|
|
# Legacy build failures are not included in the return value.
|
|
assert build_failures == ['a']
|
|
|
|
@patch('pip._internal.commands.install.is_wheel_installed')
|
|
def test_build_wheels__wheel_not_installed(self, is_wheel_installed):
|
|
is_wheel_installed.return_value = False
|
|
|
|
mock_calls, build_failures = self.check_build_wheels(
|
|
pep517_requirements=['a', 'b'],
|
|
legacy_requirements=['c', 'd'],
|
|
)
|
|
|
|
# Legacy requirements were not built.
|
|
assert mock_calls == [
|
|
call(['a', 'b'], should_unpack=True),
|
|
]
|
|
|
|
assert build_failures == ['a']
|
|
|
|
|
|
class TestDecideUserInstall:
|
|
@patch('site.ENABLE_USER_SITE', True)
|
|
@patch('pip._internal.commands.install.site_packages_writable')
|
|
def test_prefix_and_target(self, sp_writable):
|
|
sp_writable.return_value = False
|
|
|
|
assert decide_user_install(
|
|
use_user_site=None, prefix_path='foo'
|
|
) is False
|
|
|
|
assert decide_user_install(
|
|
use_user_site=None, target_dir='bar'
|
|
) is False
|
|
|
|
@pytest.mark.parametrize(
|
|
"enable_user_site,site_packages_writable,result", [
|
|
(True, True, False),
|
|
(True, False, True),
|
|
(False, True, False),
|
|
(False, False, False),
|
|
])
|
|
def test_most_cases(
|
|
self, enable_user_site, site_packages_writable, result, monkeypatch,
|
|
):
|
|
monkeypatch.setattr('site.ENABLE_USER_SITE', enable_user_site)
|
|
monkeypatch.setattr(
|
|
'pip._internal.commands.install.site_packages_writable',
|
|
lambda **kw: site_packages_writable
|
|
)
|
|
assert decide_user_install(use_user_site=None) is result
|
|
|
|
|
|
@pytest.mark.parametrize('error, show_traceback, using_user_site, expected', [
|
|
# show_traceback = True, using_user_site = True
|
|
(EnvironmentError("Illegal byte sequence"), True, True, 'Could not install'
|
|
' packages due to an EnvironmentError.\n'),
|
|
(EnvironmentError(errno.EACCES, "No file permission"), True, True, 'Could'
|
|
' not install packages due to an EnvironmentError.\nCheck the'
|
|
' permissions.\n'),
|
|
# show_traceback = True, using_user_site = False
|
|
(EnvironmentError("Illegal byte sequence"), True, False, 'Could not'
|
|
' install packages due to an EnvironmentError.\n'),
|
|
(EnvironmentError(errno.EACCES, "No file permission"), True, False, 'Could'
|
|
' not install packages due to an EnvironmentError.\nConsider using the'
|
|
' `--user` option or check the permissions.\n'),
|
|
# show_traceback = False, using_user_site = True
|
|
(EnvironmentError("Illegal byte sequence"), False, True, 'Could not'
|
|
' install packages due to an EnvironmentError: Illegal byte'
|
|
' sequence\n'),
|
|
(EnvironmentError(errno.EACCES, "No file permission"), False, True, 'Could'
|
|
' not install packages due to an EnvironmentError: [Errno 13] No file'
|
|
' permission\nCheck the permissions.\n'),
|
|
# show_traceback = False, using_user_site = False
|
|
(EnvironmentError("Illegal byte sequence"), False, False, 'Could not'
|
|
' install packages due to an EnvironmentError: Illegal byte sequence'
|
|
'\n'),
|
|
(EnvironmentError(errno.EACCES, "No file permission"), False, False,
|
|
'Could not install packages due to an EnvironmentError: [Errno 13] No'
|
|
' file permission\nConsider using the `--user` option or check the'
|
|
' permissions.\n'),
|
|
])
|
|
def test_create_env_error_message(
|
|
error, show_traceback, using_user_site, expected
|
|
):
|
|
msg = create_env_error_message(error, show_traceback, using_user_site)
|
|
assert msg == expected
|