lib-not-dr/test/logger/formatter.py

#  -------------------------------
#  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()