106 lines
3.6 KiB
Python
106 lines
3.6 KiB
Python
#!/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())
|