"""T4.7 Pre-mortem 失败推演 + Red Team 对抗分析测试。

测试 AI Pre-mortem Agent 和 Red Team Agent。
"""

import pytest

from app.services.pre_mortem_agent import run_pre_mortem
from app.services.red_team_agent import run_red_team


class TestRunPreMortem:
    """Pre-mortem 失败推演测试。"""

    async def test_returns_dict(self):
        """返回失败推演结果字典。"""
        result = await run_pre_mortem("决定进入企业级 AI 市场，预计 6 个月内获取 50 家 B 端客户")
        assert isinstance(result, dict)

    async def test_empty_input(self):
        """空输入返回字典。"""
        result = await run_pre_mortem("")
        assert isinstance(result, dict)


class TestRunRedTeam:
    """Red Team 对抗分析测试。"""

    async def test_returns_dict(self):
        """返回对抗分析结果字典。"""
        result = await run_red_team("AI SaaS 公司，年营收 2000 万，客户 50 家", "competitor")
        assert isinstance(result, dict)

    async def test_different_perspective(self):
        """不同视角分析。"""
        result = await run_red_team("AI SaaS 公司，估值 5 亿", "pessimistic_investor")
        assert isinstance(result, dict)

    async def test_empty_input(self):
        """空输入返回字典。"""
        result = await run_red_team("", "")
        assert isinstance(result, dict)