mirror of
https://github.com/pre-commit/pre-commit.git
synced 2026-01-13 20:40:08 -06:00
121 lines
3.1 KiB
Python
121 lines
3.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import absolute_import
|
|
from __future__ import unicode_literals
|
|
|
|
import pytest
|
|
|
|
from pre_commit import xargs
|
|
|
|
|
|
@pytest.fixture
|
|
def sys_win32_mock(mocker):
|
|
return mocker.Mock(
|
|
platform='win32',
|
|
getdefaultencoding=mocker.Mock(return_value='utf-8'),
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def sys_linux_mock(mocker):
|
|
return mocker.Mock(
|
|
platform='linux',
|
|
getdefaultencoding=mocker.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(mocker, sys_win32_mock):
|
|
cmd = ('ninechars',)
|
|
varargs = ('😑' * 10,)
|
|
with mocker.mock_module.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(mocker, sys_linux_mock):
|
|
cmd = ('ninechars',)
|
|
varargs = ('😑' * 5,)
|
|
with mocker.mock_module.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(mocker, sys_linux_mock):
|
|
cmd = ('ninechars',)
|
|
varargs = ('😑' * 10,) # 4 bytes * 10
|
|
with mocker.mock_module.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
|