test(main,imap): fix failing IMAP tests and add coverage for shutdown exception paths

Co-authored-by: christianlouis <361235+christianlouis@users.noreply.github.com>
Agent-Logs-Url: https://github.com/christianlouis/DocuElevate/sessions/05be730d-fcbd-43a5-98be-26d853cf57d0
This commit is contained in:
copilot-swe-agent[bot]
2026-03-23 21:32:44 +00:00
parent 0b8f967eb5
commit c03ce8cdb2
2 changed files with 64 additions and 3 deletions
+12 -3
View File
@@ -524,7 +524,10 @@ class TestTestImapConnection:
from app.api.imap_accounts import _test_imap_connection
mock_mail = MagicMock()
with patch("imaplib.IMAP4_SSL", return_value=mock_mail):
with (
patch("app.api.imap_accounts.is_private_ip", return_value=False),
patch("imaplib.IMAP4_SSL", return_value=mock_mail),
):
result = _test_imap_connection(
"imap.example.com",
993,
@@ -541,7 +544,10 @@ class TestTestImapConnection:
"""An exception raised by IMAP4_SSL returns success=False."""
from app.api.imap_accounts import _test_imap_connection
with patch("imaplib.IMAP4_SSL", side_effect=Exception("auth failed")):
with (
patch("app.api.imap_accounts.is_private_ip", return_value=False),
patch("imaplib.IMAP4_SSL", side_effect=Exception("auth failed")),
):
result = _test_imap_connection(
"imap.example.com",
993,
@@ -557,7 +563,10 @@ class TestTestImapConnection:
"""An OSError returns success=False with a network error message."""
from app.api.imap_accounts import _test_imap_connection
with patch("imaplib.IMAP4", side_effect=OSError("connection refused")):
with (
patch("app.api.imap_accounts.is_private_ip", return_value=False),
patch("imaplib.IMAP4", side_effect=OSError("connection refused")),
):
result = _test_imap_connection(
"bad-host",
143,