mirror of
https://github.com/classilla/tenfourfox.git
synced 2024-06-10 02:29:43 +00:00
43 lines
959 B
Python
43 lines
959 B
Python
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()
|