Base code

backend/test/test_calendar_generation.py

@@ -0,0 +1,66 @@
+#!/usr/bin/env python3
+"""
+Test script for calendar generation API
+"""
+
+import asyncio
+import aiohttp
+import json
+
+async def test_calendar_generation():
+    """Test the calendar generation API."""
+    
+    url = "http://localhost:8000/api/content-planning/calendar-generation/start"
+    
+    payload = {
+        "user_id": 1,
+        "strategy_id": 1,
+        "calendar_type": "monthly",
+        "industry": "technology",
+        "business_size": "sme"
+    }
+    
+    async with aiohttp.ClientSession() as session:
+        try:
+            async with session.post(url, json=payload) as response:
+                if response.status == 200:
+                    result = await response.json()
+                    print("✅ Calendar generation started successfully!")
+                    print(f"Session ID: {result.get('session_id')}")
+                    
+                    # Test progress endpoint
+                    session_id = result.get('session_id')
+                    if session_id:
+                        print(f"\n🔄 Testing progress for session: {session_id}")
+                        progress_url = f"http://localhost:8000/api/content-planning/calendar-generation/progress/{session_id}"
+                        
+                        async with session.get(progress_url) as progress_response:
+                            if progress_response.status == 200:
+                                progress_data = await progress_response.json()
+                                print("✅ Progress endpoint working!")
+                                print(f"Status: {progress_data.get('status')}")
+                                print(f"Current Step: {progress_data.get('current_step')}")
+                                print(f"Overall Progress: {progress_data.get('overall_progress')}%")
+                                
+                                # Check for Step 4 specifically
+                                step_results = progress_data.get('step_results', {})
+                                if 'step_04' in step_results:
+                                    step4_result = step_results['step_04']
+                                    print(f"\n📊 Step 4 Status: {step4_result.get('status')}")
+                                    print(f"Step 4 Quality: {step4_result.get('quality_score')}")
+                                    if step4_result.get('status') == 'error':
+                                        print(f"Step 4 Error: {step4_result.get('error_message')}")
+                                else:
+                                    print("⚠️ Step 4 results not yet available")
+                            else:
+                                print(f"❌ Progress endpoint failed: {progress_response.status}")
+                else:
+                    print(f"❌ Calendar generation failed: {response.status}")
+                    error_text = await response.text()
+                    print(f"Error: {error_text}")
+                    
+        except Exception as e:
+            print(f"❌ Error testing calendar generation: {e}")
+
+if __name__ == "__main__":
+    asyncio.run(test_calendar_generation())