From 8f71069bc85acb1703328971dd946230ebde2fd6 Mon Sep 17 00:00:00 2001
From: Masayuki Igawa <masayuki.igawa@gmail.com>
Date: Fri, 4 Dec 2015 18:34:41 +0900
Subject: [PATCH] Add unit test for colorizer

This commit adds unit-test cases for colorizer module.

Change-Id: I329a808bbed3240e6fa41ac21d3e5558a0704d6a
---
 os_testr/tests/utils/__init__.py       |  0
 os_testr/tests/utils/test_colorizer.py | 76 ++++++++++++++++++++++++++
 2 files changed, 76 insertions(+)
 create mode 100644 os_testr/tests/utils/__init__.py
 create mode 100644 os_testr/tests/utils/test_colorizer.py

diff --git a/os_testr/tests/utils/__init__.py b/os_testr/tests/utils/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/os_testr/tests/utils/test_colorizer.py b/os_testr/tests/utils/test_colorizer.py
new file mode 100644
index 0000000..fdf96b2
--- /dev/null
+++ b/os_testr/tests/utils/test_colorizer.py
@@ -0,0 +1,76 @@
+# Copyright 2016 Hewlett Packard Enterprise Development LP
+# All Rights Reserved.
+#
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License.
+
+import six
+import sys
+
+from ddt import data
+from ddt import ddt
+from ddt import unpack
+
+from os_testr.tests import base
+from os_testr.utils import colorizer
+
+
+@ddt
+class TestNullColorizer(base.TestCase):
+
+    @data(None, "foo", sys.stdout, )
+    def test_supported_always_true(self, stream):
+        self.assertTrue(colorizer.NullColorizer.supported(stream))
+
+    @data(("foo", "red"), ("foo", "bar"))
+    @unpack
+    def test_write_string_ignore_color(self, text, color):
+        output = six.StringIO()
+        c = colorizer.NullColorizer(output)
+        c.write(text, color)
+        self.assertEqual(text, output.getvalue())
+
+    @data((None, "red"), (None, None))
+    @unpack
+    def test_write_none_exception(self, text, color):
+        c = colorizer.NullColorizer(sys.stdout)
+        self.assertRaises(TypeError, c.write, text, color)
+
+
+@ddt
+class TestAnsiColorizer(base.TestCase):
+
+    def test_supported_false(self):
+        # NOTE(masayukig): This returns False because our unittest env isn't
+        #  interactive
+        self.assertFalse(colorizer.AnsiColorizer.supported(sys.stdout))
+
+    @data(None, "foo")
+    def test_supported_error(self, stream):
+        self.assertRaises(AttributeError,
+                          colorizer.AnsiColorizer.supported, stream)
+
+    @data(("foo", "red", "31"), ("foo", "blue", "34"))
+    @unpack
+    def test_write_string_valid_color(self, text, color, color_code):
+        output = six.StringIO()
+        c = colorizer.AnsiColorizer(output)
+        c.write(text, color)
+        self.assertIn(text, output.getvalue())
+        self.assertIn(color_code, output.getvalue())
+
+    @data(("foo", None), ("foo", "invalid_color"))
+    @unpack
+    def test_write_string_invalid_color(self, text, color):
+        output = six.StringIO()
+        c = colorizer.AnsiColorizer(output)
+        self.assertRaises(KeyError, c.write, text, color)