test ?

OpenBB-finance · Apr 19, 2024 · 79f1dba · 79f1dba
1 parent 85ab51b
commit 79f1dba
Showing 1 changed file with 30 additions and 18 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
@@ -37,18 +37,26 @@ def __init__(self, settings):
         """Initialize the formatter."""
         self.settings = settings
 
+    def _style(self):
+        """Return the style."""
+        return logging.PercentStyle
+
 
 def test_handlers_added_correctly():
     """Test if the handlers are added correctly."""
-    with patch(
-        "openbb_core.app.logs.handlers_manager.PosthogHandler",
-        MockPosthogHandler,
-    ), patch(
-        "openbb_core.app.logs.handlers_manager.PathTrackingFileHandler",
-        MockPathTrackingFileHandler,
-    ), patch(
-        "openbb_core.app.logs.handlers_manager.FormatterWithExceptions",
-        MockFormatterWithExceptions,
+    with (
+        patch(
+            "openbb_core.app.logs.handlers_manager.PosthogHandler",
+            MockPosthogHandler,
+        ),
+        patch(
+            "openbb_core.app.logs.handlers_manager.PathTrackingFileHandler",
+            MockPathTrackingFileHandler,
+        ),
+        patch(
+            "openbb_core.app.logs.handlers_manager.FormatterWithExceptions",
+            MockFormatterWithExceptions,
+        ),
     ):
         settings = Mock()
         settings.handler_list = ["stdout", "stderr", "noop", "file", "posthog"]
@@ -75,15 +83,19 @@ 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,
-    ), patch(
-        "openbb_core.app.logs.handlers_manager.PathTrackingFileHandler",
-        MockPathTrackingFileHandler,
-    ), patch(
-        "openbb_core.app.logs.handlers_manager.FormatterWithExceptions",
-        MockFormatterWithExceptions,
+    with (
+        patch(
+            "openbb_core.app.logs.handlers_manager.PosthogHandler",
+            MockPosthogHandler,
+        ),
+        patch(
+            "openbb_core.app.logs.handlers_manager.PathTrackingFileHandler",
+            MockPathTrackingFileHandler,
+        ),
+        patch(
+            "openbb_core.app.logs.handlers_manager.FormatterWithExceptions",
+            MockFormatterWithExceptions,
+        ),
     ):
         settings = Mock()
         settings.handler_list = ["file", "posthog"]