third-party/pre-commit
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit.git synced 2026-02-17 08:14:42 +04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Reorganize output writing

Browse source
This commit is contained in:
Anthony Sottile 2016-11-26 12:15:55 -08:00
parent 1adfa24124
commit 0dda19f691
20 changed files with 202 additions and 142 deletions
Download patch file Download diff file Expand all files Collapse all files

19
tests/output_test.py
View file

@ -4,8 +4,7 @@ import mock
import pytest
from pre_commit import color
from pre_commit.output import get_hook_message
from pre_commit.output import sys_stdout_write_wrapper
from pre_commit import output
@pytest.mark.parametrize(
@ -25,16 +24,16 @@ from pre_commit.output import sys_stdout_write_wrapper
)
def test_get_hook_message_raises(kwargs):
with pytest.raises(ValueError):
get_hook_message('start', **kwargs)
output.get_hook_message('start', **kwargs)
def test_case_with_end_len():
ret = get_hook_message('start', end_len=5, cols=15)
ret = output.get_hook_message('start', end_len=5, cols=15)
assert ret == 'start' + '.' * 4
def test_case_with_end_msg():
ret = get_hook_message(
ret = output.get_hook_message(
'start',
end_msg='end',
end_color='',
@ -45,7 +44,7 @@ def test_case_with_end_msg():
def test_case_with_end_msg_using_color():
ret = get_hook_message(
ret = output.get_hook_message(
'start',
end_msg='end',
end_color=color.RED,
@ -56,7 +55,7 @@ def test_case_with_end_msg_using_color():
def test_case_with_postfix_message():
ret = get_hook_message(
ret = output.get_hook_message(
'start',
postfix='post ',
end_msg='end',
@ -68,7 +67,7 @@ def test_case_with_postfix_message():
def test_make_sure_postfix_is_not_colored():
ret = get_hook_message(
ret = output.get_hook_message(
'start',
postfix='post ',
end_msg='end',
@ -81,7 +80,7 @@ def test_make_sure_postfix_is_not_colored():
)
def test_sys_stdout_write_wrapper_writes():
def test_output_write_writes():
fake_stream = mock.Mock()
sys_stdout_write_wrapper('hello world', fake_stream)
output.write('hello world', fake_stream)
assert fake_stream.write.call_count == 1