Base code

2026-01-08 22:39:53 +07:00
parent 697115c61a
commit c35fa52117
2169 changed files with 626670 additions and 0 deletions
--- a/backend/test/test_step1_only.py
+++ b/backend/test/test_step1_only.py
@@ -0,0 +1,126 @@
+#!/usr/bin/env python3
+"""
+Simple Test Script for Step 1 Only
+
+This script tests only Step 1 to verify imports are working correctly.
+"""
+
+import asyncio
+import sys
+import os
+from typing import Dict, Any
+from loguru import logger
+
+# Add the backend directory to the path
+backend_dir = os.path.dirname(os.path.abspath(__file__))
+if backend_dir not in sys.path:
+    sys.path.insert(0, backend_dir)
+
+# Add the services directory to the path
+services_dir = os.path.join(backend_dir, "services")
+if services_dir not in sys.path:
+    sys.path.insert(0, services_dir)
+
+async def test_step1_only():
+    """Test only Step 1 to verify imports work."""
+    
+    try:
+        logger.info("🚀 Starting test of Step 1 only")
+        
+        # Test data
+        test_context = {
+            "user_id": 1,
+            "strategy_id": 1,
+            "calendar_duration": 7,
+            "posting_preferences": {
+                "posting_frequency": "daily",
+                "preferred_days": ["monday", "wednesday", "friday"],
+                "preferred_times": ["09:00", "12:00", "15:00"],
+                "content_per_day": 2
+            }
+        }
+        
+        # Test Step 1: Content Strategy Analysis
+        logger.info("📋 Testing Step 1: Content Strategy Analysis")
+        try:
+            from services.calendar_generation_datasource_framework.prompt_chaining.steps.phase1.phase1_steps import ContentStrategyAnalysisStep
+            from services.calendar_generation_datasource_framework.data_processing.strategy_data import StrategyDataProcessor
+            
+            logger.info("✅ Imports successful")
+            
+            # Create strategy processor with mock data for testing
+            strategy_processor = StrategyDataProcessor()
+            
+            # Mock strategy data
+            mock_strategy_data = {
+                "strategy_id": 1,
+                "strategy_name": "Test Strategy",
+                "industry": "technology",
+                "target_audience": {
+                    "primary": "Tech professionals",
+                    "secondary": "Business leaders",
+                    "demographics": {"age_range": "25-45", "location": "Global"}
+                },
+                "content_pillars": [
+                    "AI and Machine Learning",
+                    "Digital Transformation",
+                    "Innovation and Technology Trends",
+                    "Business Strategy and Growth"
+                ],
+                "business_objectives": [
+                    "Increase brand awareness by 40%",
+                    "Generate 500 qualified leads per month",
+                    "Establish thought leadership"
+                ],
+                "target_metrics": {"awareness": "website_traffic", "leads": "lead_generation"},
+                "quality_indicators": {"data_completeness": 0.8, "strategic_alignment": 0.9}
+            }
+            
+            # Mock the get_strategy_data method for testing
+            async def mock_get_strategy_data(strategy_id):
+                return mock_strategy_data
+            
+            strategy_processor.get_strategy_data = mock_get_strategy_data
+            
+            # Mock the validate_data method
+            async def mock_validate_data(data):
+                return {
+                    "quality_score": 0.85,
+                    "missing_fields": [],
+                    "recommendations": []
+                }
+            
+            strategy_processor.validate_data = mock_validate_data
+            
+            step1 = ContentStrategyAnalysisStep()
+            step1.strategy_processor = strategy_processor
+            
+            result1 = await step1.execute(test_context)
+            logger.info(f"✅ Step 1 completed: {result1.get('status')}")
+            logger.info(f"   Quality Score: {result1.get('quality_score')}")
+            
+        except Exception as e:
+            logger.error(f"❌ Step 1 failed: {str(e)}")
+            return False
+        
+        logger.info("🎉 Step 1 test completed successfully!")
+        return True
+        
+    except Exception as e:
+        logger.error(f"❌ Test failed with error: {str(e)}")
+        return False
+
+if __name__ == "__main__":
+    # Configure logging
+    logger.remove()
+    logger.add(sys.stderr, level="INFO", format="<green>{time:HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>")
+    
+    # Run the test
+    success = asyncio.run(test_step1_only())
+    
+    if success:
+        logger.info("✅ Test completed successfully!")
+        sys.exit(0)
+    else:
+        logger.error("❌ Test failed!")
+        sys.exit(1)