55 lines
1.9 KiB
Python
55 lines
1.9 KiB
Python
"""基础测试模块，测试核心业务逻辑和 API 端点。"""
|
|
|
|
import pytest
|
|
from httpx import AsyncClient, ASGITransport
|
|
from app.main import app
|
|
from app.services import get_greeting, get_system_info
|
|
|
|
|
|
class TestServices:
|
|
"""服务层单元测试。"""
|
|
|
|
def test_get_greeting_default(self):
|
|
"""测试无参数时返回默认问候语。"""
|
|
result = get_greeting()
|
|
assert result == "Hello, World!"
|
|
|
|
def test_get_greeting_with_name(self):
|
|
"""测试传入名字时返回个性化问候语。"""
|
|
result = get_greeting("FastAPI")
|
|
assert result == "Hello, FastAPI!"
|
|
|
|
def test_get_system_info_structure(self):
|
|
"""测试系统信息返回正确的字段结构。"""
|
|
info = get_system_info()
|
|
assert "message" in info
|
|
assert "timestamp" in info
|
|
assert "service" in info
|
|
assert "version" in info
|
|
assert info["message"] == "Hello, World!"
|
|
assert info["service"] == "Simple Hello API"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
class TestAPI:
|
|
"""API 端点集成测试。"""
|
|
|
|
async def test_root_endpoint(self):
|
|
"""测试根路径返回正确的系统信息。"""
|
|
transport = ASGITransport(app=app)
|
|
async with AsyncClient(transport=transport, base_url="http://test") as client:
|
|
response = await client.get("/")
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert data["message"] == "Hello, World!"
|
|
assert "timestamp" in data
|
|
|
|
async def test_hello_endpoint(self):
|
|
"""测试 /hello/{name} 返回正确的问候语。"""
|
|
transport = ASGITransport(app=app)
|
|
async with AsyncClient(transport=transport, base_url="http://test") as client:
|
|
response = await client.get("/hello/Python")
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert data["message"] == "Hello, Python!"
|