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

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

import unittest

from lib_not_dr.logger.config import ConfigStorage


class TestConfigStorage(unittest.TestCase):
    def test_detect_circle_require(self):
        test_dict = {
            "a": ["b", "c"],
            "b": ["c"],
            "c": ["a"],
            "d": ["e"],
            "e": ["f"],
            "f": ["d"],
        }
        self.assertEqual(
            ["a", "b", "c", "d", "e", "f"], ConfigStorage.find_cycles(test_dict)
        )

    def test_detect_circle_require_2(self):
        # large circle
        cycles = sorted(map(str, range(100)))
        for i in range(100):
            test_dict = {}
            for j in range(100):
                test_dict[str(j)] = [str((j + 1) % 100)]
            self.assertEqual(cycles, ConfigStorage.find_cycles(test_dict))

    def test_detect_circle_require_3(self):
        # normal
        test_dict = {"a": [], "b": ["a"], "c": ["b"], "d": ["c"]}
        self.assertEqual([], ConfigStorage.find_cycles(test_dict))

    def test_parse_formatter_config(self):
        test_config = {
            "main": {"class": "MainFormatter", "sub_formatter": []},
            "cycle_a": {"class": "MainFormatter", "sub_formatter": ["cycle_b"]},
            "cycle_b": {"class": "MainFormatter", "sub_formatter": ["cycle_a"]},
        }
        test_storage = ConfigStorage()
        test_storage.parse_formatter(test_config)
        self.assertTrue(test_storage.have_formatter("main"))
        self.assertFalse(test_storage.have_formatter("cycle_a"))
        self.assertFalse(test_storage.have_formatter("cycle_b"))