tor-browser

The Tor Browser
git clone https://git.dasho.dev/tor-browser.git
Log | Files | Refs | README | LICENSE

test_base_log.py (942B)

      1 import os
      2 import shutil
      3 import unittest
      4 
      5 import mozunit
      6 from mozharness.base import log
      7 
      8 tmp_dir = "test_log_dir"
      9 log_name = "test"
     10 
     11 
     12 def clean_log_dir():
     13    if os.path.exists(tmp_dir):
     14        shutil.rmtree(tmp_dir)
     15 
     16 
     17 def get_log_file_path(level=None):
     18    if level:
     19        return os.path.join(tmp_dir, "%s_%s.log" % (log_name, level))
     20    return os.path.join(tmp_dir, "%s.log" % log_name)
     21 
     22 
     23 class TestLog(unittest.TestCase):
     24    def setUp(self):
     25        clean_log_dir()
     26 
     27    def tearDown(self):
     28        clean_log_dir()
     29 
     30    def test_log_dir(self):
     31        fh = open(tmp_dir, "w")
     32        fh.write("foo")
     33        fh.close()
     34        l = log.SimpleFileLogger(
     35            log_dir=tmp_dir, log_name=log_name, log_to_console=False
     36        )
     37        self.assertTrue(os.path.exists(tmp_dir))
     38        l.log_message("blah")
     39        self.assertTrue(os.path.exists(get_log_file_path()))
     40        del l
     41 
     42 
     43 if __name__ == "__main__":
     44    mozunit.main()