62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
# -------------------------------
|
|
# Difficult Rocket
|
|
# Copyright © 2020-2023 by shenjackyuanjie 3695888@qq.com
|
|
# All rights reserved
|
|
# -------------------------------
|
|
|
|
import inspect
|
|
import unittest
|
|
|
|
from lib_not_dr.logger.formatter import BaseFormatter, TimeFormatter, LevelFormatter, TraceFormatter
|
|
from lib_not_dr.logger.structers import LogMessage
|
|
|
|
|
|
class FormatterTest(unittest.TestCase):
|
|
|
|
test_levels = (0, 2, 5, 7, 10, 30, 50, 90)
|
|
|
|
def test_create_message(self):
|
|
message = LogMessage(messages=['test'],
|
|
level=20,
|
|
logger_name='root',
|
|
logger_tag='root',
|
|
stack_trace=inspect.currentframe())
|
|
self.assertEqual(message.messages, ['test'])
|
|
self.assertEqual(message.level, 20)
|
|
self.assertEqual(message.logger_name, 'root')
|
|
self.assertEqual(message.logger_tag, 'root')
|
|
|
|
def test_create_empty_tag_message(self):
|
|
message = LogMessage()
|
|
|
|
self.assertIsNone(message.logger_tag)
|
|
self.assertEqual(message.format_for_message()['logger_tag'], ' ')
|
|
|
|
def test_formatter(self):
|
|
formatter = BaseFormatter()
|
|
formatter.info()
|
|
|
|
def test_time_formatter(self):
|
|
formatter = TimeFormatter()
|
|
formatter.info()
|
|
|
|
def test_level_formatter(self):
|
|
formatter = LevelFormatter()
|
|
formatter.info()
|
|
message = LogMessage(messages=['test'],
|
|
level=0)
|
|
|
|
for test_level in self.test_levels:
|
|
message.level = test_level
|
|
basic_info = message.format_for_message()
|
|
formatting_message = formatter._format((message, basic_info))
|
|
self.assertEqual(formatting_message[1]['level'], formatter.level_name_map[formatting_message[0].level])
|
|
|
|
def test_std_formatter(self):
|
|
formatter = BaseFormatter()
|
|
formatter.info()
|
|
|
|
def test_trace_formatter(self):
|
|
formatter = TraceFormatter()
|
|
formatter.info()
|