fix tests

This commit is contained in:
hjoaquim
2023-08-15 12:10:34 +01:00
parent 835b0e90ba
commit 853fcf72ad
2 changed files with 14 additions and 7 deletions

View File

@@ -1,7 +1,11 @@
import logging
from unittest.mock import Mock, patch
from openbb_core.app.logs.handlers_manager import HandlersManager
from openbb_core.app.logs.handlers_manager import (
HandlersManager,
PosthogHandler,
PathTrackingFileHandler,
)
class MockPosthogHandler(logging.NullHandler):
@@ -39,7 +43,15 @@ def test_handlers_added_correctly():
assert len(handlers) >= 5
for handler in handlers:
assert isinstance(handler, (logging.NullHandler, logging.StreamHandler))
assert isinstance(
handler,
(
logging.NullHandler,
logging.StreamHandler,
PathTrackingFileHandler,
PosthogHandler,
),
)
for mock in [MockPosthogHandler, MockPathTrackingFileHandler]:
assert any(isinstance(handler, mock) for handler in handlers)

View File

@@ -52,11 +52,6 @@ def test_correctly_initialized(logging_manager):
assert logging_manager
def test_logging_settings_property(logging_manager):
assert logging_manager.logging_settings.system_settings == "mock_system_settings"
assert logging_manager.logging_settings.user_settings == "mock_user_settings"
def test_logging_settings_setter(logging_manager):
custom_user_settings = "custom_user_settings"
custom_system_settings = "custom_system_settings"