Skip to content
This repository was archived by the owner on Dec 21, 2024. It is now read-only.

Commit a910fd8

Browse files
Rename variables and functions in the tests to follow PEP8 principle (#155)
Co-authored-by: Louis Melchior <louis@jaris.io>
1 parent d42f92c commit a910fd8

File tree

1 file changed

+92
-92
lines changed

1 file changed

+92
-92
lines changed

tests/tests.py

Lines changed: 92 additions & 92 deletions
Original file line numberDiff line numberDiff line change
@@ -21,60 +21,60 @@
2121

2222
class TestJsonLogger(unittest.TestCase):
2323
def setUp(self):
24-
self.logger = logging.getLogger("logging-test-{}".format(random.randint(1, 101)))
25-
self.logger.setLevel(logging.DEBUG)
24+
self.log = logging.getLogger("logging-test-{}".format(random.randint(1, 101)))
25+
self.log.setLevel(logging.DEBUG)
2626
self.buffer = StringIO()
2727

28-
self.logHandler = logging.StreamHandler(self.buffer)
29-
self.logger.addHandler(self.logHandler)
28+
self.log_handler = logging.StreamHandler(self.buffer)
29+
self.log.addHandler(self.log_handler)
3030

31-
def testDefaultFormat(self):
31+
def test_default_format(self):
3232
fr = jsonlogger.JsonFormatter()
33-
self.logHandler.setFormatter(fr)
33+
self.log_handler.setFormatter(fr)
3434

3535
msg = "testing logging format"
36-
self.logger.info(msg)
37-
logJson = json.loads(self.buffer.getvalue())
36+
self.log.info(msg)
37+
log_json = json.loads(self.buffer.getvalue())
3838

39-
self.assertEqual(logJson["message"], msg)
39+
self.assertEqual(log_json["message"], msg)
4040

41-
def testPercentageFormat(self):
41+
def test_percentage_format(self):
4242
fr = jsonlogger.JsonFormatter(
4343
# All kind of different styles to check the regex
4444
'[%(levelname)8s] %(message)s %(filename)s:%(lineno)d %(asctime)'
4545
)
46-
self.logHandler.setFormatter(fr)
46+
self.log_handler.setFormatter(fr)
4747

4848
msg = "testing logging format"
49-
self.logger.info(msg)
50-
logJson = json.loads(self.buffer.getvalue())
49+
self.log.info(msg)
50+
log_json = json.loads(self.buffer.getvalue())
5151

52-
self.assertEqual(logJson["message"], msg)
53-
self.assertEqual(logJson.keys(), {'levelname', 'message', 'filename', 'lineno', 'asctime'})
52+
self.assertEqual(log_json["message"], msg)
53+
self.assertEqual(log_json.keys(), {'levelname', 'message', 'filename', 'lineno', 'asctime'})
5454

55-
def testRenameBaseField(self):
55+
def test_rename_base_field(self):
5656
fr = jsonlogger.JsonFormatter(rename_fields={'message': '@message'})
57-
self.logHandler.setFormatter(fr)
57+
self.log_handler.setFormatter(fr)
5858

5959
msg = "testing logging format"
60-
self.logger.info(msg)
61-
logJson = json.loads(self.buffer.getvalue())
60+
self.log.info(msg)
61+
log_json = json.loads(self.buffer.getvalue())
6262

63-
self.assertEqual(logJson["@message"], msg)
63+
self.assertEqual(log_json["@message"], msg)
6464

65-
def testAddStaticFields(self):
65+
def test_add_static_fields(self):
6666
fr = jsonlogger.JsonFormatter(static_fields={'log_stream': 'kafka'})
6767

68-
self.logHandler.setFormatter(fr)
68+
self.log_handler.setFormatter(fr)
6969

7070
msg = "testing static fields"
71-
self.logger.info(msg)
72-
logJson = json.loads(self.buffer.getvalue())
71+
self.log.info(msg)
72+
log_json = json.loads(self.buffer.getvalue())
7373

74-
self.assertEqual(logJson["log_stream"], "kafka")
75-
self.assertEqual(logJson["message"], msg)
74+
self.assertEqual(log_json["log_stream"], "kafka")
75+
self.assertEqual(log_json["message"], msg)
7676

77-
def testFormatKeys(self):
77+
def test_format_keys(self):
7878
supported_keys = [
7979
'asctime',
8080
'created',
@@ -99,96 +99,97 @@ def testFormatKeys(self):
9999
custom_format = ' '.join(log_format(supported_keys))
100100

101101
fr = jsonlogger.JsonFormatter(custom_format)
102-
self.logHandler.setFormatter(fr)
102+
self.log_handler.setFormatter(fr)
103103

104104
msg = "testing logging format"
105-
self.logger.info(msg)
105+
self.log.info(msg)
106106
log_msg = self.buffer.getvalue()
107107
log_json = json.loads(log_msg)
108108

109109
for supported_key in supported_keys:
110110
if supported_key in log_json:
111111
self.assertTrue(True)
112112

113-
def testUnknownFormatKey(self):
113+
def test_unknown_format_key(self):
114114
fr = jsonlogger.JsonFormatter('%(unknown_key)s %(message)s')
115115

116-
self.logHandler.setFormatter(fr)
116+
self.log_handler.setFormatter(fr)
117117
msg = "testing unknown logging format"
118118
try:
119-
self.logger.info(msg)
120-
except:
119+
self.log.info(msg)
120+
except Exception:
121121
self.assertTrue(False, "Should succeed")
122122

123-
def testLogADict(self):
123+
def test_log_adict(self):
124124
fr = jsonlogger.JsonFormatter()
125-
self.logHandler.setFormatter(fr)
125+
self.log_handler.setFormatter(fr)
126126

127127
msg = {"text": "testing logging", "num": 1, 5: "9",
128128
"nested": {"more": "data"}}
129-
self.logger.info(msg)
130-
logJson = json.loads(self.buffer.getvalue())
131-
self.assertEqual(logJson.get("text"), msg["text"])
132-
self.assertEqual(logJson.get("num"), msg["num"])
133-
self.assertEqual(logJson.get("5"), msg[5])
134-
self.assertEqual(logJson.get("nested"), msg["nested"])
135-
self.assertEqual(logJson["message"], "")
136-
137-
def testLogExtra(self):
129+
130+
self.log.info(msg)
131+
log_json = json.loads(self.buffer.getvalue())
132+
self.assertEqual(log_json.get("text"), msg["text"])
133+
self.assertEqual(log_json.get("num"), msg["num"])
134+
self.assertEqual(log_json.get("5"), msg[5])
135+
self.assertEqual(log_json.get("nested"), msg["nested"])
136+
self.assertEqual(log_json["message"], None)
137+
138+
def test_log_extra(self):
138139
fr = jsonlogger.JsonFormatter()
139-
self.logHandler.setFormatter(fr)
140+
self.log_handler.setFormatter(fr)
140141

141142
extra = {"text": "testing logging", "num": 1, 5: "9",
142143
"nested": {"more": "data"}}
143-
self.logger.info("hello", extra=extra)
144-
logJson = json.loads(self.buffer.getvalue())
145-
self.assertEqual(logJson.get("text"), extra["text"])
146-
self.assertEqual(logJson.get("num"), extra["num"])
147-
self.assertEqual(logJson.get("5"), extra[5])
148-
self.assertEqual(logJson.get("nested"), extra["nested"])
149-
self.assertEqual(logJson["message"], "hello")
150-
151-
def testJsonDefaultEncoder(self):
144+
self.log.info("hello", extra=extra)
145+
log_json = json.loads(self.buffer.getvalue())
146+
self.assertEqual(log_json.get("text"), extra["text"])
147+
self.assertEqual(log_json.get("num"), extra["num"])
148+
self.assertEqual(log_json.get("5"), extra[5])
149+
self.assertEqual(log_json.get("nested"), extra["nested"])
150+
self.assertEqual(log_json["message"], "hello")
151+
152+
def test_json_default_encoder(self):
152153
fr = jsonlogger.JsonFormatter()
153-
self.logHandler.setFormatter(fr)
154+
self.log_handler.setFormatter(fr)
154155

155156
msg = {"adate": datetime.datetime(1999, 12, 31, 23, 59),
156157
"otherdate": datetime.date(1789, 7, 14),
157158
"otherdatetime": datetime.datetime(1789, 7, 14, 23, 59),
158159
"otherdatetimeagain": datetime.datetime(1900, 1, 1)}
159-
self.logger.info(msg)
160-
logJson = json.loads(self.buffer.getvalue())
161-
self.assertEqual(logJson.get("adate"), "1999-12-31T23:59:00")
162-
self.assertEqual(logJson.get("otherdate"), "1789-07-14")
163-
self.assertEqual(logJson.get("otherdatetime"), "1789-07-14T23:59:00")
164-
self.assertEqual(logJson.get("otherdatetimeagain"),
160+
self.log.info(msg)
161+
log_json = json.loads(self.buffer.getvalue())
162+
self.assertEqual(log_json.get("adate"), "1999-12-31T23:59:00")
163+
self.assertEqual(log_json.get("otherdate"), "1789-07-14")
164+
self.assertEqual(log_json.get("otherdatetime"), "1789-07-14T23:59:00")
165+
self.assertEqual(log_json.get("otherdatetimeagain"),
165166
"1900-01-01T00:00:00")
166167

167168
@unittest.mock.patch('time.time', return_value=1500000000.0)
168-
def testJsonDefaultEncoderWithTimestamp(self, time_mock):
169+
def test_json_default_encoder_with_timestamp(self, time_mock):
169170
fr = jsonlogger.JsonFormatter(timestamp=True)
170-
self.logHandler.setFormatter(fr)
171+
self.log_handler.setFormatter(fr)
171172

172-
self.logger.info("Hello")
173+
self.log.info("Hello")
173174

174175
self.assertTrue(time_mock.called)
175-
logJson = json.loads(self.buffer.getvalue())
176-
self.assertEqual(logJson.get("timestamp"), "2017-07-14T02:40:00+00:00")
176+
log_json = json.loads(self.buffer.getvalue())
177+
self.assertEqual(log_json.get("timestamp"), "2017-07-14T02:40:00+00:00")
177178

178-
def testJsonCustomDefault(self):
179+
def test_json_custom_default(self):
179180
def custom(o):
180181
return "very custom"
181182
fr = jsonlogger.JsonFormatter(json_default=custom)
182-
self.logHandler.setFormatter(fr)
183+
self.log_handler.setFormatter(fr)
183184

184185
msg = {"adate": datetime.datetime(1999, 12, 31, 23, 59),
185186
"normal": "value"}
186-
self.logger.info(msg)
187-
logJson = json.loads(self.buffer.getvalue())
188-
self.assertEqual(logJson.get("adate"), "very custom")
189-
self.assertEqual(logJson.get("normal"), "value")
187+
self.log.info(msg)
188+
log_json = json.loads(self.buffer.getvalue())
189+
self.assertEqual(log_json.get("adate"), "very custom")
190+
self.assertEqual(log_json.get("normal"), "value")
190191

191-
def testJsonCustomLogicAddsField(self):
192+
def test_json_custom_logic_adds_field(self):
192193
class CustomJsonFormatter(jsonlogger.JsonFormatter):
193194

194195
def process_log_record(self, log_record):
@@ -197,43 +198,42 @@ def process_log_record(self, log_record):
197198
# style
198199
return jsonlogger.JsonFormatter.process_log_record(self, log_record)
199200

200-
self.logHandler.setFormatter(CustomJsonFormatter())
201-
self.logger.info("message")
202-
logJson = json.loads(self.buffer.getvalue())
203-
self.assertEqual(logJson.get("custom"), "value")
201+
self.log_handler.setFormatter(CustomJsonFormatter())
202+
self.log.info("message")
203+
log_json = json.loads(self.buffer.getvalue())
204+
self.assertEqual(log_json.get("custom"), "value")
204205

205-
def testExcInfo(self):
206+
def test_exc_info(self):
206207
fr = jsonlogger.JsonFormatter()
207-
self.logHandler.setFormatter(fr)
208+
self.log_handler.setFormatter(fr)
208209
try:
209210
raise Exception('test')
210211
except Exception:
211-
212-
self.logger.exception("hello")
212+
self.log.exception("hello")
213213

214214
expected_value = traceback.format_exc()
215215
# Formatter removes trailing new line
216216
if expected_value.endswith('\n'):
217217
expected_value = expected_value[:-1]
218218

219-
logJson = json.loads(self.buffer.getvalue())
220-
self.assertEqual(logJson.get("exc_info"), expected_value)
219+
log_json = json.loads(self.buffer.getvalue())
220+
self.assertEqual(log_json.get("exc_info"), expected_value)
221221

222-
def testEnsureAsciiTrue(self):
222+
def test_ensure_ascii_true(self):
223223
fr = jsonlogger.JsonFormatter()
224-
self.logHandler.setFormatter(fr)
225-
self.logger.info('Привет')
224+
self.log_handler.setFormatter(fr)
225+
self.log.info('Привет')
226226
msg = self.buffer.getvalue().split('"message": "', 1)[1].split('"', 1)[0]
227227
self.assertEqual(msg, r"\u041f\u0440\u0438\u0432\u0435\u0442")
228228

229-
def testEnsureAsciiFalse(self):
229+
def test_ensure_ascii_false(self):
230230
fr = jsonlogger.JsonFormatter(json_ensure_ascii=False)
231-
self.logHandler.setFormatter(fr)
232-
self.logger.info('Привет')
231+
self.log_handler.setFormatter(fr)
232+
self.log.info('Привет')
233233
msg = self.buffer.getvalue().split('"message": "', 1)[1].split('"', 1)[0]
234234
self.assertEqual(msg, "Привет")
235235

236-
def testCustomObjectSerialization(self):
236+
def test_custom_object_serialization(self):
237237
def encode_complex(z):
238238
if isinstance(z, complex):
239239
return (z.real, z.imag)
@@ -243,13 +243,13 @@ def encode_complex(z):
243243

244244
formatter = jsonlogger.JsonFormatter(json_default=encode_complex,
245245
json_encoder=json.JSONEncoder)
246-
self.logHandler.setFormatter(formatter)
246+
self.log_handler.setFormatter(formatter)
247247

248248
value = {
249249
"special": complex(3, 8),
250250
}
251251

252-
self.logger.info(" message", extra=value)
252+
self.log.info(" message", extra=value)
253253
msg = self.buffer.getvalue()
254254
self.assertEqual(msg, "{\"message\": \" message\", \"special\": [3.0, 8.0]}\n")
255255

0 commit comments

Comments
 (0)