mirror of
https://github.com/pre-commit/pre-commit-hooks.git
synced 2026-04-05 03:26:53 +00:00
203735ecf7
Added case for which the file has mixed line endings. In this case, default into using LF for end of file line.
46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
import io
|
|
|
|
import pytest
|
|
|
|
from pre_commit_hooks.end_of_file_fixer import fix_file
|
|
from pre_commit_hooks.end_of_file_fixer import main
|
|
|
|
|
|
# Input, expected return value, expected output
|
|
TESTS = (
|
|
(b'foo\n', 0, b'foo\n'),
|
|
(b'', 0, b''),
|
|
(b'\n\n', 1, b''),
|
|
(b'\n\n\n\n', 1, b''),
|
|
(b'foo', 1, b'foo\n'),
|
|
(b'foo\n\n\n', 1, b'foo\n'),
|
|
(b'\xe2\x98\x83', 1, b'\xe2\x98\x83\n'),
|
|
(b'foo\r\n', 0, b'foo\r\n'),
|
|
(b'foo\r\nbar', 1, b'foo\r\nbar\r\n'),
|
|
(b'foo\nbar\r\nbaz', 1, b'foo\nbar\r\nbaz\n'),
|
|
(b'foo\r\n\r\n\r\n', 1, b'foo\r\n'),
|
|
(b'foo\r', 0, b'foo\r'),
|
|
(b'foo\r\r\r\r', 1, b'foo\r'),
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(('input_s', 'expected_retval', 'output'), TESTS)
|
|
def test_fix_file(input_s, expected_retval, output):
|
|
file_obj = io.BytesIO(input_s)
|
|
ret = fix_file(file_obj)
|
|
assert file_obj.getvalue() == output
|
|
assert ret == expected_retval
|
|
|
|
|
|
@pytest.mark.parametrize(('input_s', 'expected_retval', 'output'), TESTS)
|
|
def test_integration(input_s, expected_retval, output, tmpdir):
|
|
path = tmpdir.join('file.txt')
|
|
path.write_binary(input_s)
|
|
|
|
ret = main([str(path)])
|
|
file_output = path.read_binary()
|
|
|
|
assert file_output == output
|
|
assert ret == expected_retval
|