stable-diffusion-for-fun/utilities/logger_test.py

35 lines
1.0 KiB
Python

import os
import unittest
from utilities.logger import Logger
class TestLogger(unittest.TestCase):
@classmethod
def setUpClass(self):
self.logger = Logger()
self.log_filepath = "/tmp/test/tmplog.log"
def test_logger_print(self):
self.logger.set_log_output_filepath(self.log_filepath)
self.logger.debug("A quirky message only developers care about")
self.logger.info("Curious users might want to know this")
self.logger.warn("Something is wrong and any user should be informed")
self.logger.error("Serious stuff, this is red for a reason")
self.logger.critical("OH NO everything is on fire")
self.logger.debugging_on()
self.logger.debug("A quirky message only developers care about")
self.assertTrue(os.path.isfile(self.log_filepath))
@classmethod
def tearDownClass(self):
if os.path.isfile(self.log_filepath):
os.remove(self.log_filepath)
if __name__ == "__main__":
unittest.main()