| from __future__ import absolute_import |
| from __future__ import unicode_literals |
| |
| import sys |
| |
| import pytest |
| |
| from pre_commit_hooks.trailing_whitespace_fixer import fix_trailing_whitespace |
| from testing.util import cwd |
| |
| |
| def test_fixes_trailing_whitespace(tmpdir): |
| with cwd(tmpdir.strpath): |
| for filename, contents in ( |
| ('foo.py', 'foo \nbar \n'), |
| ('bar.py', 'bar\t\nbaz\t\n'), |
| ): |
| with open(filename, 'w') as file_obj: |
| file_obj.write(contents) # pragma: no branch (26 coverage bug) |
| |
| ret = fix_trailing_whitespace(['foo.py', 'bar.py']) |
| assert ret == 1 |
| |
| for filename, after_contents in ( |
| ('foo.py', 'foo\nbar\n'), |
| ('bar.py', 'bar\nbaz\n'), |
| ): |
| assert open(filename).read() == after_contents |
| |
| |
| # filename, expected input, expected output |
| # pylint: disable=bad-whitespace |
| MD_TESTS_1 = ( |
| ('foo.md', 'foo \nbar \n ', 'foo \nbar\n\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'), |
| ) |
| # pylint: enable=bad-whitespace |
| |
| |
| @pytest.mark.parametrize(('filename', 'input_s', 'output'), MD_TESTS_1) |
| def test_fixes_trailing_markdown_whitespace(filename, input_s, output, tmpdir): |
| with cwd(tmpdir.strpath): |
| with open(filename, 'w') as file_obj: |
| file_obj.write(input_s) # pragma: no branch (26 coverage bug) |
| |
| ret = fix_trailing_whitespace([filename]) |
| assert ret == 1 |
| assert open(filename).read() == output |
| |
| |
| # filename, expected input, expected output |
| # pylint: disable=bad-whitespace |
| 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'), |
| ) |
| # pylint: enable=bad-whitespace |
| |
| |
| @pytest.mark.parametrize(('filename', 'input_s', 'output'), MD_TESTS_2) |
| def test_markdown_linebreak_ext_opt(filename, input_s, output, tmpdir): |
| with cwd(tmpdir.strpath): |
| with open(filename, 'w') as file_obj: |
| file_obj.write(input_s) # pragma: no branch (26 coverage bug) |
| |
| ret = fix_trailing_whitespace(['--markdown-linebreak-ext=TxT', |
| filename]) |
| assert ret == 1 |
| assert open(filename).read() == output |
| |
| |
| # filename, expected input, expected output |
| # pylint: disable=bad-whitespace |
| MD_TESTS_3 = ( |
| ('foo.baz', 'foo \nbar \n ', 'foo \nbar\n\n'), |
| ('bar', 'bar \nbaz\t\n\t\n', 'bar \nbaz\n\n'), |
| ) |
| # pylint: enable=bad-whitespace |
| |
| |
| @pytest.mark.parametrize(('filename', 'input_s', 'output'), MD_TESTS_3) |
| def test_markdown_linebreak_ext_opt_all(filename, input_s, output, tmpdir): |
| with cwd(tmpdir.strpath): |
| with open(filename, 'w') as file_obj: |
| file_obj.write(input_s) # pragma: no branch (26 coverage bug) |
| |
| # need to make sure filename is not treated as argument to option |
| ret = fix_trailing_whitespace(['--markdown-linebreak-ext=*', |
| filename]) |
| assert ret == 1 |
| assert open(filename).read() == output |
| |
| |
| @pytest.mark.parametrize(('arg'), ('--', 'a.b', 'a/b')) |
| def test_markdown_linebreak_ext_badopt(arg): |
| try: |
| ret = fix_trailing_whitespace(['--markdown-linebreak-ext', arg]) |
| except SystemExit: |
| ret = sys.exc_info()[1].code |
| finally: |
| assert ret == 2 |
| |
| |
| # filename, expected input, expected output |
| # pylint: disable=bad-whitespace |
| MD_TESTS_4 = ( |
| ('bar.md', 'bar \nbaz\t \n\t\n', 'bar\nbaz\n\n'), |
| ('bar.markdown', 'baz \nquux \n', 'baz\nquux\n'), |
| ) |
| # pylint: enable=bad-whitespace |
| |
| |
| @pytest.mark.parametrize(('filename', 'input_s', 'output'), MD_TESTS_4) |
| def test_no_markdown_linebreak_ext_opt(filename, input_s, output, tmpdir): |
| with cwd(tmpdir.strpath): |
| with open(filename, 'w') as file_obj: |
| file_obj.write(input_s) # pragma: no branch (26 coverage bug) |
| |
| ret = fix_trailing_whitespace(['--no-markdown-linebreak-ext', filename]) |
| assert ret == 1 |
| assert open(filename).read() == output |
| |
| |
| def test_returns_zero_for_no_changes(): |
| assert fix_trailing_whitespace([__file__]) == 0 |