progress
This commit is contained in:
@@ -21,7 +21,6 @@ class TestIMAPService:
|
||||
assert imap_service.user == user
|
||||
assert imap_service.config == {}
|
||||
|
||||
@patch('app.imap_service.imaplib.IMAP4_SSL')
|
||||
def test_test_connection_success(self, mock_imap, app):
|
||||
with app.app_context():
|
||||
# Mock successful connection
|
||||
@@ -33,17 +32,14 @@ class TestIMAPService:
|
||||
mock_connection.logout.return_value = None
|
||||
|
||||
user = User(email='test@example.com', first_name='Test', last_name='User')
|
||||
user.imap_config = {'server': 'test.com', 'port': 993, 'username': 'user', 'password': 'pass'}
|
||||
user.imap_config = {'server': 'localhost', 'port': 5143, 'username': 'user1@example.com', 'password': 'password1', 'use_ssl': False}
|
||||
imap_service = IMAPService(user)
|
||||
|
||||
success, message = imap_service.test_connection()
|
||||
|
||||
assert success is True
|
||||
assert message == "Connection successful"
|
||||
mock_imap.assert_called_once_with('test.com', 993)
|
||||
mock_connection.login.assert_called_once_with('user', 'pass')
|
||||
assert success is True
|
||||
|
||||
@patch('app.imap_service.imaplib.IMAP4_SSL')
|
||||
def test_test_connection_failure(self, mock_imap, app):
|
||||
with app.app_context():
|
||||
# Mock failed connection
|
||||
@@ -153,68 +149,43 @@ class TestIMAPService:
|
||||
|
||||
assert len(folders) == 0
|
||||
|
||||
@patch('app.imap_service.imaplib.IMAP4_SSL')
|
||||
def test_sync_folders_success(self, mock_imap, app):
|
||||
with app.app_context():
|
||||
# Mock successful folder listing and database operations
|
||||
mock_connection = Mock()
|
||||
mock_imap.return_value = mock_connection
|
||||
mock_connection.login.return_value = None
|
||||
mock_connection.list.return_value = ('OK', [b'(\\HasNoChildren) "INBOX"', b'(\\HasNoChildren) "CustomFolder"', b'(\\HasNoChildren) "AnotherFolder"'])
|
||||
mock_connection.close.return_value = None
|
||||
mock_connection.logout.return_value = None
|
||||
|
||||
user = User(email='test@example.com', first_name='Test', last_name='User')
|
||||
user.imap_config = {'server': 'test.com', 'port': 993, 'username': 'user', 'password': 'pass'}
|
||||
imap_service = IMAPService(user)
|
||||
|
||||
success, message = imap_service.sync_folders()
|
||||
|
||||
assert success is True
|
||||
assert "Successfully synced 2 folders" in message
|
||||
def test_sync_folders_success(self, app):
|
||||
user = User(email='test@example.com', first_name='Test', last_name='User')
|
||||
user.imap_config = {'server': 'localhost', 'port': 5143, 'username': 'user1@example.com', 'password': 'password1'}
|
||||
imap_service = IMAPService(user)
|
||||
|
||||
success, message = imap_service.sync_folders()
|
||||
|
||||
assert success is True
|
||||
assert "Successfully synced 2 folders" in message
|
||||
|
||||
@patch('app.imap_service.imaplib.IMAP4_SSL')
|
||||
def test_sync_folders_no_config(self, mock_imap, app):
|
||||
with app.app_context():
|
||||
user = User(email='test@example.com', first_name='Test', last_name='User')
|
||||
imap_service = IMAPService(user)
|
||||
|
||||
success, message = imap_service.sync_folders()
|
||||
|
||||
assert success is False
|
||||
assert message == "No IMAP configuration found"
|
||||
user = User(email='test@example.com', first_name='Test', last_name='User')
|
||||
imap_service = IMAPService(user)
|
||||
|
||||
success, message = imap_service.sync_folders()
|
||||
|
||||
assert success is False
|
||||
assert message == "No IMAP configuration found"
|
||||
|
||||
@patch('app.imap_service.imaplib.IMAP4_SSL')
|
||||
def test_sync_folders_no_folders(self, mock_imap, app):
|
||||
with app.app_context():
|
||||
# Mock empty folder listing
|
||||
mock_connection = Mock()
|
||||
mock_imap.return_value = mock_connection
|
||||
mock_connection.login.return_value = None
|
||||
mock_connection.list.return_value = ('OK', [])
|
||||
mock_connection.close.return_value = None
|
||||
mock_connection.logout.return_value = None
|
||||
|
||||
user = User(email='test@example.com', first_name='Test', last_name='User')
|
||||
user.imap_config = {'server': 'test.com', 'port': 993, 'username': 'user', 'password': 'pass'}
|
||||
imap_service = IMAPService(user)
|
||||
|
||||
success, message = imap_service.sync_folders()
|
||||
|
||||
assert success is False
|
||||
assert message == "No folders found on IMAP server"
|
||||
|
||||
user = User(email='test@example.com', first_name='Test', last_name='User')
|
||||
user.imap_config = {'server': 'localhost', 'port': 5143, 'username': 'user1@example.com', 'password': 'password1', 'use_ssl': False}
|
||||
imap_service = IMAPService(user)
|
||||
|
||||
success, message = imap_service.sync_folders()
|
||||
|
||||
assert success is False
|
||||
assert message == "No folders found on IMAP server"
|
||||
|
||||
@patch('app.imap_service.imaplib.IMAP4_SSL')
|
||||
def test_sync_folders_exception(self, mock_imap, app):
|
||||
with app.app_context():
|
||||
# Mock exception during sync
|
||||
mock_imap.side_effect = Exception("Sync error")
|
||||
def test_sync_folders_exception(self, app):
|
||||
|
||||
user = User(email='test@example.com', first_name='Test', last_name='User')
|
||||
user.imap_config = {'server': 'test.com', 'port': 993, 'username': 'user', 'password': 'pass'}
|
||||
imap_service = IMAPService(user)
|
||||
|
||||
success, message = imap_service.sync_folders()
|
||||
|
||||
assert success is False
|
||||
assert "Sync error" in message
|
||||
user = User(email='test@example.com', first_name='Test', last_name='User')
|
||||
user.imap_config = {'server': 'localhost', 'port': 5143, 'username': 'user1@example.com', 'password': 'password1', 'use_ssl': False }
|
||||
imap_service = IMAPService(user)
|
||||
|
||||
success, message = imap_service.sync_folders()
|
||||
|
||||
assert success is False
|
||||
assert "Sync error" in message
|
||||
Reference in New Issue
Block a user