mirror of
https://github.com/pre-commit/pre-commit.git
synced 2025-12-20 16:39:36 -06:00
145 lines
4.2 KiB
Python
145 lines
4.2 KiB
Python
|
|
import os
|
|
import mock
|
|
import pytest
|
|
import subprocess
|
|
|
|
from pre_commit.prefixed_command_runner import _replace_cmd
|
|
from pre_commit.prefixed_command_runner import CalledProcessError
|
|
from pre_commit.prefixed_command_runner import PrefixedCommandRunner
|
|
|
|
|
|
def test_CalledProcessError_str():
|
|
error = CalledProcessError(1, ['git', 'status'], 0, ('stdout', 'stderr'))
|
|
assert str(error) == (
|
|
"Command: ['git', 'status']\n"
|
|
"Return code: 1\n"
|
|
"Expected return code: 0\n"
|
|
"Output: ('stdout', 'stderr')\n"
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def popen_mock():
|
|
popen = mock.Mock(spec=subprocess.Popen)
|
|
popen.return_value.communicate.return_value = (mock.Mock(), mock.Mock())
|
|
return popen
|
|
|
|
|
|
@pytest.fixture
|
|
def makedirs_mock():
|
|
return mock.Mock(spec=os.makedirs)
|
|
|
|
|
|
@pytest.mark.parametrize(('input', 'kwargs', 'expected_output'), (
|
|
([], {}, []),
|
|
(['foo'], {}, ['foo']),
|
|
([], {'foo': 'bar'}, []),
|
|
(['{foo}/baz'], {'foo': 'bar'}, ['bar/baz']),
|
|
(['foo'], {'foo': 'bar'}, ['foo']),
|
|
(['foo', '{bar}'], {'bar': 'baz'}, ['foo', 'baz']),
|
|
))
|
|
def test_replace_cmd(input, kwargs, expected_output):
|
|
ret = _replace_cmd(input, **kwargs)
|
|
assert ret == expected_output
|
|
|
|
|
|
@pytest.mark.parametrize(('input', 'expected_prefix'), (
|
|
('.', './'),
|
|
('foo', 'foo/'),
|
|
('bar/', 'bar/'),
|
|
('foo/bar', 'foo/bar/'),
|
|
('foo/bar/', 'foo/bar/'),
|
|
))
|
|
def test_init_normalizes_path_endings(input, expected_prefix):
|
|
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(
|
|
['prefix/bar', 'baz'],
|
|
stdin=subprocess.PIPE,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
)
|
|
assert ret == (
|
|
popen_mock.return_value.returncode,
|
|
popen_mock.return_value.communicate.return_value[0],
|
|
popen_mock.return_value.communicate.return_value[1],
|
|
)
|
|
|
|
|
|
PATH_TESTS = (
|
|
('foo', '', 'foo'),
|
|
('foo', 'bar', 'foo/bar'),
|
|
('foo/bar', '../baz', 'foo/baz'),
|
|
('./', 'bar', '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 == 'foo/bar/baz'
|
|
|
|
|
|
@pytest.mark.parametrize(('prefix', 'path_end', 'expected_output'),
|
|
tuple(
|
|
(prefix, path_end, expected_output + os.sep)
|
|
for prefix, path_end, expected_output in PATH_TESTS
|
|
),
|
|
)
|
|
def test_from_command_runner(prefix, path_end, expected_output):
|
|
first = PrefixedCommandRunner(prefix)
|
|
second = PrefixedCommandRunner.from_command_runner(first, path_end)
|
|
assert second.prefix_dir == expected_output
|
|
|
|
|
|
def test_from_command_runner_preserves_popen(popen_mock, makedirs_mock):
|
|
first = PrefixedCommandRunner('foo', popen=popen_mock, makedirs=makedirs_mock)
|
|
second = PrefixedCommandRunner.from_command_runner(first, 'bar')
|
|
second.run(['foo/bar/baz'], retcode=None)
|
|
popen_mock.assert_called_once_with(
|
|
['foo/bar/baz'],
|
|
stdin=subprocess.PIPE,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
)
|
|
makedirs_mock.assert_called_once_with('foo/bar/')
|
|
|
|
|
|
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(['foo'])
|