def test_merge_color_configs_mixed():
second = eg_config.get_default_color_config()
first = eg_config.ColorConfig(pound='pound_color',
heading=None,
code='code_color',
backticks=None,
prompt=None,
pound_reset=None,
heading_reset=None,
code_reset=None,
backticks_reset=None,
prompt_reset=None)
merged = eg_config.merge_color_configs(first, second)
target = eg_config.ColorConfig(pound=first.pound,
heading=second.heading,
code=first.code,
backticks=second.backticks,
prompt=second.prompt,
pound_reset=second.pound_reset,
heading_reset=second.heading_reset,
code_reset=second.code_reset,
backticks_reset=second.backticks_reset,
prompt_reset=second.prompt_reset)
assert_equal(merged, target)
def _get_color_config_from_egrc_withdata():
"""Get the color_config that is defined in the egrc_withdata test file."""
test_color_config = eg_config.ColorConfig(
pound='\x1b[32m',
heading='heading_val',
code='code_val',
backticks='backticks_val',
prompt='prompt_val',
pound_reset='pound_reset_val',
heading_reset='heading_reset_val',
code_reset='code_reset_val',
backticks_reset='backticks_reset_val',
prompt_reset='prompt_reset_val')
return test_color_config
def test_merge_color_configs_take_all_first():
second = eg_config.get_default_color_config()
first = eg_config.ColorConfig(pound='pound_color',
heading='heading_color',
code='code_color',
backticks='backticks_color',
prompt='prompt_color',
pound_reset='p_reset',
heading_reset='h_reset',
code_reset='c_reset',
backticks_reset='b_reset',
prompt_reset='prmpt_reset')
merged = eg_config.merge_color_configs(first, second)
assert_equal(merged, first)
def test_merge_color_configs_first_all_none():
second = eg_config.get_default_color_config()
first = eg_config.ColorConfig(pound=None,
heading=None,
code=None,
backticks=None,
prompt=None,
pound_reset=None,
heading_reset=None,
code_reset=None,
backticks_reset=None,
prompt_reset=None)
merged = eg_config.merge_color_configs(first, second)
assert_equal(merged, second)
def test_colorize_heading():
"""Makes sure we colorize things like '# find' correctly"""
color_config = eg_config.ColorConfig('P', 'H', _YELLOW, _MAGENTA, _BLACK,
'RES', 'RES', '', '', '')
clean = get_clean_find_file()
raw_file = get_raw_find_test_file()
target = get_data_with_subs(raw_file,
pound=color_config.pound,
pound_reset=color_config.pound_reset,
heading=color_config.heading,
heading_reset=color_config.heading_reset)
colorizer = eg_colorizer.EgColorizer(color_config)
actual = colorizer.colorize_heading(clean)
assert_equal(actual, target)
def test_colorize_backticks():
"""Makes sure we colorize backticks correctly."""
color_config = eg_config.ColorConfig(_BLACK, _MAGENTA, _YELLOW, 'B',
_GREEN, '', '', '', 'res', '')
clean = get_clean_find_file()
raw_file = get_raw_find_test_file()
target = get_data_with_subs(
raw_file,
backticks=color_config.backticks,
backticks_reset=color_config.backticks_reset,
)
colorizer = eg_colorizer.EgColorizer(color_config)
actual = colorizer.colorize_backticks(clean)
assert_equal(actual, target)
def test_colorize_block_indents():
"""Makes sure we colorize block indents correctly."""
color_config = eg_config.ColorConfig(_BLACK, _MAGENTA, 'C', _YELLOW, 'P',
'', '', 'res', '', 'res')
clean = get_clean_find_file()
raw_file = get_raw_find_test_file()
target = get_data_with_subs(raw_file,
code=color_config.code,
code_reset=color_config.code_reset,
prompt=color_config.prompt,
prompt_reset=color_config.prompt_reset)
colorizer = eg_colorizer.EgColorizer(color_config)
actual = colorizer.colorize_block_indent(clean)
assert_equal(actual, target)