test_workflow.py

import pytest
from workflows.content_generator import generate_instagram_content
from config.schemas import InstagramPost

def test_content_generation():
    """Test content generation"""
    test_entry = {
        "title": "Test Article",
        "content": "This is a test article about AI and technology.",
        "entry_id": "test123"
    }
    
    result = generate_instagram_content(test_entry)
    
    # Validate structure
    assert "caption" in result
    assert "hashtags" in result
    assert "image_prompt" in result
    assert len(result["hashtags"]) >= 3

if __name__ == "__main__":
    pytest.main([__file__])