pre-commit-hooks/tests/mixed_line_ending_test.py
2017-07-09 09:24:21 +02:00

107 lines
3.4 KiB
Python

import pytest
from pre_commit_hooks.mixed_line_ending import mixed_line_ending
# Input, expected return value, expected output
TESTS_FIX_AUTO = (
# only 'LF'
(b'foo\nbar\nbaz\n', 0, b'foo\nbar\nbaz\n'),
# only 'CRLF'
(b'foo\r\nbar\r\nbaz\r\n', 0, b'foo\r\nbar\r\nbaz\r\n'),
# mixed with majority of 'LF'
(b'foo\r\nbar\nbaz\n', 1, b'foo\nbar\nbaz\n'),
# mixed with majority of 'CRLF'
(b'foo\r\nbar\nbaz\r\n', 1, b'foo\r\nbar\r\nbaz\r\n'),
# mixed with as much 'LF' as 'CRLF'
(b'foo\r\nbar\nbaz', 1, b'foo\r\nbar\nbaz'),
)
@pytest.mark.parametrize(('input_s', 'expected_retval', 'output'),
TESTS_FIX_AUTO)
def test_mixed_line_ending_fix_auto(input_s, expected_retval, output, tmpdir):
path = tmpdir.join('file.txt')
path.write(input_s)
ret = mixed_line_ending(('--fix=auto', '-vv', path.strpath))
assert ret == expected_retval
assert path.read_binary() == output
# Input, expected return value, expected output
TESTS_NO_FIX = (
# only 'LF'
(b'foo\nbar\nbaz\n', 0, b'foo\nbar\nbaz\n'),
# only 'CRLF'
(b'foo\r\nbar\r\nbaz\r\n', 0, b'foo\r\nbar\r\nbaz\r\n'),
# mixed with majority of 'LF'
(b'foo\r\nbar\nbaz\n', 1, b'foo\r\nbar\nbaz\n'),
# mixed with majority of 'CRLF'
(b'foo\r\nbar\nbaz\r\n', 1, b'foo\r\nbar\nbaz\r\n'),
)
@pytest.mark.parametrize(('input_s', 'expected_retval', 'output'),
TESTS_NO_FIX)
def test_detect_mixed_line_ending(input_s, expected_retval, output, tmpdir):
path = tmpdir.join('file.txt')
path.write(input_s)
ret = mixed_line_ending(('--fix=no', '-vv', path.strpath))
assert ret == expected_retval
assert path.read_binary() == output
# Input, expected return value, expected output
TESTS_FIX_FORCE_LF = (
# only 'LF'
(b'foo\nbar\nbaz\n', 1, b'foo\nbar\nbaz\n'),
# only 'CRLF'
(b'foo\r\nbar\r\nbaz\r\n', 1, b'foo\nbar\nbaz\n'),
# mixed with majority of 'LF'
(b'foo\r\nbar\nbaz\n', 1, b'foo\nbar\nbaz\n'),
# mixed with majority of 'CRLF'
(b'foo\r\nbar\nbaz\r\n', 1, b'foo\nbar\nbaz\n'),
)
@pytest.mark.parametrize(('input_s', 'expected_retval', 'output'),
TESTS_FIX_FORCE_LF)
def test_mixed_line_ending_fix_force_lf(input_s, expected_retval, output,
tmpdir):
path = tmpdir.join('file.txt')
path.write(input_s)
ret = mixed_line_ending(('--fix=lf', '-vv', path.strpath))
assert ret == expected_retval
assert path.read_binary() == output
# Input, expected return value, expected output
TESTS_FIX_FORCE_CRLF = (
# only 'LF'
(b'foo\nbar\nbaz\n', 1, b'foo\r\nbar\r\nbaz\r\n'),
# only 'CRLF'
(b'foo\r\nbar\r\nbaz\r\n', 1, b'foo\r\nbar\r\nbaz\r\n'),
# mixed with majority of 'LF'
(b'foo\r\nbar\nbaz\n', 1, b'foo\r\nbar\r\nbaz\r\n'),
# mixed with majority of 'CRLF'
(b'foo\r\nbar\nbaz\r\n', 1, b'foo\r\nbar\r\nbaz\r\n'),
)
@pytest.mark.parametrize(('input_s', 'expected_retval', 'output'),
TESTS_FIX_FORCE_CRLF)
def test_mixed_line_ending_fix_force_crlf(input_s, expected_retval, output,
tmpdir):
path = tmpdir.join('file.txt')
path.write(input_s)
ret = mixed_line_ending(('--fix=crlf', '-vv', path.strpath))
assert ret == expected_retval
assert path.read_binary() == output
def test_check_filenames():
with pytest.raises(IOError):
mixed_line_ending(['/dev/null'])