mirror of
https://github.com/pre-commit/pre-commit.git
synced 2026-01-13 04:20:28 -06:00
136 lines
3.7 KiB
Python
136 lines
3.7 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import os
|
|
import subprocess
|
|
|
|
import mock
|
|
import pytest
|
|
|
|
from pre_commit import five
|
|
from pre_commit.prefixed_command_runner import PrefixedCommandRunner
|
|
from pre_commit.util import CalledProcessError
|
|
|
|
|
|
def norm_slash(input_tup):
|
|
return tuple(x.replace('/', os.sep) for x in input_tup)
|
|
|
|
|
|
def test_CalledProcessError_str():
|
|
error = CalledProcessError(
|
|
1,
|
|
[five.n('git'), five.n('status')],
|
|
0,
|
|
(five.n('stdout'), five.n('stderr')),
|
|
)
|
|
assert str(error) == (
|
|
"Command: ['git', 'status']\n"
|
|
"Return code: 1\n"
|
|
"Expected return code: 0\n"
|
|
"Output: \n"
|
|
" stdout\n"
|
|
"Errors: \n"
|
|
" stderr\n"
|
|
)
|
|
|
|
|
|
def test_CalledProcessError_str_nooutput():
|
|
error = CalledProcessError(
|
|
1, [five.n('git'), five.n('status')], 0, (five.n(''), five.n(''))
|
|
)
|
|
assert str(error) == (
|
|
"Command: ['git', 'status']\n"
|
|
"Return code: 1\n"
|
|
"Expected return code: 0\n"
|
|
"Output: (none)\n"
|
|
"Errors: (none)\n"
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def popen_mock():
|
|
popen = mock.Mock(spec=subprocess.Popen)
|
|
popen.return_value.communicate.return_value = (b'stdout', b'stderr')
|
|
return popen
|
|
|
|
|
|
@pytest.fixture
|
|
def makedirs_mock():
|
|
return mock.Mock(spec=os.makedirs)
|
|
|
|
|
|
@pytest.mark.parametrize(('input', 'expected_prefix'), (
|
|
norm_slash(('.', './')),
|
|
norm_slash(('foo', 'foo/')),
|
|
norm_slash(('bar/', 'bar/')),
|
|
norm_slash(('foo/bar', 'foo/bar/')),
|
|
norm_slash(('foo/bar/', 'foo/bar/')),
|
|
))
|
|
def test_init_normalizes_path_endings(input, expected_prefix):
|
|
input = input.replace('/', os.sep)
|
|
expected_prefix = expected_prefix.replace('/', os.sep)
|
|
instance = PrefixedCommandRunner(input)
|
|
assert instance.prefix_dir == expected_prefix
|
|
|
|
|
|
def test_run_substitutes_prefix(popen_mock, makedirs_mock):
|
|
instance = PrefixedCommandRunner(
|
|
'prefix', popen=popen_mock, makedirs=makedirs_mock,
|
|
)
|
|
ret = instance.run(['{prefix}bar', 'baz'], retcode=None)
|
|
popen_mock.assert_called_once_with(
|
|
(five.n(os.path.join('prefix', 'bar')), five.n('baz')),
|
|
env=None,
|
|
stdin=subprocess.PIPE,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
)
|
|
assert ret == (popen_mock.return_value.returncode, 'stdout', 'stderr')
|
|
|
|
|
|
PATH_TESTS = (
|
|
norm_slash(('foo', '', 'foo')),
|
|
norm_slash(('foo', 'bar', 'foo/bar')),
|
|
norm_slash(('foo/bar', '../baz', 'foo/baz')),
|
|
norm_slash(('./', 'bar', 'bar')),
|
|
norm_slash(('./', '', '.')),
|
|
norm_slash(('/tmp/foo', '/tmp/bar', '/tmp/bar')),
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(('prefix', 'path_end', 'expected_output'), PATH_TESTS)
|
|
def test_path(prefix, path_end, expected_output):
|
|
instance = PrefixedCommandRunner(prefix)
|
|
ret = instance.path(path_end)
|
|
assert ret == expected_output
|
|
|
|
|
|
def test_path_multiple_args():
|
|
instance = PrefixedCommandRunner('foo')
|
|
ret = instance.path('bar', 'baz')
|
|
assert ret == os.path.join('foo', 'bar', 'baz')
|
|
|
|
|
|
def test_create_path_if_not_exists(in_tmpdir):
|
|
instance = PrefixedCommandRunner('foo')
|
|
assert not os.path.exists('foo')
|
|
instance._create_path_if_not_exists()
|
|
assert os.path.exists('foo')
|
|
|
|
|
|
def test_exists_does_not_exist(in_tmpdir):
|
|
assert not PrefixedCommandRunner('.').exists('foo')
|
|
|
|
|
|
def test_exists_does_exist(in_tmpdir):
|
|
os.mkdir('foo')
|
|
assert PrefixedCommandRunner('.').exists('foo')
|
|
|
|
|
|
def test_raises_on_error(popen_mock, makedirs_mock):
|
|
popen_mock.return_value.returncode = 1
|
|
with pytest.raises(CalledProcessError):
|
|
instance = PrefixedCommandRunner(
|
|
'.', popen=popen_mock, makedirs=makedirs_mock,
|
|
)
|
|
instance.run(['echo'])
|