1 files changed, 57 insertions, 0 deletions
diff --git a/scripts/devscripts/test/test_logger.py b/scripts/devscripts/test/test_logger.py
new file mode 100644
index 0000000..e6322ea
--- /dev/null
+++ b/scripts/devscripts/test/test_logger.py
@@ -0,0 +1,57 @@
+# test_logger.py - Test devscripts.logger.Logger.
+#
+# Copyright (C) 2012, Stefano Rivera <stefanor@debian.org>
+#
+# Permission to use, copy, modify, and/or distribute this software for any
+# purpose with or without fee is hereby granted, provided that the above
+# copyright notice and this permission notice appear in all copies.
+#
+# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
+# REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
+# AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
+# INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
+# LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
+# OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
+# PERFORMANCE OF THIS SOFTWARE.
+
+import io
+import sys
+
+from devscripts.logger import Logger
+from devscripts.test import unittest
+
+
+class LoggerTestCase(unittest.TestCase):
+    def setUp(self):
+        Logger.stdout = io.StringIO()
+        Logger.stderr = io.StringIO()
+        self._script_name = Logger.script_name
+        Logger.script_name = "test"
+        self._verbose = Logger.verbose
+
+    def tearDown(self):
+        Logger.stdout = sys.stdout
+        Logger.stderr = sys.stderr
+        Logger.script_name = self._script_name
+        Logger.verbose = self._verbose
+
+    def test_command(self):
+        # pylint: disable=no-member
+        Logger.command(("ls", "a b"))
+        self.assertEqual(Logger.stdout.getvalue(), "")
+        Logger.set_verbosity(True)
+        Logger.command(("ls", "a b"))
+        self.assertEqual(Logger.stdout.getvalue(), 'test: I: ls "a b"\n')
+        self.assertEqual(Logger.stderr.getvalue(), "")
+
+    def test_no_args(self):
+        # pylint: disable=no-member
+        Logger.normal("hello %s")
+        self.assertEqual(Logger.stdout.getvalue(), "test: hello %s\n")
+        self.assertEqual(Logger.stderr.getvalue(), "")
+
+    def test_args(self):
+        # pylint: disable=no-member
+        Logger.normal("hello %s", "world")
+        self.assertEqual(Logger.stdout.getvalue(), "test: hello world\n")
+        self.assertEqual(Logger.stderr.getvalue(), "")