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
|
"""
test json related things
"""
import json
import tempfile
import logzero
def _test_json_obj_content(obj):
# Check that all fields are contained
attrs = ['asctime', 'filename', 'funcName', 'levelname', 'levelno', 'lineno', 'module', 'message', 'name', 'pathname', 'process', 'processName', 'threadName']
assert obj["message"] == "info"
for attr in attrs:
if attr not in obj:
raise Exception("obj missing key '%s'" % attr)
def test_json(capsys):
"""
Test json logging
"""
# Test setup_logger
logger = logzero.setup_logger(json=True)
logger.info('info')
out, err = capsys.readouterr()
_test_json_obj_content(json.loads(err))
def test_json_default_logger(capsys):
# Test default logger
logzero.reset_default_logger()
logzero.logger.info('info')
out, err = capsys.readouterr()
assert "] info" in err
logzero.json()
logzero.logger.info('info')
out, err = capsys.readouterr()
_test_json_obj_content(json.loads(err))
logzero.json(False)
logzero.logger.info('info')
out, err = capsys.readouterr()
assert "] info" in err
def test_json_logfile(capsys):
# Test default logger
logzero.reset_default_logger()
temp = tempfile.NamedTemporaryFile()
try:
logger = logzero.setup_logger(logfile=temp.name, json=True)
logger.info('info')
with open(temp.name) as f:
content = f.read()
_test_json_obj_content(json.loads(content))
finally:
temp.close()
def test_json_encoding(capsys):
"""
see logzero.json(json_ensure_ascii=True)
"""
logzero.reset_default_logger()
# UTF-8 mode
logzero.json(json_ensure_ascii=False)
logzero.logger.info('ß')
out, err = capsys.readouterr()
json.loads(err) # make sure JSON is valid
assert 'ß' in err
assert 'u00df' not in err
# ASCII mode
logzero.json(json_ensure_ascii=True)
logzero.logger.info('ß')
out, err = capsys.readouterr()
json.loads(err) # make sure JSON is valid
assert 'u00df' in err
assert 'ß' not in err
# Default JSON mode should be utf-8
logzero.json()
logzero.logger.info('ß')
out, err = capsys.readouterr()
json.loads(err) # make sure JSON is valid
assert 'ß' in err
assert 'u00df' not in err
|
