summaryrefslogtreecommitdiffstats
path: root/testing/mozharness/test/test_base_log.py
diff options
context:
space:
mode:
Diffstat (limited to 'testing/mozharness/test/test_base_log.py')
-rw-r--r--testing/mozharness/test/test_base_log.py42
1 files changed, 42 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 000000000..0947834f7
--- /dev/null
+++ b/testing/mozharness/test/test_base_log.py
@@ -0,0 +1,42 @@
+import os
+import shutil
+import subprocess
+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()