mirror of
https://github.com/pre-commit/pre-commit-hooks.git
synced 2026-04-05 03:26:53 +00:00
336f2ed6d1
Instead of writing \n we try to detect the line ending used in the file. We use the first line ending sequence we find.
47 lines
1.3 KiB
Python
47 lines
1.3 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\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'),
|
|
(b'foo\r\nbar', 1, b'foo\r\nbar\r\n'),
|
|
(b'foo\nbar', 1, b'foo\nbar\n'),
|
|
(b'foo\rbar', 1, b'foo\rbar\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
|