"""仪表盘 API 测试。"""

import pytest
from fastapi.testclient import TestClient


@pytest.fixture
def auth_headers(client: TestClient):
    """注册并登录。"""
    client.post(
        "/api/v1/auth/register",
        json={
            "email": "dashboard_test@example.com",
            "password": "password123",
            "name": "测试投资经理",
            "tenant_name": "测试机构",
            "role": "investor",
        },
    )
    resp = client.post(
        "/api/v1/auth/login",
        json={"email": "dashboard_test@example.com", "password": "password123"},
    )
    token = resp.json()["data"]["access_token"]
    return {"Authorization": f"Bearer {token}"}


class TestDashboardSummary:
    """仪表盘汇总测试。"""

    def test_summary_empty(self, client: TestClient, auth_headers: dict):
        """空租户的汇总数据。"""
        response = client.get("/api/v1/dashboard/summary", headers=auth_headers)
        assert response.status_code == 200
        data = response.json()
        assert data["code"] == 0
        assert data["data"]["total_companies"] == 0
        assert data["data"]["avg_health_score"] == 0.0
        assert data["data"]["high_risk_count"] == 0
        assert data["data"]["pending_reports"] == 0

    def test_summary_without_auth(self, client: TestClient):
        """无认证应返回 401。"""
        response = client.get("/api/v1/dashboard/summary")
        assert response.status_code == 401

    def test_summary_with_data(self, client: TestClient, auth_headers: dict):
        """有企业数据后的汇总。"""
        # 创建企业
        client.post(
            "/api/v1/companies",
            json={"name": "仪表盘测试公司", "industry": "AI"},
            headers=auth_headers,
        )

        response = client.get("/api/v1/dashboard/summary", headers=auth_headers)
        assert response.status_code == 200
        data = response.json()
        assert data["data"]["total_companies"] >= 1


class TestHealthScores:
    """健康度评分列表测试。"""

    def test_list_scores_empty(self, client: TestClient, auth_headers: dict):
        """空评分列表。"""
        response = client.get("/api/v1/dashboard/scores", headers=auth_headers)
        assert response.status_code == 200
        data = response.json()
        assert data["code"] == 0
        assert isinstance(data["data"], list)