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_linkedin_service.py
+++ b/backend/test/test_linkedin_service.py
@@ -0,0 +1,105 @@
+#!/usr/bin/env python3
+"""
+Test script for LinkedIn service functionality.
+
+This script tests that the LinkedIn service can be initialized and
+basic functionality works without errors.
+
+Usage:
+    python test_linkedin_service.py
+"""
+
+import asyncio
+import sys
+import os
+from pathlib import Path
+
+# Add the backend directory to the Python path
+backend_dir = Path(__file__).parent
+sys.path.insert(0, str(backend_dir))
+
+from loguru import logger
+from models.linkedin_models import LinkedInPostRequest, GroundingLevel
+from services.linkedin_service import LinkedInService
+
+
+async def test_linkedin_service():
+    """Test the LinkedIn service functionality."""
+    try:
+        logger.info("🧪 Testing LinkedIn Service Functionality")
+        
+        # Initialize the service
+        logger.info("📦 Initializing LinkedIn Service...")
+        service = LinkedInService()
+        logger.info("✅ LinkedIn Service initialized successfully")
+        
+        # Create a test request
+        test_request = LinkedInPostRequest(
+            topic="AI in Marketing",
+            industry="Technology",
+            tone="professional",
+            max_length=500,
+            target_audience="Marketing professionals",
+            key_points=["AI automation", "Personalization", "ROI improvement"],
+            research_enabled=True,
+            search_engine="google",
+            grounding_level=GroundingLevel.BASIC,
+            include_citations=True
+        )
+        
+        logger.info("📝 Testing LinkedIn Post Generation...")
+        
+        # Test post generation
+        response = await service.generate_linkedin_post(test_request)
+        
+        if response.success:
+            logger.info("✅ LinkedIn post generation successful")
+            logger.info(f"📊 Content length: {len(response.data.content)} characters")
+            logger.info(f"🔗 Sources: {len(response.research_sources)}")
+            logger.info(f"📚 Citations: {len(response.data.citations)}")
+            logger.info(f"🏆 Quality score: {response.data.quality_metrics.overall_score if response.data.quality_metrics else 'N/A'}")
+            
+            # Display a snippet of the generated content
+            content_preview = response.data.content[:200] + "..." if len(response.data.content) > 200 else response.data.content
+            logger.info(f"📄 Content preview: {content_preview}")
+            
+        else:
+            logger.error(f"❌ LinkedIn post generation failed: {response.error}")
+            return False
+        
+        logger.info("🎉 LinkedIn service test completed successfully!")
+        return True
+        
+    except Exception as e:
+        logger.error(f"❌ LinkedIn service test failed: {str(e)}")
+        return False
+
+
+async def main():
+    """Main test function."""
+    logger.info("🚀 Starting LinkedIn Service Test")
+    logger.info("=" * 50)
+    
+    success = await test_linkedin_service()
+    
+    if success:
+        logger.info("\n🎉 SUCCESS: LinkedIn service is working correctly!")
+        logger.info("✅ Service initialization successful")
+        logger.info("✅ Post generation working")
+        logger.info("✅ Ready for production use")
+    else:
+        logger.error("\n❌ FAILURE: LinkedIn service test failed")
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    # Configure logging
+    logger.remove()
+    logger.add(
+        sys.stderr,
+        format="<green>{time:YYYY-MM-DD HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>",
+        level="INFO"
+    )
+    
+    # Run the test
+    asyncio.run(main())