blob: 0947834f70ded044ec97a129aeee75efe41c6937 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
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()
|
