112 lines
4.4 KiB
Python
112 lines
4.4 KiB
Python
"""
|
||
基础单元测试 —— 测试 TodoService 的核心功能。
|
||
|
||
运行方式：
|
||
cd codegen-runs/codegen_a2cef5070965479b93c5b3c07d4c8216
|
||
pytest tests/ -v
|
||
"""
|
||
|
||
import pytest
|
||
from app.services import TodoService, TodoCreate, TodoUpdate
|
||
|
||
|
||
@pytest.fixture
|
||
def service() -> TodoService:
|
||
"""每个测试用例使用独立的 TodoService 实例。"""
|
||
return TodoService()
|
||
|
||
|
||
class TestTodoService:
|
||
"""TodoService 核心功能测试。"""
|
||
|
||
def test_list_todos_should_return_mock_data(self, service: TodoService):
|
||
"""测试初始化时自动生成假数据。"""
|
||
todos = service.list_todos()
|
||
assert len(todos) == 5 # 初始化了 5 条假数据
|
||
|
||
def test_create_todo_should_return_created_todo(self, service: TodoService):
|
||
"""测试创建待办事项。"""
|
||
todo_in = TodoCreate(title="新任务", completed=True)
|
||
todo = service.create_todo(todo_in)
|
||
assert todo.title == "新任务"
|
||
assert todo.completed is True
|
||
assert todo.id > 0
|
||
|
||
def test_get_todo_should_return_correct_todo(self, service: TodoService):
|
||
"""测试根据 ID 获取待办事项。"""
|
||
# 先创建
|
||
created = service.create_todo(TodoCreate(title="测试任务"))
|
||
# 再获取
|
||
fetched = service.get_todo(created.id)
|
||
assert fetched is not None
|
||
assert fetched.id == created.id
|
||
assert fetched.title == "测试任务"
|
||
|
||
def test_get_todo_should_return_none_when_not_found(self, service: TodoService):
|
||
"""测试获取不存在的待办事项返回 None。"""
|
||
todo = service.get_todo(99999)
|
||
assert todo is None
|
||
|
||
def test_update_todo_should_update_title(self, service: TodoService):
|
||
"""测试部分更新待办事项的标题。"""
|
||
created = service.create_todo(TodoCreate(title="旧标题"))
|
||
updated = service.update_todo(created.id, TodoUpdate(title="新标题"))
|
||
assert updated is not None
|
||
assert updated.title == "新标题"
|
||
assert updated.completed is False # 未更新的字段保持不变
|
||
|
||
def test_update_todo_should_update_completed(self, service: TodoService):
|
||
"""测试部分更新待办事项的完成状态。"""
|
||
created = service.create_todo(TodoCreate(title="任务", completed=False))
|
||
updated = service.update_todo(created.id, TodoUpdate(completed=True))
|
||
assert updated is not None
|
||
assert updated.completed is True
|
||
assert updated.title == "任务" # 未更新的字段保持不变
|
||
|
||
def test_update_todo_should_return_none_when_not_found(self, service: TodoService):
|
||
"""测试更新不存在的待办事项返回 None。"""
|
||
result = service.update_todo(99999, TodoUpdate(title="无"))
|
||
assert result is None
|
||
|
||
def test_delete_todo_should_return_true(self, service: TodoService):
|
||
"""测试删除存在的待办事项返回 True。"""
|
||
created = service.create_todo(TodoCreate(title="待删除"))
|
||
result = service.delete_todo(created.id)
|
||
assert result is True
|
||
# 确认已删除
|
||
assert service.get_todo(created.id) is None
|
||
|
||
def test_delete_todo_should_return_false_when_not_found(self, service: TodoService):
|
||
"""测试删除不存在的待办事项返回 False。"""
|
||
result = service.delete_todo(99999)
|
||
assert result is False
|
||
|
||
def test_list_todos_after_create_and_delete(self, service: TodoService):
|
||
"""测试增删操作后列表长度的变化。"""
|
||
initial_count = len(service.list_todos())
|
||
# 创建 2 个
|
||
t1 = service.create_todo(TodoCreate(title="A"))
|
||
t2 = service.create_todo(TodoCreate(title="B"))
|
||
assert len(service.list_todos()) == initial_count + 2
|
||
# 删除 1 个
|
||
service.delete_todo(t1.id)
|
||
assert len(service.list_todos()) == initial_count + 1
|
||
# 删除另一个
|
||
service.delete_todo(t2.id)
|
||
assert len(service.list_todos()) == initial_count
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_fastapi_health_check():
|
||
"""简单的 FastAPI 健康检查测试（使用 TestClient）。"""
|
||
from httpx import AsyncClient, ASGITransport
|
||
from app.main import app
|
||
|
||
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 "service" in data
|
||
assert data["service"] == "Todo 管理服务"
|