def tearDownClass(cls):
log = logging.getLogger(__name__)
reusables.remove_all_handlers(log)
if os.path.exists(my_stream_path):
try:
os.unlink(my_stream_path)
except WindowsError:
pass
def test_remove_stream_handlers(self):
logger = reusables.get_logger("sample_stream_logger", file_path=my_stream_path)
logger.addHandler(logging.NullHandler())
for _ in range(10):
logger.addHandler(reusables.get_stream_handler())
reusables.remove_stream_handlers("sample_stream_logger")
assert len(logger.handlers) == 2, logger.handlers
assert isinstance(logger.handlers[0], logging.FileHandler)
reusables.remove_all_handlers(logger)
def test_add_file_logger(self):
logger = reusables.get_logger(__name__)
reusables.add_file_handler(logger, my_fiie_path)
logger.info("Test log")
logger.error("Example error log")
reusables.remove_all_handlers(logger)
with open(my_fiie_path) as f:
lines = f.readlines()
assert "INFO" in lines[0]
assert "ERROR" in lines[1]
assert "Example error log" in lines[1]
def test_get_stream_logger(self):
my_stream = open(my_stream_path, "w")
logger = reusables.get_logger(__name__, stream=my_stream)
logger.info("Test log")
logger.error("Example error log")
my_stream.close()
reusables.remove_all_handlers(logger)
with open(my_stream_path) as f:
lines = f.readlines()
assert "INFO" in lines[0]
assert "ERROR" in lines[1]
assert "Example error log" in lines[1]
def test_remove_file_handlers(self):
logger = reusables.get_logger("sample_file_logger", file_path=my_stream_path)
logger.addHandler(logging.FileHandler("test_file"))
logger.addHandler(logging.NullHandler())
reusables.remove_file_handlers("sample_file_logger")
assert len(logger.handlers) == 2
assert isinstance(logger.handlers[0], logging.StreamHandler)
try:
os.unlink("test_file")
except Exception:
pass
reusables.remove_all_handlers(logger)
def test_add_rotate_file_handlers(self):
from logging.handlers import RotatingFileHandler,\
TimedRotatingFileHandler
logger = reusables.get_logger("add_file")
reusables.remove_all_handlers(logger)
reusables.add_rotating_file_handler("add_file")
assert isinstance(logger.handlers[0], RotatingFileHandler), logger.handlers
reusables.remove_all_handlers("add_file")
reusables.add_timed_rotating_file_handler("add_file")
assert isinstance(logger.handlers[0], TimedRotatingFileHandler)
reusables.remove_all_handlers("add_file")
def test_add_simple_handlers(self):
logger = reusables.get_logger("test1")
reusables.remove_all_handlers("test1")
reusables.add_stream_handler("test1")
assert isinstance(logger.handlers[0], logging.StreamHandler)
reusables.remove_all_handlers("test1")