Skip to content

Commit

Permalink
add logging tests, and fix issue if cwd is not writable
Browse files Browse the repository at this point in the history
  • Loading branch information
rettigl committed Feb 8, 2025
1 parent c9a8855 commit 0d337da
Show file tree
Hide file tree
Showing 3 changed files with 108 additions and 17 deletions.
1 change: 1 addition & 0 deletions .cspell/custom-dictionary.txt
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ basepath
bitshift
bysource
calib
caplog
checkscan
clim
codemirror
Expand Down
38 changes: 21 additions & 17 deletions src/specsanalyzer/logging.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,28 +41,32 @@ def setup_logging(
# Create base logger
base_logger = logging.getLogger("specsanalyzer")
base_logger.setLevel(logging.DEBUG) # Set the minimum log level for the logger
if set_base_handler or not base_logger.hasHandlers():
if base_logger.hasHandlers():
if set_base_handler or len(base_logger.handlers) == 0:
if len(base_logger.handlers):
base_logger.handlers.clear()

# Determine log file path
if user_log_path is None:
user_log_path = DEFAULT_LOG_DIR
os.makedirs(user_log_path, exist_ok=True)
log_file = os.path.join(user_log_path, f"sed_{datetime.now().strftime('%Y-%m-%d')}.log")

# Create file handler and set level to debug
file_handler = logging.FileHandler(log_file)
file_handler.setLevel(FILE_VERBOSITY)

# Create formatter for file
file_formatter = logging.Formatter(
"%(asctime)s - %(name)s - %(levelname)s - %(message)s in %(filename)s:%(lineno)d",
)
file_handler.setFormatter(file_formatter)

# Add file handler to logger
base_logger.addHandler(file_handler)
try:
os.makedirs(user_log_path, exist_ok=True)
log_file = os.path.join(user_log_path, f"sed_{datetime.now().strftime('%Y-%m-%d')}.log")

# Create file handler and set level to debug
file_handler = logging.FileHandler(log_file)
file_handler.setLevel(FILE_VERBOSITY)

# Create formatter for file
file_formatter = logging.Formatter(
"%(asctime)s - %(name)s - %(levelname)s - %(message)s in %(filename)s:%(lineno)d",
)
file_handler.setFormatter(file_formatter)

# Add file handler to logger
base_logger.addHandler(file_handler)
except PermissionError:
logging.warning(f"Cannot create logfile in Folder {user_log_path}, disabling logfile.")
base_logger.addHandler(logging.NullHandler())

# create named logger
logger = base_logger.getChild(name)
Expand Down
86 changes: 86 additions & 0 deletions tests/test_logging.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,86 @@
import io
import logging
import os
from datetime import datetime

import pytest

from specsanalyzer.logging import set_verbosity
from specsanalyzer.logging import setup_logging


@pytest.fixture
def logger_():
logger = setup_logging("test_logger")
log_capture_string = io.StringIO()
ch = logging.StreamHandler(log_capture_string)
ch.setLevel(logging.DEBUG)
logger.addHandler(ch)
yield logger, log_capture_string


def test_debug_logging(logger_):
logger, log_capture_string = logger_
logger.debug("This is a debug message")
assert "This is a debug message" in log_capture_string.getvalue()


def test_info_logging(logger_):
logger, log_capture_string = logger_
logger.info("This is an info message")
assert "This is an info message" in log_capture_string.getvalue()


def test_warning_logging(logger_):
logger, log_capture_string = logger_
logger.warning("This is a warning message")
assert "This is a warning message" in log_capture_string.getvalue()


def test_error_logging(logger_):
logger, log_capture_string = logger_
logger.error("This is an error message")
assert "This is an error message" in log_capture_string.getvalue()


def test_critical_logging(logger_):
logger, log_capture_string = logger_
logger.critical("This is a critical message")
assert "This is a critical message" in log_capture_string.getvalue()


def test_set_verbosity(logger_):
logger, log_capture_string = logger_
set_verbosity(logger, verbose=True)
assert logger.handlers[0].level == logging.INFO
set_verbosity(logger, verbose=False)
assert logger.handlers[0].level == logging.WARNING


def test_logger_has_base_logger(logger_):
logger, log_capture_string = logger_
assert logger.name == "specsanalyzer.test_logger"
assert logger.parent.name == "specsanalyzer"
assert logger.parent.parent.name == "root"
assert logger.parent.level == logging.DEBUG
assert isinstance(logger.parent.handlers[0], logging.FileHandler)


def test_logger_creates_logfile(tmp_path):
logger = setup_logging("test_logger", set_base_handler=True, user_log_path=tmp_path)
log_file = os.path.join(tmp_path, f"sed_{datetime.now().strftime('%Y-%m-%d')}.log")
assert os.path.exists(log_file)
with open(log_file) as f:
assert f.read() == ""
logger.debug("This is a debug message")
with open(log_file) as f:
assert "This is a debug message" in f.read()


def test_readonly_path(tmp_path, caplog):
os.chmod(tmp_path, 0o444)
with caplog.at_level(logging.WARNING):
setup_logging("test_logger", set_base_handler=True, user_log_path=tmp_path)
assert f"Cannot create logfile in Folder {tmp_path}, disabling logfile." in caplog.messages[0]
log_file = os.path.join(tmp_path, f"sed_{datetime.now().strftime('%Y-%m-%d')}.log")
assert not os.path.exists(log_file)

0 comments on commit 0d337da

Please sign in to comment.