Adding upstream version 86.0.1.upstream/86.0.1upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 44 insertions, 0 deletions

diff --git a/testing/mozharness/test/test_base_log.py b/testing/mozharness/test/test_base_log.py
new file mode 100644
index 0000000000..cf2e52356f
--- /dev/null
+++ b/testing/mozharness/test/test_base_log.py
@@ -0,0 +1,44 @@
+from __future__ import absolute_import
+import os
+import shutil
+import unittest
+
+import mozharness.base.log as log
+
+tmp_dir = "test_log_dir"
+log_name = "test"
+
+
+def clean_log_dir():
+    if os.path.exists(tmp_dir):
+        shutil.rmtree(tmp_dir)
+
+
+def get_log_file_path(level=None):
+    if level:
+        return os.path.join(tmp_dir, "%s_%s.log" % (log_name, level))
+    return os.path.join(tmp_dir, "%s.log" % log_name)
+
+
+class TestLog(unittest.TestCase):
+    def setUp(self):
+        clean_log_dir()
+
+    def tearDown(self):
+        clean_log_dir()
+
+    def test_log_dir(self):
+        fh = open(tmp_dir, "w")
+        fh.write("foo")
+        fh.close()
+        l = log.SimpleFileLogger(
+            log_dir=tmp_dir, log_name=log_name, log_to_console=False
+        )
+        self.assertTrue(os.path.exists(tmp_dir))
+        l.log_message("blah")
+        self.assertTrue(os.path.exists(get_log_file_path()))
+        del l
+
+
+if __name__ == "__main__":
+    unittest.main()