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
|
from loguru import logger
def test_patch_after_add(writer):
logger.add(writer, format="{extra[a]} {message}")
logger_patched = logger.patch(lambda r: r["extra"].update(a=0))
logger_patched.debug("A")
assert writer.read() == "0 A\n"
def test_patch_before_add(writer):
logger_patched = logger.patch(lambda r: r["extra"].update(a=0))
logger.add(writer, format="{extra[a]} {message}")
logger_patched.debug("A")
assert writer.read() == "0 A\n"
def test_add_using_patched(writer):
logger.configure(patcher=lambda r: r["extra"].update(a=-1))
logger_patched = logger.patch(lambda r: r["extra"].update(a=0))
logger_patched.add(writer, format="{extra[a]} {message}")
logger.debug("A")
logger_patched.debug("B")
assert writer.read() == "-1 A\n0 B\n"
def test_not_override_parent_logger(writer):
logger_1 = logger.patch(lambda r: r["extra"].update(a="a"))
logger_2 = logger_1.patch(lambda r: r["extra"].update(a="A"))
logger.add(writer, format="{extra[a]} {message}")
logger_1.debug("1")
logger_2.debug("2")
assert writer.read() == "a 1\nA 2\n"
def test_override_previous_patched(writer):
logger.add(writer, format="{extra[x]} {message}")
logger2 = logger.patch(lambda r: r["extra"].update(x=3))
logger2.patch(lambda r: r["extra"].update(x=2)).debug("4")
assert writer.read() == "2 4\n"
def test_no_conflict(writer):
logger_ = logger.patch(lambda r: None)
logger_2 = logger_.patch(lambda r: r["extra"].update(a=2))
logger_3 = logger_.patch(lambda r: r["extra"].update(a=3))
logger.add(writer, format="{extra[a]} {message}")
logger_2.debug("222")
logger_3.debug("333")
assert writer.read() == "2 222\n3 333\n"
def test_override_configured(writer):
logger.configure(patcher=lambda r: r["extra"].update(a=123, b=678))
logger2 = logger.patch(lambda r: r["extra"].update(a=456))
logger2.add(writer, format="{extra[a]} {extra[b]} {message}")
logger2.debug("!")
assert writer.read() == "456 678 !\n"
def test_multiple_patches(writer):
def patch_1(record):
record["extra"]["a"] = 5
def patch_2(record):
record["extra"]["a"] += 1
def patch_3(record):
record["extra"]["a"] *= 2
logger.add(writer, format="{extra[a]} {message}")
logger.patch(patch_1).patch(patch_2).patch(patch_3).info("Test")
assert writer.read() == "12 Test\n"
|
