76 lines
2.7 KiB
Python
76 lines
2.7 KiB
Python
|
import os.path
|
||
|
|
import sys
|
||
|
|
from unittest import mock
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
import pre_commit.constants as C
|
||
|
|
from pre_commit.languages import python
|
||
|
|
from pre_commit.prefix import Prefix
|
||
|
|
|
||
|
|
|
||
|
|
def test_norm_version_expanduser():
|
||
|
|
home = os.path.expanduser('~')
|
||
|
|
if os.name == 'nt': # pragma: nt cover
|
||
|
|
path = r'~\python343'
|
||
|
|
expected_path = fr'{home}\python343'
|
||
|
|
else: # pragma: nt no cover
|
||
|
|
path = '~/.pyenv/versions/3.4.3/bin/python'
|
||
|
|
expected_path = f'{home}/.pyenv/versions/3.4.3/bin/python'
|
||
|
|
result = python.norm_version(path)
|
||
|
|
assert result == expected_path
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize('v', ('python3.6', 'python3', 'python'))
|
||
|
|
def test_sys_executable_matches(v):
|
||
|
|
with mock.patch.object(sys, 'version_info', (3, 6, 7)):
|
||
|
|
assert python._sys_executable_matches(v)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize('v', ('notpython', 'python3.x'))
|
||
|
|
def test_sys_executable_matches_does_not_match(v):
|
||
|
|
with mock.patch.object(sys, 'version_info', (3, 6, 7)):
|
||
|
|
assert not python._sys_executable_matches(v)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
('exe', 'realpath', 'expected'), (
|
||
|
|
('/usr/bin/python3', '/usr/bin/python3.7', 'python3'),
|
||
|
|
('/usr/bin/python', '/usr/bin/python3.7', 'python3.7'),
|
||
|
|
('/usr/bin/python', '/usr/bin/python', None),
|
||
|
|
('/usr/bin/python3.6m', '/usr/bin/python3.6m', 'python3.6m'),
|
||
|
|
('v/bin/python', 'v/bin/pypy', 'pypy'),
|
||
|
|
),
|
||
|
|
)
|
||
|
|
def test_find_by_sys_executable(exe, realpath, expected):
|
||
|
|
with mock.patch.object(sys, 'executable', exe):
|
||
|
|
with mock.patch.object(os.path, 'realpath', return_value=realpath):
|
||
|
|
with mock.patch.object(python, 'find_executable', lambda x: x):
|
||
|
|
assert python._find_by_sys_executable() == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_healthy_types_py_in_cwd(tmpdir):
|
||
|
|
with tmpdir.as_cwd():
|
||
|
|
prefix = tmpdir.join('prefix').ensure_dir()
|
||
|
|
prefix.join('setup.py').write('import setuptools; setuptools.setup()')
|
||
|
|
prefix = Prefix(str(prefix))
|
||
|
|
python.install_environment(prefix, C.DEFAULT, ())
|
||
|
|
|
||
|
|
# even if a `types.py` file exists, should still be healthy
|
||
|
|
tmpdir.join('types.py').ensure()
|
||
|
|
assert python.healthy(prefix, C.DEFAULT) is True
|
||
|
|
|
||
|
|
|
||
|
|
def test_healthy_python_goes_missing(tmpdir):
|
||
|
|
with tmpdir.as_cwd():
|
||
|
|
prefix = tmpdir.join('prefix').ensure_dir()
|
||
|
|
prefix.join('setup.py').write('import setuptools; setuptools.setup()')
|
||
|
|
prefix = Prefix(str(prefix))
|
||
|
|
python.install_environment(prefix, C.DEFAULT, ())
|
||
|
|
|
||
|
|
exe_name = 'python' if sys.platform != 'win32' else 'python.exe'
|
||
|
|
py_exe = prefix.path(python.bin_dir('py_env-default'), exe_name)
|
||
|
|
os.remove(py_exe)
|
||
|
|
|
||
|
|
assert python.healthy(prefix, C.DEFAULT) is False
|