1 files changed, 77 insertions, 0 deletions
diff --git a/tests/test_logging.py b/tests/test_logging.py
new file mode 100644
index 00000000000..848c736a8d2
--- /dev/null
+++ b/tests/test_logging.py
@@ -0,0 +1,77 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+Unit tests for the dogtail.logging package
+"""
+import unittest
+import dogtail.tree
+from gtkdemotest import trap_stdout
+
+
+class TestLogging(unittest.TestCase):
+
+    def setUp(self):
+        self.old_log_dir = dogtail.config.config.logDir
+
+    def tearDown(self):
+        dogtail.config.config.logDebugToFile = False
+        dogtail.config.config.logDir = self.old_log_dir
+
+    def test_entryStamp_is_not_empty(self):
+        ts = dogtail.logging.TimeStamp()
+        self.assertEquals(len(ts.entryStamp()) > 0, True)
+
+    def test_correct_error_if_log_dir_does_not_exist(self):
+        import shutil
+        shutil.rmtree(dogtail.config.config.logDir)
+        self.assertRaises(IOError, dogtail.logging.Logger, "log", file=True)
+
+    def test_unique_name(self):
+        logger1 = dogtail.logging.Logger("log", file=True)
+        logger1.createFile()
+        logger2 = dogtail.logging.Logger("log", file=True)
+        logger2.createFile()
+        logger3 = dogtail.logging.Logger("log", file=True)
+        self.assertNotEquals(logger1.fileName, logger2.fileName)
+        self.assertNotEquals(logger2.fileName, logger3.fileName)
+
+    def test_no_new_line_to_file(self):
+        dogtail.config.config.logDebugToFile = True
+        logger = dogtail.logging.Logger("log", file=True, stdOut=False)
+        logger.log("hello world", newline=False)
+        self.assertTrue("hello world " in open(logger.fileName, 'r').read())
+
+    def test_no_new_line_to_stdout(self):
+        dogtail.config.config.logDebugToFile = False
+        logger = dogtail.logging.Logger("log", file=False, stdOut=True)
+        output = trap_stdout(
+            logger.log, {'message': 'hello world', 'newline': False})
+        self.assertEquals(output, "hello world")
+
+    def test_no_new_line_to_both_file_and_stdout(self):
+        dogtail.config.config.logDebugToFile = True
+        logger = dogtail.logging.Logger("log", file=True, stdOut=True)
+        output = trap_stdout(
+            logger.log, {'message': 'hello world', 'newline': False})
+        self.assertTrue("hello world" in output)
+        self.assertTrue("hello world " in open(logger.fileName, 'r').read())
+
+    def test_empty_script_name(self):
+        dogtail.config.config.scriptName = None
+        logger = dogtail.logging.Logger("log", file=True, stdOut=True)
+        self.assertTrue(logger.fileName, "log")
+
+    def test_force_to_file(self):
+        dogtail.config.config.logDebugToFile = False
+        logger = dogtail.logging.Logger("log", file=True, stdOut=False)
+        logger.log("hello world", force=True)
+        self.assertTrue("hello world" in open(logger.fileName, 'r').read())
+
+    def test_results_logger_correct_dict(self):
+        logger = dogtail.logging.ResultsLogger("log")
+        output = trap_stdout(logger.log, {'entry': {'a': '1'}})
+        self.assertEquals('a:      1' in output, True)
+
+    def test_results_logger_incorrect_dict(self):
+        logger = dogtail.logging.ResultsLogger("log")
+        self.assertRaises(ValueError, logger.log, "not a dict")