def test_all_colors():
"""Makes sure the COLOR_NAMES is in sync with the colors in the printer"""
cp = Printer()
for color_name in COLOR_NAMES:
out = StringIO()
cp.msg(_u('msg'), color_name, file=out)
out.seek(0)
assert out.read() == _u(cp.colors[color_name] + 'msg' + '\033[0m')
def test_all_colors():
"""Makes sure the COLOR_NAMES is in sync with the colors in the printer"""
cp = Printer()
for color_name in COLOR_NAMES:
out = StringIO()
cp.msg(_u('msg'), color_name, file=out)
out.seek(0)
assert out.read() == _u(cp.colors[color_name] + 'msg' + '\033[0m')
def test_no_color():
cp = Printer(use_color=False)
out = StringIO()
cp.msg(_u('msg'), 'red', file=out)
out.seek(0)
assert out.read() == _u('msg')
def test_conky_red_msg():
cp = Printer(conky=True)
out = StringIO()
cp.msg(_u('msg'), 'red', file=out)
out.seek(0)
assert out.read() == _u('${color red}msg')
def test_red_msg():
cp = Printer()
out = StringIO()
cp.msg(_u('msg'), 'red', file=out)
out.seek(0)
assert out.read() == _u('\033[0;31mmsg\033[0m')
def test_no_color():
cp = Printer(use_color=False)
out = StringIO()
cp.msg(_u('msg'), 'red', file=out)
out.seek(0)
assert out.read() == _u('msg')
def test_conky_red_msg():
cp = Printer(conky=True)
out = StringIO()
cp.msg(_u('msg'), 'red', file=out)
out.seek(0)
assert out.read() == _u('${color red}msg${color}')
def test_red_msg():
cp = Printer()
out = StringIO()
cp.msg(_u('msg'), 'red', file=out)
out.seek(0)
assert out.read() == _u('\033[0;31mmsg\033[0m')