1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
import functools
import re
import warnings
from io import StringIO
from random import randrange
import pytest
import logbook
import logbook.compat
from .utils import capturing_stderr_context
__file_without_pyc__ = __file__
if __file_without_pyc__.endswith(".pyc"):
__file_without_pyc__ = __file_without_pyc__[:-1]
@pytest.mark.parametrize("set_root_logger_level", [True, False])
def test_basic_compat(request, set_root_logger_level):
import logging
from logbook.compat import redirected_logging
# mimic the default logging setting
request.addfinalizer(functools.partial(logging.root.setLevel, logging.root.level))
logging.root.setLevel(logging.WARNING)
name = "test_logbook-%d" % randrange(1 << 32)
logger = logging.getLogger(name)
with logbook.TestHandler(bubble=True) as handler:
with capturing_stderr_context() as captured:
with redirected_logging(set_root_logger_level):
logger.debug("This is from the old system")
logger.info("This is from the old system")
logger.warning("This is from the old %s", "system")
logger.error("This is from the old system")
logger.critical("This is from the old system")
logger.error(
"This is a %(what)s %(where)s", {"what": "mapping", "where": "test"}
)
assert (
"WARNING: %s: This is from the old system" % name
) in captured.getvalue()
assert ("ERROR: %s: This is a mapping test" % name) in captured.getvalue()
if set_root_logger_level:
assert handler.records[0].level == logbook.DEBUG
else:
assert handler.records[0].level == logbook.WARNING
assert handler.records[0].msg == "This is from the old %s"
def test_redirect_logbook():
import logging
out = StringIO()
logger = logging.getLogger()
logbook_logger = logbook.Logger("testlogger")
old_handlers = logger.handlers[:]
handler = logging.StreamHandler(out)
handler.setFormatter(logging.Formatter("%(name)s:%(levelname)s:%(message)s"))
logger.handlers[:] = [handler]
try:
with logbook.compat.LoggingHandler():
logbook_logger.warn("This goes to logging")
pieces = out.getvalue().strip().split(":")
assert pieces == ["testlogger", "WARNING", "This goes to logging"]
finally:
logger.handlers[:] = old_handlers
from itertools import count
test_warning_redirections_i = count()
def test_warning_redirections():
from logbook.compat import redirected_warnings
with logbook.TestHandler() as handler:
with redirected_warnings():
warnings.warn(
RuntimeWarning(f"Testing {next(test_warning_redirections_i)}")
)
assert len(handler.formatted_records) == 1
assert handler.formatted_records[0].startswith("[WARNING] RuntimeWarning: Testing")
assert __file_without_pyc__ in handler.records[0].filename
|
