76 lines
3.1 KiB
Python
76 lines
3.1 KiB
Python
"""Unit tests for the buildscripts.resmokelib.logging.loggers module."""
|
|
|
|
import unittest
|
|
from unittest.mock import MagicMock
|
|
|
|
from buildscripts.resmokelib import config
|
|
from buildscripts.resmokelib.logging import loggers
|
|
|
|
|
|
class TestLoggers(unittest.TestCase):
|
|
"""Unit tests for the resmoke loggers."""
|
|
|
|
def setUp(self):
|
|
loggers.ROOT_EXECUTOR_LOGGER = MagicMock()
|
|
loggers.ROOT_FIXTURE_LOGGER = MagicMock()
|
|
loggers.ROOT_TESTS_LOGGER = MagicMock()
|
|
config.LOGGING_CONFIG = MagicMock()
|
|
|
|
def test_resmoke_logger(self):
|
|
logger = loggers.new_resmoke_logger()
|
|
self.assertEqual(logger.parent, loggers.ROOT_EXECUTOR_LOGGER)
|
|
|
|
def test_job_logger(self):
|
|
logger = loggers.new_job_logger("dummy_test_kind", 55)
|
|
self.assertEqual(logger.parent, loggers.ROOT_EXECUTOR_LOGGER)
|
|
|
|
def test_fixture_node_logger(self):
|
|
mock_fixture_logger = MagicMock()
|
|
loggers._FIXTURE_LOGGER_REGISTRY[24] = mock_fixture_logger
|
|
logger = loggers.new_fixture_node_logger("dummy_class", 24, "dummy_node")
|
|
self.assertEqual(logger.parent, mock_fixture_logger)
|
|
|
|
def test_testqueue_logger(self):
|
|
logger = loggers.new_testqueue_logger("dummy_test_kind")
|
|
self.assertEqual(logger.parent, loggers.ROOT_TESTS_LOGGER)
|
|
|
|
def test_test_thread_logger(self):
|
|
logger = loggers.new_test_thread_logger("dummy_parent", "dummy_kind", "dummy_id")
|
|
self.assertEqual(logger.parent, "dummy_parent")
|
|
|
|
def test_hook_logger(self):
|
|
loggers._FIXTURE_LOGGER_REGISTRY[22] = "dummy_fixture_logger"
|
|
logger = loggers.new_hook_logger("dummy_class", 22)
|
|
self.assertEqual(logger.parent, "dummy_fixture_logger")
|
|
|
|
def test_shorten_logger_name(self):
|
|
config.SHORTEN_LOGGER_NAME_CONFIG = {
|
|
"remove": ["MongoDFixture", "ReplicaSetFixture", "ShardedClusterFixture"],
|
|
"replace": {
|
|
"primary": "prim",
|
|
"secondary": "sec",
|
|
"node": "n",
|
|
"shard": "s",
|
|
"configsvr": "c",
|
|
"mongos": "s",
|
|
"job": "j",
|
|
},
|
|
}
|
|
transform = [
|
|
("MongoDFixture:job0", "j0"),
|
|
("ReplicaSetFixture:job0:primary", "j0:prim"),
|
|
("ReplicaSetFixture:job0:secondary", "j0:sec"),
|
|
("ReplicaSetFixture:job0:secondary0", "j0:sec0"),
|
|
("ReplicaSetFixture:job0:node0", "j0:n0"),
|
|
("ShardedClusterFixture:job0:mongos", "j0:s"),
|
|
("ShardedClusterFixture:job0:configsvr:primary", "j0:c:prim"),
|
|
("ShardedClusterFixture:job0:configsvr:secondary", "j0:c:sec"),
|
|
("ShardedClusterFixture:job0:configsvr:secondary0", "j0:c:sec0"),
|
|
("ShardedClusterFixture:job0:shard0:primary", "j0:s0:prim"),
|
|
("ShardedClusterFixture:job0:shard0:secondary", "j0:s0:sec"),
|
|
("ShardedClusterFixture:job0:shard0:secondary0", "j0:s0:sec0"),
|
|
]
|
|
for full_name, expected_short_name in transform:
|
|
short_name = loggers._shorten(full_name)
|
|
self.assertEqual(short_name, expected_short_name)
|