36 lines
1.5 KiB
Python
36 lines
1.5 KiB
Python
import pytest
|
|
from app import create_app
|
|
from app.models import User, db, Folder
|
|
|
|
class TestIMAPRoutes:
|
|
def test_imap_config_modal(self, authenticated_client):
|
|
response = authenticated_client.get('/api/imap/config')
|
|
assert response.status_code == 200
|
|
assert b'Configure IMAP Connection' in response.data
|
|
|
|
def test_imap_connection_test_success(self, authenticated_client, app):
|
|
response = authenticated_client.post('/api/imap/test', data={
|
|
'server': 'localhost',
|
|
'port': '5143',
|
|
'username': 'user1@example.com',
|
|
'password': 'password1',
|
|
'use_ssl': False
|
|
})
|
|
print(response.data)
|
|
|
|
assert response.status_code == 200
|
|
# Should show either success or error message
|
|
assert b'Test Connection' in response.data
|
|
|
|
def test_imap_sync_folders(self, authenticated_client, app, mock_user):
|
|
# Create a test user and log in
|
|
mock_user.imap_config = {'server': 'localhost', 'port': 5143, 'username': 'user1@example.com', 'password': 'password1', 'use_ssl': False}
|
|
db.session.commit()
|
|
folders = Folder.query.filter_by(user_id=mock_user.id).all()
|
|
response = authenticated_client.post('/api/imap/sync')
|
|
print('respo', response.data, response)
|
|
new_folders = Folder.query.filter_by(user_id=mock_user.id).all()
|
|
assert len(new_folders) > len(folders)
|
|
# Should fail without real IMAP server but return proper response
|
|
assert response.status_code in [200, 400]
|