mirror of
https://github.com/pre-commit/pre-commit-hooks.git
synced 2026-04-05 19:46:54 +00:00
a17514d55e
This change adds an advisory mode via `--check` that only warns of formatting issues with files, but does not address them. This support is desirable because--while I don't mind the automagic changes when done in a mechanical way--some individuals who I described the current behavior of these fixers to were a bit uneasy about the magic that went along with them. Adding `--check` so others can opt out (similar to `black --check`) is a compromise on this front. Signed-off-by: Enji Cooper <yaneurabeya@gmail.com>
54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
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'),
|
|
)
|
|
|
|
|
|
@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([path.strpath])
|
|
file_output = path.read_binary()
|
|
|
|
assert file_output == output
|
|
assert ret == expected_retval
|
|
|
|
|
|
@pytest.mark.parametrize(('input_s', 'expected_retval', 'output'), TESTS)
|
|
def test_integration_check(input_s, expected_retval, output, tmpdir):
|
|
path = tmpdir.join('file.txt')
|
|
path.write_binary(input_s)
|
|
|
|
ret = main(['--check', path.strpath])
|
|
file_output = path.read_binary()
|
|
|
|
assert file_output == input_s
|
|
assert ret == expected_retval
|