def test_format_with_error(self):
msg = 'Hello'
level = 'info'
now = 'now'
err = Exception('Some error')
try:
raise err
except Exception as e:
err = e
err_data = None
if err != None:
err_data = uio.StringIO()
sys.print_exception(err, err_data)
err_data.seek(0)
err_data = err_data.read().strip().split('\n')
want_payload = {
'msg': msg,
'time': now,
'level': level,
'v': 1,
}
want_payload['err'] = err_data[-1]
want_payload['trace'] = err_data[:-1]
got_message = logger.json_formatter(msg=msg,
level=level,
now=now,
err=err)
self.assertEqual(got_message, ujson.dumps(want_payload))
def test_format_with_msg(self):
msg = 'Something is happening %d' % random.randint(10, 1000)
level = 'info'
now = 'time %d' % random.randint(10, 1000)
want_message = ujson.dumps({
'msg': msg,
'time': now,
'level': level,
'v': 1
})
got_message = logger.json_formatter(msg=msg, level=level, now=now)
self.assertEqual(got_message, want_message)
def test_format_with_context(self):
msg = 'Hello'
level = 'info'
now = 'now'
context = {
'key1': 'val1',
'key2': 'val2',
'key3': 'val3',
}
want_payload = {
'msg': msg,
'time': now,
'level': level,
'v': 1,
}
want_payload['context'] = context
got_message = logger.json_formatter(msg=msg,
level=level,
now=now,
context=context)
self.assertEqual(got_message, ujson.dumps(want_payload))
def test_write_msg(self):
class target():
pass
err = Exception('Test error')
test_context = {"ctx-key1": 'val1', "ctx-key2": 'val2'}
test_data = {"key1": 'val1', "key2": 'val2'}
now = '2020-01-02T03:04:05Z'
got_messages = []
def mock_transport(message):
got_messages.append(message)
logger.setup(
target=target,
transport=mock_transport,
now_fn=lambda: now,
)
target.error('Hello Error',
context=test_context,
data=test_data,
err=err)
target.warn('Hello Warn',
context=test_context,
data=test_data,
err=err)
target.info('Hello Info',
context=test_context,
data=test_data,
err=err)
target.debug('Hello Debug',
context=test_context,
data=test_data,
err=err)
self.assertEqual(got_messages, [
logger.json_formatter(msg='Hello Error',
level='error',
now=now,
context=test_context,
data=test_data,
err=err),
logger.json_formatter(msg='Hello Warn',
level='warn',
now=now,
context=test_context,
data=test_data,
err=err),
logger.json_formatter(msg='Hello Info',
level='info',
now=now,
context=test_context,
data=test_data,
err=err),
logger.json_formatter(msg='Hello Debug',
level='debug',
now=now,
context=test_context,
data=test_data,
err=err),
])