mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-06-14 15:25:17 +02:00
modify by comment
This commit is contained in:
parent
4a991b4cd4
commit
3369c9e536
3 changed files with 14 additions and 15 deletions
|
|
@ -1,5 +1,4 @@
|
|||
import os
|
||||
from unittest.mock import Mock, patch
|
||||
|
||||
import pytest
|
||||
|
||||
|
|
@ -16,14 +15,13 @@ incorrect_email_password = "incorrect_password"
|
|||
|
||||
|
||||
@pytest.fixture
|
||||
def imap_server_setup():
|
||||
# Use patch to mock the behavior of MailBox from the correct module path
|
||||
with patch("metagpt.tools.libs.email_login.MailBox") as mock_mailbox:
|
||||
# Setup for successful login
|
||||
mock_mail_instance = Mock()
|
||||
mock_mail_instance.login.return_value = mock_mail_instance
|
||||
mock_mailbox.return_value = mock_mail_instance
|
||||
yield mock_mail_instance
|
||||
def imap_server_setup(mocker):
|
||||
# Use the mocker fixture to mock the MailBox class
|
||||
mock_mailbox = mocker.patch("metagpt.tools.libs.email_login.MailBox")
|
||||
mock_mail_instance = mocker.Mock()
|
||||
mock_mail_instance.login.return_value = mock_mail_instance
|
||||
mock_mailbox.return_value = mock_mail_instance
|
||||
return mock_mail_instance
|
||||
|
||||
|
||||
def test_email_login_imap_success(imap_server_setup):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue