20
20
from tests import mockserver
21
21
22
22
23
+ def get_logger (name , level = logging .INFO ):
24
+ logger = logging .getLogger (name )
25
+ logger .setLevel (level )
26
+ return logger
27
+
28
+
23
29
class TestHandler (unittest .TestCase ):
24
30
def setUp (self ):
25
31
super (TestHandler , self ).setUp ()
@@ -40,8 +46,7 @@ def test_simple(self):
40
46
handler = self .get_handler_class ()('app.follow' , port = self ._port )
41
47
42
48
with handler :
43
- logging .basicConfig (level = logging .INFO )
44
- log = logging .getLogger ('fluent.test' )
49
+ log = get_logger ('fluent.test' )
45
50
handler .setFormatter (fluent .handler .FluentRecordFormatter ())
46
51
log .addHandler (handler )
47
52
log .info ({
@@ -63,8 +68,7 @@ def test_custom_fmt(self):
63
68
handler = self .get_handler_class ()('app.follow' , port = self ._port )
64
69
65
70
with handler :
66
- logging .basicConfig (level = logging .INFO )
67
- log = logging .getLogger ('fluent.test' )
71
+ log = get_logger ('fluent.test' )
68
72
handler .setFormatter (
69
73
fluent .handler .FluentRecordFormatter (fmt = {
70
74
'name' : '%(name)s' ,
@@ -86,8 +90,7 @@ def test_custom_fmt_with_format_style(self):
86
90
handler = self .get_handler_class ()('app.follow' , port = self ._port )
87
91
88
92
with handler :
89
- logging .basicConfig (level = logging .INFO )
90
- log = logging .getLogger ('fluent.test' )
93
+ log = get_logger ('fluent.test' )
91
94
handler .setFormatter (
92
95
fluent .handler .FluentRecordFormatter (fmt = {
93
96
'name' : '{name}' ,
@@ -109,8 +112,7 @@ def test_custom_fmt_with_template_style(self):
109
112
handler = self .get_handler_class ()('app.follow' , port = self ._port )
110
113
111
114
with handler :
112
- logging .basicConfig (level = logging .INFO )
113
- log = logging .getLogger ('fluent.test' )
115
+ log = get_logger ('fluent.test' )
114
116
handler .setFormatter (
115
117
fluent .handler .FluentRecordFormatter (fmt = {
116
118
'name' : '${name}' ,
@@ -131,8 +133,7 @@ def test_custom_field_raise_exception(self):
131
133
handler = self .get_handler_class ()('app.follow' , port = self ._port )
132
134
133
135
with handler :
134
- logging .basicConfig (level = logging .INFO )
135
- log = logging .getLogger ('fluent.test' )
136
+ log = get_logger ('fluent.test' )
136
137
handler .setFormatter (
137
138
fluent .handler .FluentRecordFormatter (fmt = {
138
139
'name' : '%(name)s' ,
@@ -147,8 +148,7 @@ def test_custom_field_raise_exception(self):
147
148
def test_custom_field_fill_missing_fmt_key_is_true (self ):
148
149
handler = self .get_handler_class ()('app.follow' , port = self ._port )
149
150
with handler :
150
- logging .basicConfig (level = logging .INFO )
151
- log = logging .getLogger ('fluent.test' )
151
+ log = get_logger ('fluent.test' )
152
152
handler .setFormatter (
153
153
fluent .handler .FluentRecordFormatter (fmt = {
154
154
'name' : '%(name)s' ,
@@ -172,8 +172,7 @@ def test_json_encoded_message(self):
172
172
handler = self .get_handler_class ()('app.follow' , port = self ._port )
173
173
174
174
with handler :
175
- logging .basicConfig (level = logging .INFO )
176
- log = logging .getLogger ('fluent.test' )
175
+ log = get_logger ('fluent.test' )
177
176
handler .setFormatter (fluent .handler .FluentRecordFormatter ())
178
177
log .addHandler (handler )
179
178
log .info ('{"key": "hello world!", "param": "value"}' )
@@ -186,8 +185,7 @@ def test_unstructured_message(self):
186
185
handler = self .get_handler_class ()('app.follow' , port = self ._port )
187
186
188
187
with handler :
189
- logging .basicConfig (level = logging .INFO )
190
- log = logging .getLogger ('fluent.test' )
188
+ log = get_logger ('fluent.test' )
191
189
handler .setFormatter (fluent .handler .FluentRecordFormatter ())
192
190
log .addHandler (handler )
193
191
log .info ('hello %s' , 'world' )
@@ -200,8 +198,7 @@ def test_unstructured_formatted_message(self):
200
198
handler = self .get_handler_class ()('app.follow' , port = self ._port )
201
199
202
200
with handler :
203
- logging .basicConfig (level = logging .INFO )
204
- log = logging .getLogger ('fluent.test' )
201
+ log = get_logger ('fluent.test' )
205
202
handler .setFormatter (fluent .handler .FluentRecordFormatter ())
206
203
log .addHandler (handler )
207
204
log .info ('hello world, %s' , 'you!' )
@@ -214,8 +211,7 @@ def test_number_string_simple_message(self):
214
211
handler = self .get_handler_class ()('app.follow' , port = self ._port )
215
212
216
213
with handler :
217
- logging .basicConfig (level = logging .INFO )
218
- log = logging .getLogger ('fluent.test' )
214
+ log = get_logger ('fluent.test' )
219
215
handler .setFormatter (fluent .handler .FluentRecordFormatter ())
220
216
log .addHandler (handler )
221
217
log .info ("1" )
@@ -227,8 +223,7 @@ def test_non_string_simple_message(self):
227
223
handler = self .get_handler_class ()('app.follow' , port = self ._port )
228
224
229
225
with handler :
230
- logging .basicConfig (level = logging .INFO )
231
- log = logging .getLogger ('fluent.test' )
226
+ log = get_logger ('fluent.test' )
232
227
handler .setFormatter (fluent .handler .FluentRecordFormatter ())
233
228
log .addHandler (handler )
234
229
log .info (42 )
@@ -240,8 +235,7 @@ def test_non_string_dict_message(self):
240
235
handler = self .get_handler_class ()('app.follow' , port = self ._port )
241
236
242
237
with handler :
243
- logging .basicConfig (level = logging .INFO )
244
- log = logging .getLogger ('fluent.test' )
238
+ log = get_logger ('fluent.test' )
245
239
handler .setFormatter (fluent .handler .FluentRecordFormatter ())
246
240
log .addHandler (handler )
247
241
log .info ({42 : 'root' })
@@ -254,8 +248,7 @@ def test_exception_message(self):
254
248
handler = self .get_handler_class ()('app.follow' , port = self ._port )
255
249
256
250
with handler :
257
- logging .basicConfig (level = logging .INFO )
258
- log = logging .getLogger ('fluent.test' )
251
+ log = get_logger ('fluent.test' )
259
252
handler .setFormatter (fluent .handler .FluentRecordFormatter ())
260
253
log .addHandler (handler )
261
254
try :
@@ -297,8 +290,7 @@ def test_simple(self):
297
290
self .assertEqual (handler .sender .queue_circular , True )
298
291
self .assertEqual (handler .sender .queue_maxsize , self .Q_SIZE )
299
292
300
- logging .basicConfig (level = logging .INFO )
301
- log = logging .getLogger ('fluent.test' )
293
+ log = get_logger ('fluent.test' )
302
294
handler .setFormatter (fluent .handler .FluentRecordFormatter ())
303
295
log .addHandler (handler )
304
296
log .info ({'cnt' : 1 , 'from' : 'userA' , 'to' : 'userB' })
@@ -359,8 +351,7 @@ def custom_full_queue():
359
351
self .assertEqual (handler .sender .queue_circular , True )
360
352
self .assertEqual (handler .sender .queue_maxsize , self .Q_SIZE )
361
353
362
- logging .basicConfig (level = logging .INFO )
363
- log = logging .getLogger ('fluent.test' )
354
+ log = get_logger ('fluent.test' )
364
355
handler .setFormatter (fluent .handler .FluentRecordFormatter ())
365
356
log .addHandler (handler )
366
357
0 commit comments