iframe-proxy | Sunbelt Computer Software

History

87 lines (71 loc) · 2.09 KB

Raw

from __future__ import unicode_literals

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

@pytest.mark.parametrize(

'kwargs',

(

# both end_msg and end_len

{'end_msg': 'end', 'end_len': 1, 'end_color': '', 'use_color': True},

# Neither end_msg nor end_len

{},

# Neither color option for end_msg

{'end_msg': 'end'},

# No use_color for end_msg

{'end_msg': 'end', 'end_color': ''},

# No end_color for end_msg

{'end_msg': 'end', 'use_color': ''},

)

def test_get_hook_message_raises(kwargs):

with pytest.raises(ValueError):

get_hook_message('start', **kwargs)

def test_case_with_end_len():

ret = get_hook_message('start', end_len=5, cols=15)

assert ret == 'start' + '.' * 4

def test_case_with_end_msg():

ret = get_hook_message(

'start',

end_msg='end',

end_color='',

use_color=False,

cols=15,

)

assert ret == 'start' + '.' * 6 + 'end' + '\n'

def test_case_with_end_msg_using_color():

ret = get_hook_message(

'start',

end_msg='end',

end_color=color.RED,

use_color=True,

cols=15,

)

assert ret == 'start' + '.' * 6 + color.RED + 'end' + color.NORMAL + '\n'

def test_case_with_postfix_message():

ret = get_hook_message(

'start',

postfix='post ',

end_msg='end',

end_color='',

use_color=False,

cols=20,

)

assert ret == 'start' + '.' * 6 + 'post ' + 'end' + '\n'

def test_make_sure_postfix_is_not_colored():

ret = get_hook_message(

'start',

postfix='post ',

end_msg='end',

end_color=color.RED,

use_color=True,

cols=20,

)

assert ret == (

'start' + '.' * 6 + 'post ' + color.RED + 'end' + color.NORMAL + '\n'

)

def test_sys_stdout_write_wrapper_writes():

fake_stream = mock.Mock()

sys_stdout_write_wrapper('hello world', fake_stream)

assert fake_stream.write.call_count == 1

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

output_test.py

Latest commit

History

output_test.py

File metadata and controls