75 lines
3.1 KiB
Python
75 lines
3.1 KiB
Python
"""Unit tests for the buildscripts.resmokelib.logging.loggers module."""
|
|
|
|
import json
|
|
import unittest
|
|
from unittest.mock import MagicMock
|
|
|
|
from buildscripts.resmokelib.logging import loggers
|
|
from buildscripts.resmokelib import config
|
|
|
|
# pylint: disable=missing-docstring,protected-access
|
|
|
|
|
|
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()
|
|
loggers.BUILDLOGGER_SERVER = MagicMock()
|
|
loggers._get_buildlogger_handler_info = 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):
|
|
loggers.BUILDLOGGER_SERVER.new_build_id.return_value = 78
|
|
logger = loggers.new_job_logger("dummy_test_kind", 55)
|
|
self.assertEqual(loggers._BUILD_ID_REGISTRY[55], 78)
|
|
self.assertEqual(logger.parent, loggers.ROOT_EXECUTOR_LOGGER)
|
|
|
|
def test_fixture_logger(self):
|
|
loggers._BUILD_ID_REGISTRY[32] = 29
|
|
loggers._get_buildlogger_handler_info.return_value = True
|
|
mock_handler = MagicMock()
|
|
loggers.BUILDLOGGER_SERVER.get_global_handler.return_value = mock_handler
|
|
logger = loggers.new_fixture_logger("dummy_class", 32)
|
|
self.assertEqual(logger.handlers[0], mock_handler)
|
|
|
|
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_logger(self):
|
|
loggers._BUILD_ID_REGISTRY[88] = 47
|
|
loggers._get_buildlogger_handler_info.return_value = True
|
|
mock_handler = MagicMock()
|
|
loggers.BUILDLOGGER_SERVER.get_test_handler.return_value = mock_handler
|
|
loggers.BUILDLOGGER_SERVER.new_test_id.return_value = 22
|
|
loggers.BUILDLOGGER_SERVER.get_test_log_url.return_value = "dummy_url"
|
|
|
|
mock_parent = MagicMock()
|
|
(logger, url) = loggers.new_test_logger("dummy_shortname", "dummy_basename",
|
|
"dummy_command", mock_parent, 88, MagicMock())
|
|
self.assertEqual(logger.handlers[0], mock_handler)
|
|
self.assertEqual(logger.parent, mock_parent)
|
|
self.assertEqual(url, "dummy_url")
|
|
|
|
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")
|