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