salt/tests/unit/utils/test_color.py

# -*- coding: utf-8 -*-
"""
Unit tests for salt.utils.color.py
"""

# Import python libs
from __future__ import absolute_import, print_function, unicode_literals

# Import Salt libs
import salt.utils.color
from salt.ext import six

# Import Salt Testing libs
from tests.support.unit import TestCase


class ColorUtilsTestCase(TestCase):
    def test_get_colors(self):
        ret = salt.utils.color.get_colors()
        self.assertEqual("\x1b[0;37m", six.text_type(ret["LIGHT_GRAY"]))

        ret = salt.utils.color.get_colors(use=False)
        self.assertDictContainsSubset({"LIGHT_GRAY": ""}, ret)

        ret = salt.utils.color.get_colors(use="LIGHT_GRAY")
        # LIGHT_YELLOW now == LIGHT_GRAY
        self.assertEqual(
            six.text_type(ret["LIGHT_YELLOW"]), six.text_type(ret["LIGHT_GRAY"])
        )