def test_format_matches(self):
"""Test format includes expected output."""
base_str = 'blue'
blue_str = colors.ColorStr(base_str, colors.COLOR_MAP['b'])
expected = 'INFO: this is {0}\n'.format(blue_str)
self.log.info('this is #b<{0}>'.format(base_str))
self.assertEqual(self.stream.getvalue(), expected)
def format(self, record):
"""Format the message."""
message = _cast_message(record.msg)
for val in COLOR_EXP.findall(message):
message = message.replace(
val, colors.ColorStr(val[3:-1], colors.COLOR_MAP[val[1]]))
record.msg = message
return logging.Formatter.format(self, record)
def test_make_str_white_unsupported(self):
"""Verify no color sequences on white"""
self.assertEqual(
colors.ColorStr('colored', colors.COLOR_MAP['w'], force_seq=False),
'colored')
def test_make_str_magenta(self):
"""Verify can color strings magenta"""
self.assertEqual(
colors.ColorStr('colored', colors.COLOR_MAP['m'], force_seq=True),
'\033[95mcolored\033[0m')
def test_make_str_white(self):
"""Verify can color strings white"""
self.assertEqual(
colors.ColorStr('colored', colors.COLOR_MAP['w'], force_seq=True),
'\033[97mcolored\033[0m')
def test_make_str_cyan(self):
"""Verify can color strings cyan"""
self.assertEqual(
colors.ColorStr('colored', colors.COLOR_MAP['c'], force_seq=True),
'\033[96mcolored\033[0m')
def test_make_str_red(self):
"""Verify can color strings red"""
self.assertEqual(
colors.ColorStr('colored', colors.COLOR_MAP['r'], force_seq=True),
'\033[91mcolored\033[0m')
def test_make_str_yellow(self):
"""Verify can color strings yellow"""
self.assertEqual(
colors.ColorStr('colored', colors.COLOR_MAP['y'], force_seq=True),
"\033[93mcolored\033[0m")
