Fix pytest for logging

andreasgriffin · Nov 10, 2024 · ca4789d · ca4789d
1 parent 98c333a
commit ca4789d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/.vscode/launch.json b/.vscode/launch.json
@@ -22,6 +22,7 @@
       "args": [
         "-vvv", 
         "--ignore=coding_tests",
+        "--ignore=tools",
       ],
       "console": "integratedTerminal",
       //  "justMyCode": false

diff --git a/tests/non_gui/test_mail_handler.py b/tests/non_gui/test_mail_handler.py
@@ -55,11 +55,15 @@ def test_no_error_logging(caplog: LogCaptureFixture):
 
 
 def test_exception_logging(caplog: LogCaptureFixture):
-    with patch("bitcoin_safe.logging_handlers.compose_email") as mock_compose_email:
-        mock_compose_email.return_value = "Mocked Function"
+    with patch("bitcoin_safe.logging_handlers.OpenLogHandler.emit") as mock_OpenLogHandler_emit:
+        mock_OpenLogHandler_emit.return_value = "Mocked OpenLogHandler.emit"
+        with patch("bitcoin_safe.logging_handlers.compose_email") as mock_compose_email:
+            mock_compose_email.return_value = "Mocked Function"
 
-        with caplog.at_level(logging.INFO):
-            logger.critical("this should compose an email", exc_info=sys.exc_info())
+            with caplog.at_level(logging.INFO):
+                logger.critical("this should compose an email", exc_info=sys.exc_info())
 
-            # Assert that the mocked function was called
-            mock_compose_email.assert_called_once()
+                # Assert that the mocked function was called
+                mock_compose_email.assert_called_once()
+                # Assert that the mocked function was called
+                mock_OpenLogHandler_emit.assert_called_once()