31 lines
1.0 KiB
Python
31 lines
1.0 KiB
Python
import pytest
|
|
from app.models import User
|
|
import uuid
|
|
|
|
def test_index_route(client, app):
|
|
"""Test that the index route loads successfully."""
|
|
with app.app_context():
|
|
# Create a mock user for the test
|
|
mock_user = User(
|
|
id=uuid.UUID('123e4567-e89b-12d3-a456-426614174000'),
|
|
email='test@example.com'
|
|
)
|
|
from app import db
|
|
db.session.add(mock_user)
|
|
db.session.commit()
|
|
|
|
response = client.get('/')
|
|
assert response.status_code == 200
|
|
# Check if the page contains expected elements
|
|
assert b'Email Organizer' in response.data
|
|
assert b'Folders' in response.data
|
|
|
|
def test_add_folder_route(client):
|
|
"""Test the add folder API endpoint."""
|
|
response = client.post('/api/folders',
|
|
json={'name': 'Test Folder', 'rule_text': 'Test rule'},
|
|
content_type='application/json')
|
|
|
|
assert response.status_code == 201
|
|
assert b'Folder added (mock)' in response.data
|
|
assert b'Test Folder' in response.data |