Skip to content

Commit

Permalink
tests: fix small error
Browse files Browse the repository at this point in the history
  • Loading branch information
jjjermiah committed Feb 5, 2024
1 parent d2a6f55 commit 977df6c
Showing 1 changed file with 17 additions and 8 deletions.
25 changes: 17 additions & 8 deletions tests/test_logger.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,33 +5,42 @@

def test_setup_logger_name():
# Test case 1: Verify that the logger is created with the correct name
logger = setup_logger("test_logger")
assert logger.name == "test_logger"
logger = setup_logger("test_logger1")
assert logger.name == "test_logger1"

def test_setup_logger_level():
# Test case 2: Verify that the logger level is set correctly
logger = setup_logger("test_logger", log_level="DEBUG")
logger = setup_logger("test_logger2", log_level="DEBUG")
assert logger.level == logging.DEBUG

def test_setup_logger_console_logging():
# Test case 3: Verify that the logger logs to console when console_logging is True
logger = setup_logger("test_logger", console_logging=True)
logger = setup_logger("test_logger3", console_logging=True)
assert len(logger.handlers) == 1
assert isinstance(logger.handlers[0], logging.StreamHandler)
# close the file handler
logger.handlers[0].close()


def test_setup_logger_log_file():
# Test case 4: Verify that the logger logs to file when log_file is provided
log_file = "test.log"
logger = setup_logger("test_logger", log_file=log_file)
logger = setup_logger("test_logger4", log_file=log_file)
assert len(logger.handlers) == 1
assert isinstance(logger.handlers[0], TimedRotatingFileHandler)
assert logger.handlers[0].baseFilename == os.path.abspath(log_file)

# close the file handler
logger.handlers[0].close()


def test_setup_logger_log_dir():
# Test case 5: Verify that the logger logs to file in the specified log_dir
log_file = "test.log"
log_dir = "logs"
logger = setup_logger("test_logger", log_file=log_file, log_dir=log_dir)
assert len(logger.handlers) == 1
log_dir = "logdir"
logger = setup_logger("test_logger5", log_file=log_file, log_dir=log_dir)
assert isinstance(logger.handlers[0], TimedRotatingFileHandler)
assert logger.handlers[0].baseFilename == os.path.abspath(os.path.join(log_dir, log_file))

# close the file handler
logger.handlers[0].close()

0 comments on commit 977df6c

Please sign in to comment.