"""
API端到端测试
"""

import pytest
import requests


API_BASE = "http://localhost:7865"


@pytest.mark.e2e
class TestAPIEndpoints:
    """测试API端点"""
    
    def test_api_info(self):
        """测试API信息端点"""
        response = requests.get(f"{API_BASE}/api/info", timeout=10)
        
        assert response.status_code == 200
        data = response.json()
        assert "job_id" in data
        assert "working_dir" in data
    
    @pytest.mark.parametrize("query,expected_mode", [
        ("帮我分析拟南芥SPL基因家族", "analysis"),
        ("什么是SPL基因家族？", "faq"),
        ("推荐用什么参数做HMMER搜索？", "consultation"),
    ])
    def test_chat_intent_recognition(self, query, expected_mode):
        """测试聊天意图识别"""
        response = requests.post(
            f"{API_BASE}/api/agent/chat",
            json={
                "message": query,
                "session_id": "e2e_test_session"
            },
            timeout=30
        )
        
        assert response.status_code == 200
        data = response.json()
        
        # 检查响应结构
        assert "intent" in data or "response" in data
    
    def test_plan_endpoint(self):
        """测试规划端点"""
        response = requests.post(
            f"{API_BASE}/api/agent/plan",
            json={
                "message": "分析水稻SPL基因家族",
                "gene_family": "SPL",
                "species": "rice",
                "session_id": "e2e_test_plan"
            },
            timeout=30
        )
        
        assert response.status_code == 200
        data = response.json()
        
        # 检查是否有规划结果
        if "plan" in data:
            plan = data["plan"]
            assert "stages" in plan
    
    def test_tasks_endpoint(self):
        """测试任务列表端点"""
        response = requests.get(f"{API_BASE}/api/agent/tasks", timeout=10)
        
        assert response.status_code == 200
        # 任务列表可以是空数组
        data = response.json()
        assert isinstance(data, list) or isinstance(data, dict)


@pytest.mark.e2e
class TestServiceHealth:
    """测试服务健康状态"""
    
    def test_service_running(self):
        """测试服务是否运行"""
        try:
            response = requests.get(f"{API_BASE}/api/info", timeout=5)
            assert response.status_code == 200
        except requests.exceptions.ConnectionError:
            pytest.skip("服务未运行，跳过E2E测试")
    
    def test_response_time(self):
        """测试响应时间"""
        import time
        
        start = time.time()
        response = requests.get(f"{API_BASE}/api/info", timeout=10)
        elapsed = time.time() - start
        
        assert response.status_code == 200
        assert elapsed < 2.0  # 响应时间应小于2秒