mirror of
https://github.com/pre-commit/pre-commit.git
synced 2026-02-17 08:14:42 +04:00
fa4c03da65
Change how xargs.partition computes the command length (including arguments) depending on the plataform. More specifically, 'win32' uses the amount of characters while posix system uses the byte count.
121 lines
3 KiB
Python
121 lines
3 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import unicode_literals
|
|
|
|
from unittest import mock
|
|
|
|
import pytest
|
|
|
|
from pre_commit import xargs
|
|
|
|
|
|
@pytest.fixture
|
|
def sys_win32_mock():
|
|
return mock.Mock(
|
|
platform='win32',
|
|
getdefaultencoding=mock.Mock(return_value='utf-8'),
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def sys_linux_mock():
|
|
return mock.Mock(
|
|
platform='linux',
|
|
getdefaultencoding=mock.Mock(return_value='utf-8'),
|
|
)
|
|
|
|
|
|
def test_partition_trivial():
|
|
assert xargs.partition(('cmd',), ()) == (('cmd',),)
|
|
|
|
|
|
def test_partition_simple():
|
|
assert xargs.partition(('cmd',), ('foo',)) == (('cmd', 'foo'),)
|
|
|
|
|
|
def test_partition_limits():
|
|
ret = xargs.partition(
|
|
('ninechars',), (
|
|
# Just match the end (with spaces)
|
|
'.' * 5, '.' * 4,
|
|
# Just match the end (single arg)
|
|
'.' * 10,
|
|
# Goes over the end
|
|
'.' * 5,
|
|
'.' * 6,
|
|
),
|
|
_max_length=20,
|
|
)
|
|
assert ret == (
|
|
('ninechars', '.' * 5, '.' * 4),
|
|
('ninechars', '.' * 10),
|
|
('ninechars', '.' * 5),
|
|
('ninechars', '.' * 6),
|
|
)
|
|
|
|
|
|
def test_partition_limit_win32(sys_win32_mock):
|
|
cmd = ('ninechars',)
|
|
varargs = ('😑' * 10,)
|
|
with mock.patch('pre_commit.xargs.sys', sys_win32_mock):
|
|
ret = xargs.partition(cmd, varargs, _max_length=20)
|
|
|
|
assert ret == (cmd + varargs,)
|
|
|
|
|
|
def test_partition_limit_linux(sys_linux_mock):
|
|
cmd = ('ninechars',)
|
|
varargs = ('😑' * 5,)
|
|
with mock.patch('pre_commit.xargs.sys', sys_linux_mock):
|
|
ret = xargs.partition(cmd, varargs, _max_length=30)
|
|
|
|
assert ret == (cmd + varargs,)
|
|
|
|
|
|
def test_argument_too_long_with_large_unicode(sys_linux_mock):
|
|
cmd = ('ninechars',)
|
|
varargs = ('😑' * 10,) # 4 bytes * 10
|
|
with mock.patch('pre_commit.xargs.sys', sys_linux_mock):
|
|
with pytest.raises(xargs.ArgumentTooLongError):
|
|
xargs.partition(cmd, varargs, _max_length=20)
|
|
|
|
|
|
def test_argument_too_long():
|
|
with pytest.raises(xargs.ArgumentTooLongError):
|
|
xargs.partition(('a' * 5,), ('a' * 5,), _max_length=10)
|
|
|
|
|
|
def test_xargs_smoke():
|
|
ret, out, err = xargs.xargs(('echo',), ('hello', 'world'))
|
|
assert ret == 0
|
|
assert out == b'hello world\n'
|
|
assert err == b''
|
|
|
|
|
|
exit_cmd = ('bash', '-c', 'exit $1', '--')
|
|
# Abuse max_length to control the exit code
|
|
max_length = len(' '.join(exit_cmd)) + 2
|
|
|
|
|
|
def test_xargs_negate():
|
|
ret, _, _ = xargs.xargs(
|
|
exit_cmd, ('1',), negate=True, _max_length=max_length,
|
|
)
|
|
assert ret == 0
|
|
|
|
ret, _, _ = xargs.xargs(
|
|
exit_cmd, ('1', '0'), negate=True, _max_length=max_length,
|
|
)
|
|
assert ret == 1
|
|
|
|
|
|
def test_xargs_negate_command_not_found():
|
|
ret, _, _ = xargs.xargs(('cmd-not-found',), ('1',), negate=True)
|
|
assert ret != 0
|
|
|
|
|
|
def test_xargs_retcode_normal():
|
|
ret, _, _ = xargs.xargs(exit_cmd, ('0',), _max_length=max_length)
|
|
assert ret == 0
|
|
|
|
ret, _, _ = xargs.xargs(exit_cmd, ('0', '1'), _max_length=max_length)
|
|
assert ret == 1
|