diff options
Diffstat (limited to 'openbb_platform/core/tests/app/logs/test_handlers_manager.py')
-rw-r--r-- | openbb_platform/core/tests/app/logs/test_handlers_manager.py | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/openbb_platform/core/tests/app/logs/test_handlers_manager.py b/openbb_platform/core/tests/app/logs/test_handlers_manager.py index 12cb4565d9d..d46451c6314 100644 --- a/openbb_platform/core/tests/app/logs/test_handlers_manager.py +++ b/openbb_platform/core/tests/app/logs/test_handlers_manager.py @@ -1,3 +1,5 @@ +"""Tests for the handlers manager.""" + import logging from unittest.mock import Mock, patch @@ -7,25 +9,37 @@ from openbb_core.app.logs.handlers_manager import ( PosthogHandler, ) +# pylint: disable=W0231 + class MockPosthogHandler(logging.NullHandler): + """Mock posthog handler.""" + def __init__(self, settings): + """Initialize the handler.""" self.settings = settings self.level = logging.DEBUG class MockPathTrackingFileHandler(logging.NullHandler): + """Mock path tracking file handler.""" + def __init__(self, settings): + """Initialize the handler.""" self.settings = settings self.level = logging.DEBUG class MockFormatterWithExceptions(logging.Formatter): + """Mock formatter with exceptions.""" + def __init__(self, settings): + """Initialize the formatter.""" self.settings = settings def test_handlers_added_correctly(): + """Test if the handlers are added correctly.""" with patch( "openbb_core.app.logs.handlers_manager.PosthogHandler", MockPosthogHandler, @@ -60,6 +74,7 @@ def test_handlers_added_correctly(): def test_update_handlers(): + """Test if the handlers are updated correctly.""" with patch( "openbb_core.app.logs.handlers_manager.PosthogHandler", MockPosthogHandler, @@ -85,4 +100,4 @@ def test_update_handlers(): for hdlr in handlers: if isinstance(hdlr, (MockPosthogHandler, MockPathTrackingFileHandler)): assert hdlr.settings == changed_settings - assert hdlr.formatter.settings == changed_settings + assert hdlr.formatter.settings == changed_settings # type: ignore[union-attr] |