pywiki/tests/logging-pyhmy/test_logging.py

import os

import pytest

from pyhmy import logging


def test_basic_logger():
    if os.path.exists(f"{os.getcwd()}/logs/pytest.log"):
        os.remove(f"{os.getcwd()}/logs/pytest.log")
    logger = logging.ControlledLogger("pytest", "logs/")
    assert os.path.exists(f"{os.getcwd()}/logs/pytest.log")
    logger.info("test info")
    logger.debug("test debug")
    logger.error("test error")
    logger.warning("test warning")
    with open(f"{os.getcwd()}/logs/pytest.log", "r") as f:
        log_file_contents = f.readlines()
    assert not log_file_contents
    logger.write()
    with open(f"{os.getcwd()}/logs/pytest.log", "r") as f:
        log_file_contents = f.readlines()
    for line in log_file_contents:
        if "INFO" in line:
            assert "test info" in line
        if "DEBUG" in line:
            assert "test debug" in line
        if "WARNING" in line:
            assert "test warning" in line
        if "ERROR" in line:
            assert "test error" in line