47 lines
1.4 KiB
Python
47 lines
1.4 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, MainFormatter
|
|
from lib_not_dr.logger.structure import LogMessage
|
|
|
|
|
|
class FormatterTest(unittest.TestCase):
|
|
test_levels = (0, 5, 7, 10, 20, 30, 40, 50)
|
|
|
|
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_format_level(self):
|
|
formatter = MainFormatter()
|
|
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()
|