copywriting_production/tests/ai/test_parallel_analysis.py

from src.ai.ai_executor.parallel_runner import ParallelAnalysisRunner
from src.models.product_model import Product
from src.models.ai_execution_record import AnalysisType, ModelProvider
from src.models.config_model import UserConfig
import pytest
from src.manager.core.db_mongo import BaseMongoManager

@pytest.mark.asyncio
async def test_analysis():
    db_mongo = BaseMongoManager()  # 自动初始化
    await db_mongo.initialize()
    product = await Product.find_one(Product.basic_info.name == "电线保护套")
    user = await UserConfig.find_one(UserConfig.user_name == "test_user")
    competitor_prompt = user.prompting.get("竞品和长尾词分析")
    marketing_prompt = user.prompting.get("营销文案")
    runner = ParallelAnalysisRunner([ModelProvider.DEEPSEEK, ModelProvider.OPENAI])
    results = await runner.run_parallel(product, AnalysisType.COMPETITOR)