From 01ff1ef091cb9bb3a8ba4c144061ccfe11c549ca Mon Sep 17 00:00:00 2001 From: shamoon <4887959+shamoon@users.noreply.github.com> Date: Sun, 6 Oct 2024 08:06:32 -0700 Subject: [PATCH] Ok maybe fix this last one --- src/paperless_mail/tests/test_mail.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/src/paperless_mail/tests/test_mail.py b/src/paperless_mail/tests/test_mail.py index c0749e861..061f736ca 100644 --- a/src/paperless_mail/tests/test_mail.py +++ b/src/paperless_mail/tests/test_mail.py @@ -1609,8 +1609,15 @@ class TestMailAccountTestView(APITestCase): @mock.patch("paperless_mail.mail.get_mailbox") @mock.patch("paperless_mail.mail.mailbox_login") def test_mail_account_test_view_success(self, mock_mailbox_login, mock_get_mailbox): - mock_get_mailbox.return_value.__enter__.return_value = mock.MagicMock() - mock_mailbox_login.return_value = True + def get_mailbox(server, port, security): + return mock.MagicMock() + + mock_get_mailbox.side_effect = get_mailbox + + def mock_mailbox_login(mailbox, account): + return True + + mock_mailbox_login.side_effect = mock_mailbox_login data = { "imap_server": "imap.example.com", "imap_port": 993,