New upstream version 0.0.15

tests/test_logs.py

@@ -0,0 +1,95 @@
+#!/usr/bin/python
+# Copyright (C) 2022 Jelmer Vernooij <jelmer@jelmer.uk>
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+
+import os
+import sys
+import tempfile
+from unittest import TestCase
+
+from ognibuild.logs import (
+    copy_output,
+    redirect_output,
+    rotate_logfile,
+    DirectoryLogManager,
+)
+
+
+class TestCopyOutput(TestCase):
+
+    def test_no_tee(self):
+        with tempfile.TemporaryDirectory() as td:
+            p = os.path.join(td, 'foo.log')
+            with copy_output(p, tee=False):
+                sys.stdout.write('lala\n')
+                sys.stdout.flush()
+            with open(p, 'r') as f:
+                self.assertEqual('lala\n', f.read())
+
+    def test_tee(self):
+        with tempfile.TemporaryDirectory() as td:
+            p = os.path.join(td, 'foo.log')
+            with copy_output(p, tee=True):
+                sys.stdout.write('lala\n')
+                sys.stdout.flush()
+            with open(p, 'r') as f:
+                self.assertEqual('lala\n', f.read())
+
+
+class TestRedirectOutput(TestCase):
+
+    def test_simple(self):
+        with tempfile.TemporaryDirectory() as td:
+            p = os.path.join(td, 'foo.log')
+            with open(p, 'w') as f:
+                with redirect_output(f):
+                    sys.stdout.write('lala\n')
+                    sys.stdout.flush()
+            with open(p, 'r') as f:
+                self.assertEqual('lala\n', f.read())
+
+
+class TestRotateLogfile(TestCase):
+
+    def test_does_not_exist(self):
+        with tempfile.TemporaryDirectory() as td:
+            p = os.path.join(td, 'foo.log')
+            rotate_logfile(p)
+            self.assertEqual([], os.listdir(td))
+
+    def test_simple(self):
+        with tempfile.TemporaryDirectory() as td:
+            p = os.path.join(td, 'foo.log')
+            with open(p, 'w') as f:
+                f.write('contents\n')
+            rotate_logfile(p)
+            self.assertEqual(['foo.log.1'], os.listdir(td))
+
+
+class TestLogManager(TestCase):
+
+    def test_simple(self):
+        with tempfile.TemporaryDirectory() as td:
+            p = os.path.join(td, 'foo.log')
+            lm = DirectoryLogManager(p, mode='redirect')
+
+            def writesomething():
+                sys.stdout.write('foo\n')
+                sys.stdout.flush()
+            fn = lm.wrap(writesomething)
+            fn()
+            with open(p, 'r') as f:
+                self.assertEqual('foo\n', f.read())