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

#  -------------------------------
#  Difficult Rocket
#  Copyright © 2020-2023 by shenjackyuanjie 3695888@qq.com
#  All rights reserved
#  -------------------------------

import time
import unittest

from lib_not_dr.logger.structure import LogMessage


class LogMessageTest(unittest.TestCase):
    def test_create_empty_message(self):
        """
        测试空消息的创建
        这玩意甚至是 codeiume/github copilot 自动补全的
        """
        message = LogMessage()
        self.assertEqual(message.messages, [])
        self.assertEqual(message.level, 20)
        self.assertEqual(message.logger_name, 'root')
        self.assertEqual(message.logger_tag, None)

    def test_format_message_empty(self):
        """
        测试空消息的格式化
        因为才发现空信息之前的格式化是有问题的(因为没有初始化 messages)
        """
        message = LogMessage()
        self.assertEqual(message.format_message(), '\n')

    def test_format_message_end(self):
        """
        测试消息的结尾
        """
        message = LogMessage(messages=['test'],
                             end='\n')
        for testing in ('\n', '\r\n', '\r'):
            message.end = testing
            self.assertEqual(message.format_message(), 'test' + testing)

    def test_format_message_split(self):
        """
        测试消息的分割
        """
        message = LogMessage(messages=['test', 'test2'], end='')
        for testing in (' ', ' | ', ' | '):
            message.split = testing
            self.assertEqual(message.format_message(), 'test' + testing + 'test2')

    def test_format_message_end_with_split(self):
        """
        测试消息的结尾和分割
        """
        message = LogMessage(messages=['test', 'test2'], end='\n')
        for test_split in (' ', ' | ', ' | '):
            for test_end in ('\n', '\r\n', '\r'):
                message.split = test_split
                message.end = test_end
                self.assertEqual(message.format_message(), 'test' + test_split + 'test2' + test_end)

    def test_msec_3(self):
        start_time = time.time()
        message = LogMessage(log_time=start_time)
        msec_3 = int(start_time / 1000000) % 1000
        self.assertEqual(message.create_msec_3, msec_3)