mirror of
https://github.com/pre-commit/pre-commit-hooks.git
synced 2026-04-01 18:56:52 +00:00
130 lines
3.9 KiB
Python
130 lines
3.9 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import unicode_literals
|
|
|
|
import pytest
|
|
|
|
from pre_commit_hooks.trailing_whitespace_fixer import main
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
('input_s', 'expected'),
|
|
(
|
|
('foo \nbar \n', 'foo\nbar\n'),
|
|
('bar\t\nbaz\t\n', 'bar\nbaz\n'),
|
|
),
|
|
)
|
|
def test_fixes_trailing_whitespace(input_s, expected, tmpdir):
|
|
path = tmpdir.join('file.txt')
|
|
path.write(input_s)
|
|
assert main((path.strpath,)) == 1
|
|
assert path.read() == expected
|
|
|
|
|
|
def test_ok_no_newline_end_of_file(tmpdir):
|
|
filename = tmpdir.join('f')
|
|
filename.write_binary(b'foo\nbar')
|
|
ret = main((filename.strpath,))
|
|
assert filename.read_binary() == b'foo\nbar'
|
|
assert ret == 0
|
|
|
|
|
|
def test_ok_with_dos_line_endings(tmpdir):
|
|
filename = tmpdir.join('f')
|
|
filename.write_binary(b'foo\r\nbar\r\nbaz\r\n')
|
|
ret = main((filename.strpath,))
|
|
assert filename.read_binary() == b'foo\r\nbar\r\nbaz\r\n'
|
|
assert ret == 0
|
|
|
|
|
|
def test_markdown_ok(tmpdir):
|
|
filename = tmpdir.join('foo.md')
|
|
filename.write_binary(b'foo \n')
|
|
ret = main((filename.strpath,))
|
|
assert filename.read_binary() == b'foo \n'
|
|
assert ret == 0
|
|
|
|
|
|
# filename, expected input, expected output
|
|
MD_TESTS_1 = (
|
|
('foo.md', 'foo \nbar \n ', 'foo \nbar\n'),
|
|
('bar.Markdown', 'bar \nbaz\t\n\t\n', 'bar \nbaz\n\n'),
|
|
('.md', 'baz \nquux \t\n\t\n', 'baz\nquux\n\n'),
|
|
('txt', 'foo \nbaz \n\t\n', 'foo\nbaz\n\n'),
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(('filename', 'input_s', 'output'), MD_TESTS_1)
|
|
def test_fixes_trailing_markdown_whitespace(filename, input_s, output, tmpdir):
|
|
path = tmpdir.join(filename)
|
|
path.write(input_s)
|
|
ret = main([path.strpath])
|
|
assert ret == 1
|
|
assert path.read() == output
|
|
|
|
|
|
# filename, expected input, expected output
|
|
MD_TESTS_2 = (
|
|
('foo.txt', 'foo \nbar \n \n', 'foo \nbar\n\n'),
|
|
('bar.Markdown', 'bar \nbaz\t\n\t\n', 'bar \nbaz\n\n'),
|
|
('bar.MD', 'bar \nbaz\t \n\t\n', 'bar \nbaz \n\n'),
|
|
('.txt', 'baz \nquux \t\n\t\n', 'baz\nquux\n\n'),
|
|
('txt', 'foo \nbaz \n\t\n', 'foo\nbaz\n\n'),
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(('filename', 'input_s', 'output'), MD_TESTS_2)
|
|
def test_markdown_linebreak_ext_opt(filename, input_s, output, tmpdir):
|
|
path = tmpdir.join(filename)
|
|
path.write(input_s)
|
|
ret = main(('--markdown-linebreak-ext=TxT', path.strpath))
|
|
assert ret == 1
|
|
assert path.read() == output
|
|
|
|
|
|
# filename, expected input, expected output
|
|
MD_TESTS_3 = (
|
|
('foo.baz', 'foo \nbar \n ', 'foo \nbar\n'),
|
|
('bar', 'bar \nbaz\t\n\t\n', 'bar \nbaz\n\n'),
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(('filename', 'input_s', 'output'), MD_TESTS_3)
|
|
def test_markdown_linebreak_ext_opt_all(filename, input_s, output, tmpdir):
|
|
path = tmpdir.join(filename)
|
|
path.write(input_s)
|
|
# need to make sure filename is not treated as argument to option
|
|
ret = main(('--markdown-linebreak-ext=*', path.strpath))
|
|
assert ret == 1
|
|
assert path.read() == output
|
|
|
|
|
|
@pytest.mark.parametrize(('arg'), ('--', 'a.b', 'a/b'))
|
|
def test_markdown_linebreak_ext_badopt(arg):
|
|
with pytest.raises(SystemExit) as excinfo:
|
|
main(['--markdown-linebreak-ext', arg])
|
|
assert excinfo.value.code == 2
|
|
|
|
|
|
# filename, expected input, expected output
|
|
MD_TESTS_4 = (
|
|
('bar.md', 'bar \nbaz\t \n\t\n', 'bar\nbaz\n\n'),
|
|
('bar.markdown', 'baz \nquux \n', 'baz\nquux\n'),
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(('filename', 'input_s', 'output'), MD_TESTS_4)
|
|
def test_no_markdown_linebreak_ext_opt(filename, input_s, output, tmpdir):
|
|
path = tmpdir.join(filename)
|
|
path.write(input_s)
|
|
ret = main(['--no-markdown-linebreak-ext', path.strpath])
|
|
assert ret == 1
|
|
assert path.read() == output
|
|
|
|
|
|
def test_preserve_non_utf8_file(tmpdir):
|
|
non_utf8_bytes_content = b'<a>\xe9 \n</a>\n'
|
|
path = tmpdir.join('file.txt')
|
|
path.write_binary(non_utf8_bytes_content)
|
|
ret = main([path.strpath])
|
|
assert ret == 1
|
|
assert path.size() == (len(non_utf8_bytes_content) - 1)
|