65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
import pytest
|
|
|
|
from pre_commit.languages import pygrep
|
|
|
|
|
|
@pytest.fixture
|
|
def some_files(tmpdir):
|
|
tmpdir.join('f1').write_binary(b'foo\nbar\n')
|
|
tmpdir.join('f2').write_binary(b'[INFO] hi\n')
|
|
tmpdir.join('f3').write_binary(b"with'quotes\n")
|
|
with tmpdir.as_cwd():
|
|
yield
|
|
|
|
|
|
@pytest.mark.usefixtures('some_files')
|
|
@pytest.mark.parametrize(
|
|
('pattern', 'expected_retcode', 'expected_out'),
|
|
(
|
|
('baz', 0, ''),
|
|
('foo', 1, 'f1:1:foo\n'),
|
|
('bar', 1, 'f1:2:bar\n'),
|
|
(r'(?i)\[info\]', 1, 'f2:1:[INFO] hi\n'),
|
|
("h'q", 1, "f3:1:with'quotes\n"),
|
|
),
|
|
)
|
|
def test_main(some_files, cap_out, pattern, expected_retcode, expected_out):
|
|
ret = pygrep.main((pattern, 'f1', 'f2', 'f3'))
|
|
out = cap_out.get()
|
|
assert ret == expected_retcode
|
|
assert out == expected_out
|
|
|
|
|
|
def test_ignore_case(some_files, cap_out):
|
|
ret = pygrep.main(('--ignore-case', 'info', 'f1', 'f2', 'f3'))
|
|
out = cap_out.get()
|
|
assert ret == 1
|
|
assert out == 'f2:1:[INFO] hi\n'
|
|
|
|
|
|
def test_multiline(some_files, cap_out):
|
|
ret = pygrep.main(('--multiline', r'foo\nbar', 'f1', 'f2', 'f3'))
|
|
out = cap_out.get()
|
|
assert ret == 1
|
|
assert out == 'f1:1:foo\nbar\n'
|
|
|
|
|
|
def test_multiline_line_number(some_files, cap_out):
|
|
ret = pygrep.main(('--multiline', r'ar', 'f1', 'f2', 'f3'))
|
|
out = cap_out.get()
|
|
assert ret == 1
|
|
assert out == 'f1:2:bar\n'
|
|
|
|
|
|
def test_multiline_dotall_flag_is_enabled(some_files, cap_out):
|
|
ret = pygrep.main(('--multiline', r'o.*bar', 'f1', 'f2', 'f3'))
|
|
out = cap_out.get()
|
|
assert ret == 1
|
|
assert out == 'f1:1:foo\nbar\n'
|
|
|
|
|
|
def test_multiline_multiline_flag_is_enabled(some_files, cap_out):
|
|
ret = pygrep.main(('--multiline', r'foo$.*bar', 'f1', 'f2', 'f3'))
|
|
out = cap_out.get()
|
|
assert ret == 1
|
|
assert out == 'f1:1:foo\nbar\n'
|