celery/celery/tests/utilities/test_term.py

# -*- coding: utf-8 -*-
from __future__ import absolute_import

from celery.utils import term
from celery.utils.term import colored, fg

from celery.tests.utils import Case


class test_colored(Case):

    def test_colors(self):
        colors = (
            ("black", term.BLACK),
            ("red", term.RED),
            ("green", term.GREEN),
            ("yellow", term.YELLOW),
            ("blue", term.BLUE),
            ("magenta", term.MAGENTA),
            ("cyan", term.CYAN),
            ("white", term.WHITE),
        )

        for name, key in colors:
            self.assertIn(fg(30 + key), str(colored().names[name]("foo")))

        self.assertTrue(str(colored().bold("f")))
        self.assertTrue(str(colored().underline("f")))
        self.assertTrue(str(colored().blink("f")))
        self.assertTrue(str(colored().reverse("f")))
        self.assertTrue(str(colored().bright("f")))
        self.assertTrue(str(colored().ired("f")))
        self.assertTrue(str(colored().igreen("f")))
        self.assertTrue(str(colored().iyellow("f")))
        self.assertTrue(str(colored().iblue("f")))
        self.assertTrue(str(colored().imagenta("f")))
        self.assertTrue(str(colored().icyan("f")))
        self.assertTrue(str(colored().iwhite("f")))
        self.assertTrue(str(colored().reset("f")))

        self.assertTrue(str(colored().green(u"∂bar")))

        self.assertTrue(
            colored().red(u"éefoo") + colored().green(u"∂bar"))

        self.assertEqual(
            colored().red("foo").no_color(), "foo")

        self.assertTrue(
            repr(colored().blue(u"åfoo")))

        self.assertEqual(repr(colored()), "''")

        c = colored()
        s = c.red("foo", c.blue("bar"), c.green("baz"))
        self.assertTrue(s.no_color())

        c._fold_no_color(s, u"øfoo")
        c._fold_no_color(u"fooå", s)

        c = colored().red(u"åfoo")
        self.assertEqual(c._add(c, u"baræ"),
            u'\x1b[1;31m\xe5foo\x1b[0mbar\xe6')

        c2 = colored().blue(u"ƒƒz")
        c3 = c._add(c, c2)
        self.assertEqual(c3,
            u'\x1b[1;31m\xe5foo\x1b[0m\x1b[1;34m\u0192\u0192z\x1b[0m')