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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
test_logzero
----------------------------------
Tests for `logzero` module.
"""
import os
import tempfile
import logzero
def test_api_logfile(capsys):
"""
logzero.logfile(..) should work as expected
"""
logzero.reset_default_logger()
temp = tempfile.NamedTemporaryFile()
try:
logzero.logger.info("info1")
# Set logfile
logzero.logfile(temp.name)
logzero.logger.info("info2")
# Remove logfile
logzero.logfile(None)
logzero.logger.info("info3")
# Set logfile again
logzero.logfile(temp.name)
logzero.logger.info("info4")
with open(temp.name) as f:
content = f.read()
assert "] info1" not in content
assert "] info2" in content
assert "] info3" not in content
assert "] info4" in content
finally:
temp.close()
def test_api_loglevel(capsys):
"""
Should reconfigure the internal logger loglevel
"""
logzero.reset_default_logger()
temp = tempfile.NamedTemporaryFile()
try:
logzero.logfile(temp.name)
logzero.logger.info("info1")
logzero.loglevel(logzero.WARN)
logzero.logger.info("info2")
logzero.logger.warning("warn1")
with open(temp.name) as f:
content = f.read()
assert "] info1" in content
assert "] info2" not in content
assert "] warn1" in content
finally:
temp.close()
def test_api_loglevel_custom_handlers(capsys):
"""
Should reconfigure the internal logger loglevel and custom handlers
"""
logzero.reset_default_logger()
# TODO
pass
# temp = tempfile.NamedTemporaryFile()
# try:
# logzero.logfile(temp.name)
# logzero.logger.info("info1")
# logzero.loglevel(logzero.WARN)
# logzero.logger.info("info2")
# logzero.logger.warning("warn1")
# with open(temp.name) as f:
# content = f.read()
# assert "] info1" in content
# assert "] info2" not in content
# assert "] warn1" in content
# finally:
# temp.close()
def test_api_rotating_logfile(capsys):
"""
logzero.rotating_logfile(..) should work as expected
"""
logzero.reset_default_logger()
temp = tempfile.NamedTemporaryFile()
try:
logzero.logger.info("info1")
# Set logfile
logzero.logfile(temp.name, maxBytes=10, backupCount=3)
logzero.logger.info("info2")
logzero.logger.info("info3")
with open(temp.name) as f:
content = f.read()
assert "] info1" not in content # logged before setting up logfile
assert "] info2" not in content # already rotated out
assert "] info3" in content # already rotated out
fn_rotated = temp.name + ".1"
assert os.path.exists(fn_rotated)
with open(fn_rotated) as f:
content = f.read()
assert "] info2" in content
finally:
temp.close()
def test_api_logfile_custom_loglevel():
"""
logzero.logfile(..) should be able to use a custom loglevel
"""
logzero.reset_default_logger()
temp = tempfile.NamedTemporaryFile()
try:
# Set logfile with custom loglevel
logzero.logfile(temp.name, loglevel=logzero.WARN)
logzero.logger.info("info1")
logzero.logger.warning("warn1")
# If setting a loglevel with logzero.loglevel(..) it will not overwrite
# the custom loglevel of the file handler
logzero.loglevel(logzero.INFO)
logzero.logger.info("info2")
logzero.logger.warning("warn2")
with open(temp.name) as f:
content = f.read()
assert "] info1" not in content
assert "] warn1" in content
assert "] info2" not in content
assert "] warn2" in content
finally:
temp.close()
|
