def test_merge_color_configs_mixed(): second = config.get_default_color_config() first = 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 = config.merge_color_configs(first, second) target = 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 = 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 = config.get_default_color_config() first = 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 = config.merge_color_configs(first, second) assert_equal(merged, first)
def test_merge_color_configs_first_all_none(): second = config.get_default_color_config() first = 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 = config.merge_color_configs(first, second) assert_equal(merged, second)
def test_colorize_heading(): """Makes sure we colorize things like '# find' correctly""" color_config = 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 = color.EgColorizer(color_config) actual = colorizer.colorize_heading(clean) assert_equal(actual, target)
def test_colorize_backticks(): """Makes sure we colorize backticks correctly.""" color_config = 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 = color.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 = 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 = color.EgColorizer(color_config) actual = colorizer.colorize_block_indent(clean) assert_equal(actual, target)