forked from fluent/fluent-logger-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_handler.py
More file actions
123 lines (99 loc) · 3.93 KB
/
Copy pathtest_handler.py
File metadata and controls
123 lines (99 loc) · 3.93 KB
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
# -*- coding: utf-8 -*-
import logging
import unittest
import fluent.handler
from tests import mockserver
class TestHandler(unittest.TestCase):
def setUp(self):
super(TestHandler, self).setUp()
for port in range(10000, 20000):
try:
self._server = mockserver.MockRecvServer('localhost', port)
self._port = port
break
except IOError:
pass
def get_data(self):
return self._server.get_recieved()
def test_simple(self):
handler = fluent.handler.FluentHandler('app.follow', port=self._port)
logging.basicConfig(level=logging.INFO)
log = logging.getLogger('fluent.test')
handler.setFormatter(fluent.handler.FluentRecordFormatter())
log.addHandler(handler)
log.info({
'from': 'userA',
'to': 'userB'
})
handler.close()
data = self.get_data()
eq = self.assertEqual
eq(1, len(data))
eq(3, len(data[0]))
eq('app.follow', data[0][0])
eq('userA', data[0][2]['from'])
eq('userB', data[0][2]['to'])
self.assertTrue(data[0][1])
self.assertTrue(isinstance(data[0][1], int))
def test_custom_fmt(self):
handler = fluent.handler.FluentHandler('app.follow', port=self._port)
logging.basicConfig(level=logging.INFO)
log = logging.getLogger('fluent.test')
handler.setFormatter(
fluent.handler.FluentRecordFormatter(fmt={
'name': '%(name)s',
'lineno': '%(lineno)d',
'emitted_at': '%(asctime)s',
})
)
log.addHandler(handler)
log.info({'sample': 'value'})
handler.close()
data = self.get_data()
self.assertTrue('name' in data[0][2])
self.assertEqual('fluent.test', data[0][2]['name'])
self.assertTrue('lineno' in data[0][2])
self.assertTrue('emitted_at' in data[0][2])
def test_json_encoded_message(self):
handler = fluent.handler.FluentHandler('app.follow', port=self._port)
logging.basicConfig(level=logging.INFO)
log = logging.getLogger('fluent.test')
handler.setFormatter(fluent.handler.FluentRecordFormatter())
log.addHandler(handler)
log.info('{"key": "hello world!", "param": "value"}')
handler.close()
data = self.get_data()
self.assertTrue('key' in data[0][2])
self.assertEqual('hello world!', data[0][2]['key'])
def test_unstructured_message(self):
handler = fluent.handler.FluentHandler('app.follow', port=self._port)
logging.basicConfig(level=logging.INFO)
log = logging.getLogger('fluent.test')
handler.setFormatter(fluent.handler.FluentRecordFormatter())
log.addHandler(handler)
log.info('hello world')
handler.close()
data = self.get_data()
self.assertTrue('message' in data[0][2])
self.assertEqual('hello world', data[0][2]['message'])
def test_non_string_simple_message(self):
handler = fluent.handler.FluentHandler('app.follow', port=self._port)
logging.basicConfig(level=logging.INFO)
log = logging.getLogger('fluent.test')
handler.setFormatter(fluent.handler.FluentRecordFormatter())
log.addHandler(handler)
log.info(42)
handler.close()
data = self.get_data()
self.assertTrue('message' in data[0][2])
def test_non_string_dict_message(self):
handler = fluent.handler.FluentHandler('app.follow', port=self._port)
logging.basicConfig(level=logging.INFO)
log = logging.getLogger('fluent.test')
handler.setFormatter(fluent.handler.FluentRecordFormatter())
log.addHandler(handler)
log.info({42: 'root'})
handler.close()
data = self.get_data()
# For some reason, non-string keys are ignored
self.assertFalse(42 in data[0][2])