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