Roog (顾新培)
b1077e78e9
变更内容： - 移除冗余文档，包括 Grafana 指南、指标对比、修复总结、OpenAPI 规范等。 - 精简项目文档结构，优化 README 文件内容。 - 提升文档层次清晰度，集中核心指南。
108 lines
3.3 KiB
Python
108 lines
3.3 KiB
Python
"""API 集成测试"""
|
|
|
|
import pytest
|
|
from fastapi import status
|
|
|
|
|
|
class TestInvokeEndpoint:
|
|
"""测试 /invoke 端点"""
|
|
|
|
def test_invoke_prime_number(self, client):
|
|
"""测试质数判断"""
|
|
response = client.post("/invoke", json={"number": 17})
|
|
|
|
assert response.status_code == status.HTTP_200_OK
|
|
data = response.json()
|
|
|
|
assert "request_id" in data
|
|
assert data["status"] == "success"
|
|
assert data["result"]["number"] == 17
|
|
assert data["result"]["is_prime"] is True
|
|
assert data["result"]["factors"] == []
|
|
|
|
def test_invoke_composite_number(self, client):
|
|
"""测试合数判断"""
|
|
response = client.post("/invoke", json={"number": 12})
|
|
|
|
assert response.status_code == status.HTTP_200_OK
|
|
data = response.json()
|
|
|
|
assert data["status"] == "success"
|
|
assert data["result"]["number"] == 12
|
|
assert data["result"]["is_prime"] is False
|
|
assert len(data["result"]["factors"]) > 0
|
|
|
|
def test_invoke_edge_cases(self, client):
|
|
"""测试边界情况"""
|
|
# 测试 0
|
|
response = client.post("/invoke", json={"number": 0})
|
|
assert response.status_code == status.HTTP_200_OK
|
|
assert response.json()["result"]["is_prime"] is False
|
|
|
|
# 测试 1
|
|
response = client.post("/invoke", json={"number": 1})
|
|
assert response.status_code == status.HTTP_200_OK
|
|
assert response.json()["result"]["is_prime"] is False
|
|
|
|
# 测试 2
|
|
response = client.post("/invoke", json={"number": 2})
|
|
assert response.status_code == status.HTTP_200_OK
|
|
assert response.json()["result"]["is_prime"] is True
|
|
|
|
def test_invoke_invalid_input(self, client):
|
|
"""测试无效输入"""
|
|
# 缺少必需字段
|
|
response = client.post("/invoke", json={})
|
|
assert response.status_code == status.HTTP_422_UNPROCESSABLE_ENTITY
|
|
|
|
# 错误的数据类型
|
|
response = client.post("/invoke", json={"number": "not a number"})
|
|
assert response.status_code == status.HTTP_422_UNPROCESSABLE_ENTITY
|
|
|
|
# 浮点数
|
|
response = client.post("/invoke", json={"number": 3.14})
|
|
assert response.status_code == status.HTTP_422_UNPROCESSABLE_ENTITY
|
|
|
|
|
|
class TestHealthEndpoints:
|
|
"""测试健康检查端点"""
|
|
|
|
def test_healthz(self, client):
|
|
"""测试存活检查"""
|
|
response = client.get("/healthz")
|
|
|
|
assert response.status_code == status.HTTP_200_OK
|
|
data = response.json()
|
|
|
|
assert data["status"] == "healthy"
|
|
assert "timestamp" in data
|
|
|
|
def test_readyz(self, client):
|
|
"""测试就绪检查"""
|
|
response = client.get("/readyz")
|
|
|
|
assert response.status_code == status.HTTP_200_OK
|
|
data = response.json()
|
|
|
|
assert data["status"] == "ready"
|
|
assert "timestamp" in data
|
|
assert "checks" in data
|
|
|
|
|
|
class TestMetricsEndpoint:
|
|
"""测试指标端点"""
|
|
|
|
def test_metrics(self, client):
|
|
"""测试 Prometheus 指标"""
|
|
response = client.get("/metrics")
|
|
|
|
assert response.status_code == status.HTTP_200_OK
|
|
assert "text/plain" in response.headers["content-type"]
|
|
|
|
|
|
class TestJobsEndpoint:
|
|
"""测试异步任务端点"""
|
|
|
|
# 详细测试在 test_job_manager.py 中
|
|
pass
|