1 files changed, 133 insertions, 0 deletions
diff --git a/pylog/tests/test_color_formatter.py b/pylog/tests/test_color_formatter.py
new file mode 100644
index 0000000..56bf9c2
--- /dev/null
+++ b/pylog/tests/test_color_formatter.py
@@ -0,0 +1,133 @@
+# Copyright (c) 2020 Deutsche Telekom.
+# 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.
+
+import unittest
+import sys
+from logging import LogRecord
+
+if sys.version_info[0] < 3:
+    from mock import patch
+if sys.version_info[0] >= 3:
+    from unittest.mock import patch
+import pytest
+
+from onaplogging.colorFormatter import (
+    ATTRIBUTES,
+    BaseColorFormatter,
+    colored,
+    COLORS,
+    HIGHLIGHTS,
+    FMT_STR,
+    RESET,
+)
+from onaplogging.utils.system import is_above_python_3_2
+
+
+class TestColorFormatter(unittest.TestCase):
+
+    TEST_TEXT = "test"
+
+    def test_colored_os_name_nt(self):
+
+        with patch("onaplogging.colorFormatter.os.name", "nt"):
+
+            text = colored(self.TEST_TEXT)
+            assert text == self.TEST_TEXT
+
+            text = colored(self.TEST_TEXT, color="black")
+            assert text == self.TEST_TEXT
+
+            text = colored(self.TEST_TEXT, on_color="black")
+            assert text == self.TEST_TEXT
+
+            text = colored(self.TEST_TEXT, attrs="bold")
+            assert text == self.TEST_TEXT
+
+    def test_colored_os_name_ce(self):
+
+        with patch("onaplogging.colorFormatter.os.name", "ce"):
+
+            text = colored(self.TEST_TEXT)
+            assert text == self.TEST_TEXT
+
+            text = colored(self.TEST_TEXT, color="black")
+            assert text == self.TEST_TEXT
+
+            text = colored(self.TEST_TEXT, on_color="black")
+            assert text == self.TEST_TEXT
+
+            text = colored(self.TEST_TEXT, attrs="bold")
+            assert text == self.TEST_TEXT
+
+    def test_colored_os_name_posix(self):
+
+        with patch("onaplogging.colorFormatter.os.name", "posix"):
+            text = colored(self.TEST_TEXT)
+            assert text == self.TEST_TEXT + RESET
+
+            text = colored(self.TEST_TEXT, color="black")
+            assert text == FMT_STR % (COLORS["black"], self.TEST_TEXT) + RESET
+
+            text = colored(self.TEST_TEXT, color="invalid")
+            assert text == FMT_STR % (0, self.TEST_TEXT) + RESET
+
+            text = colored(self.TEST_TEXT, on_color="red")
+            assert text == FMT_STR % (HIGHLIGHTS["red"], self.TEST_TEXT) + RESET
+
+            text = colored(self.TEST_TEXT, on_color="invalid")
+            assert text == FMT_STR % (0, self.TEST_TEXT) + RESET
+
+            text = colored(self.TEST_TEXT, attrs="bold")
+            assert text == FMT_STR % (ATTRIBUTES["bold"], self.TEST_TEXT) + RESET
+
+            text = colored(self.TEST_TEXT, attrs=["bold", "blink"])
+            assert (
+                text
+                == FMT_STR % (ATTRIBUTES["blink"], FMT_STR % (ATTRIBUTES["bold"], self.TEST_TEXT))
+                + RESET
+            )
+
+            text = colored(self.TEST_TEXT, attrs="invalid")
+            assert text == FMT_STR % (0, self.TEST_TEXT) + RESET
+
+    def test_base_color_formatter(self):
+
+        if is_above_python_3_2():
+            with pytest.raises(ValueError):
+                BaseColorFormatter(style="!")
+
+        TEST_MESSAGE = "TestMessage"
+        record = LogRecord(
+            name="TestName",
+            level=0,
+            pathname="TestPathName",
+            lineno=1,
+            msg=TEST_MESSAGE,
+            args=None,
+            exc_info=None,
+        )
+
+        base_formatter = BaseColorFormatter()
+        assert base_formatter.format(record) == TEST_MESSAGE + RESET
+
+        base_formatter = BaseColorFormatter(fmt="TEST %(message)s")
+        assert base_formatter.format(record) == "TEST " + TEST_MESSAGE + RESET
+
+        colorfmt = {record.levelname: {"color": "black", "highlight": "red", "attribute": "bold"}}
+        base_formatter = BaseColorFormatter(colorfmt=colorfmt)
+        assert (
+            base_formatter.format(record)
+            == FMT_STR
+            % (
+                ATTRIBUTES["bold"],
+                FMT_STR % (HIGHLIGHTS["red"], FMT_STR % (COLORS["black"], "TestMessage")),
+            )
+            + RESET
+        )