mirror of
https://github.com/pre-commit/pre-commit.git
synced 2026-02-17 08:14:42 +04:00
96 lines
2.8 KiB
Python
96 lines
2.8 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 PrefixedCommandRunner
|
|
|
|
|
|
@pytest.fixture
|
|
def popen_mock():
|
|
popen = mock.Mock()
|
|
popen.return_value.communicate.return_value = (mock.Mock(), mock.Mock())
|
|
return popen
|
|
|
|
|
|
@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):
|
|
instance = PrefixedCommandRunner('prefix', popen=popen_mock)
|
|
ret = instance.run(['{prefix}bar', 'baz'])
|
|
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
|
|
|
|
|
|
@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):
|
|
first = PrefixedCommandRunner('foo', popen=popen_mock)
|
|
second = PrefixedCommandRunner.from_command_runner(first, 'bar')
|
|
second.run(['foo/bar/baz'])
|
|
popen_mock.assert_called_once_with(
|
|
['foo/bar/baz'],
|
|
stdin=subprocess.PIPE,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
)
|