Files
AIPortPilot/backend/tests/test_dashboard.py
T
selfrelease fad458b2a7 docs(uiux): UIUX 设计方案大改 + 5 份作业指导书对齐 + 开发任务文档
- UIUX 文档:填充 19 个缺口(多主体画像/健康度/AI+看板/增长域/洞察域/创始人端/OODA/助推/商密)
- UIUX 文档:插入 6 个新章节(十四~十九),旧章节重编号为二十~三十一,更新目录和交叉引用
- 作业指导书 x5:导航改为 6 域分组,新增 Context Bar/工作模式/Insight Rail/决策线程/多工作区等 UI 概念
- 新建 docs/2-task-uiux.md:50 个代码落地开发任务,按 P0-P6 分优先级 + 8 Sprint 规划
- 后端/前端:大量新增模型、路由、组件(来自之前 Phase 开发)
2026-07-19 11:53:38 +08:00

72 lines
2.4 KiB
Python

"""仪表盘 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)