|
26 | 26 | import unittest
|
27 | 27 | import logging
|
28 | 28 |
|
29 |
| -from inputremapper.logger import logger, add_filehandler, update_verbosity, log_info |
| 29 | +from inputremapper.logger import logger, update_verbosity, log_info, ColorfulFormatter |
30 | 30 | from inputremapper.configs.paths import remove
|
31 | 31 |
|
32 | 32 |
|
| 33 | +def add_filehandler(log_path): |
| 34 | + """Start logging to a file.""" |
| 35 | + log_path = os.path.expanduser(log_path) |
| 36 | + os.makedirs(os.path.dirname(log_path), exist_ok=True) |
| 37 | + file_handler = logging.FileHandler(log_path) |
| 38 | + file_handler.setFormatter(ColorfulFormatter()) |
| 39 | + logger.addHandler(file_handler) |
| 40 | + logger.info('Starting logging to "%s"', log_path) |
| 41 | + |
| 42 | + |
33 | 43 | class TestLogger(unittest.TestCase):
|
34 | 44 | def tearDown(self):
|
35 | 45 | update_verbosity(debug=True)
|
@@ -77,22 +87,6 @@ def test_makes_path(self):
|
77 | 87 | add_filehandler(new_path)
|
78 | 88 | self.assertTrue(os.path.exists(new_path))
|
79 | 89 |
|
80 |
| - def test_clears_log(self): |
81 |
| - path = os.path.join(tmp, "logger-test") |
82 |
| - os.makedirs(os.path.dirname(path), exist_ok=True) |
83 |
| - os.mknod(path) |
84 |
| - |
85 |
| - with open(path, "w") as f: |
86 |
| - f.write("aaaa\n" * 2000 + "end") |
87 |
| - |
88 |
| - add_filehandler(os.path.join(tmp, "logger-test")) |
89 |
| - with open(path, "r") as f: |
90 |
| - # it only keeps the newest information |
91 |
| - content = f.readlines() |
92 |
| - self.assertLess(abs(len(content) - 1000), 10) |
93 |
| - # whatever the logging module decides to log into that file |
94 |
| - self.assertNotIn("aaaa", content[-1]) |
95 |
| - |
96 | 90 | def test_debug(self):
|
97 | 91 | path = os.path.join(tmp, "logger-test")
|
98 | 92 | add_filehandler(path)
|
|
0 commit comments