-
Notifications
You must be signed in to change notification settings - Fork 507
/
Copy pathtest_logging.py
133 lines (106 loc) · 4.22 KB
/
test_logging.py
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
import logging
import flask_restplus as restplus
class LoggingTest(object):
def test_namespace_loggers_log_to_flask_app_logger(self, app, client, caplog):
# capture Flask app logs
caplog.set_level(logging.INFO, logger=app.logger.name)
api = restplus.Api(app)
ns1 = api.namespace('ns1', path='/ns1')
ns2 = api.namespace('ns2', path='/ns2')
@ns1.route('/')
class Ns1(restplus.Resource):
def get(self):
ns1.logger.info("hello from ns1")
pass
@ns2.route('/')
class Ns2(restplus.Resource):
def get(self):
ns2.logger.info("hello from ns2")
pass
# debug log not shown
client.get("/ns1/")
matching = [r for r in caplog.records if r.message == "hello from ns1"]
assert len(matching) == 1
# info log shown
client.get("/ns2/")
matching = [r for r in caplog.records if r.message == "hello from ns2"]
assert len(matching) == 1
def test_defaults_to_app_level(self, app, client, caplog):
caplog.set_level(logging.INFO, logger=app.logger.name)
api = restplus.Api(app)
ns1 = api.namespace('ns1', path='/ns1')
ns2 = api.namespace('ns2', path='/ns2')
@ns1.route('/')
class Ns1(restplus.Resource):
def get(self):
ns1.logger.debug("hello from ns1")
pass
@ns2.route('/')
class Ns2(restplus.Resource):
def get(self):
ns2.logger.info("hello from ns2")
pass
# debug log not shown
client.get("/ns1/")
matching = [r for r in caplog.records if r.message == "hello from ns1"]
assert len(matching) == 0
# info log shown
client.get("/ns2/")
matching = [r for r in caplog.records if r.message == "hello from ns2"]
assert len(matching) == 1
def test_override_app_level(self, app, client, caplog):
caplog.set_level(logging.INFO, logger=app.logger.name)
api = restplus.Api(app)
ns1 = api.namespace('ns1', path='/ns1')
ns1.logger.setLevel(logging.DEBUG)
ns2 = api.namespace('ns2', path='/ns2')
@ns1.route('/')
class Ns1(restplus.Resource):
def get(self):
ns1.logger.debug("hello from ns1")
pass
@ns2.route('/')
class Ns2(restplus.Resource):
def get(self):
ns2.logger.debug("hello from ns2")
pass
# debug log shown from ns1
client.get("/ns1/")
matching = [r for r in caplog.records if r.message == "hello from ns1"]
assert len(matching) == 1
# debug not shown from ns2
client.get("/ns2/")
matching = [r for r in caplog.records if r.message == "hello from ns2"]
assert len(matching) == 0
def test_namespace_additional_handler(self, app, client, caplog, tmp_path):
caplog.set_level(logging.INFO, logger=app.logger.name)
log_file = tmp_path / "v1.log"
api = restplus.Api(app)
ns1 = api.namespace('ns1', path='/ns1')
# set up a file handler for ns1 only
# FileHandler only supports Path object on Python >= 3.6 -> cast to str
fh = logging.FileHandler(str(log_file))
ns1.logger.addHandler(fh)
ns2 = api.namespace('ns2', path='/ns2')
@ns1.route('/')
class Ns1(restplus.Resource):
def get(self):
ns1.logger.info("hello from ns1")
pass
@ns2.route('/')
class Ns2(restplus.Resource):
def get(self):
ns2.logger.info("hello from ns2")
pass
# ns1 logs go to stdout and log_file
client.get("/ns1/")
matching = [r for r in caplog.records if r.message == "hello from ns1"]
assert len(matching) == 1
with log_file.open() as f:
assert "hello from ns1" in f.read()
# ns2 logs go to stdout only
client.get("/ns2/")
matching = [r for r in caplog.records if r.message == "hello from ns2"]
assert len(matching) == 1
with log_file.open() as f:
assert "hello from ns1" in f.read()