zhaoxi
4aa1dab283
- control_node 标注 DEPRECATED：保留目录壳子供未来远程探针节点复用，删除调用路径与相关测试
- 新增 task 表：极简元数据持久化 regulatory_node 完成的短任务（出报告/写文件/查询整理）
- regulatory_node 自标注：MessageResponse 扩展 task_action/title/summary，_run 末尾非阻塞落库
- query_task_list 改查 task 表，符合用户对"任务列表"的直觉，与 workflow 体系解耦
- 新增 /api/v1/task/list|/{id} 只读 API（task 由 regulatory 内部触发，不开放对外创建）
Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
98 lines
2.9 KiB
Python
98 lines
2.9 KiB
Python
"""``TaskDatabase`` 单元测试：覆盖 create / get / list / update_status 路径。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
from unittest.mock import AsyncMock, MagicMock
|
|
|
|
from kilostar.core.postgres_database.module.task import TaskDatabase
|
|
|
|
|
|
def _make_db():
|
|
session = AsyncMock()
|
|
session.__aenter__ = AsyncMock(return_value=session)
|
|
session.__aexit__ = AsyncMock(return_value=False)
|
|
session_maker = MagicMock(return_value=session)
|
|
return TaskDatabase(session_maker), session
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_create_task_persists_row():
|
|
db, session = _make_db()
|
|
session.add = MagicMock()
|
|
session.commit = AsyncMock()
|
|
|
|
await db.create_task(
|
|
task_id="t1",
|
|
user_id="alice",
|
|
command="写一份周报",
|
|
title="Q2 周报",
|
|
chat_id="chat-1",
|
|
status="completed",
|
|
result_summary="已生成报告",
|
|
)
|
|
session.add.assert_called_once()
|
|
added = session.add.call_args[0][0]
|
|
assert added.task_id == "t1"
|
|
assert added.user_id == "alice"
|
|
assert added.title == "Q2 周报"
|
|
assert added.status == "completed"
|
|
session.commit.assert_awaited_once()
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_get_task_returns_none_when_missing():
|
|
db, session = _make_db()
|
|
execute_result = MagicMock()
|
|
execute_result.scalar_one_or_none.return_value = None
|
|
session.execute = AsyncMock(return_value=execute_result)
|
|
|
|
result = await db.get_task("missing")
|
|
assert result is None
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_list_tasks_by_user_filters_status():
|
|
"""传 status 时 SQL 应进入 status 过滤分支（execute 被调用一次即视为路径已走通）。"""
|
|
db, session = _make_db()
|
|
execute_result = MagicMock()
|
|
execute_result.scalars.return_value.all.return_value = []
|
|
session.execute = AsyncMock(return_value=execute_result)
|
|
|
|
result = await db.list_tasks_by_user(user_id="alice", status="completed", limit=10)
|
|
assert result == []
|
|
session.execute.assert_awaited_once()
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_list_tasks_by_user_no_status():
|
|
db, session = _make_db()
|
|
execute_result = MagicMock()
|
|
execute_result.scalars.return_value.all.return_value = []
|
|
session.execute = AsyncMock(return_value=execute_result)
|
|
|
|
await db.list_tasks_by_user(user_id="alice")
|
|
session.execute.assert_awaited_once()
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_update_status_with_summary():
|
|
db, session = _make_db()
|
|
session.execute = AsyncMock()
|
|
session.commit = AsyncMock()
|
|
|
|
await db.update_status("t1", status="failed", result_summary="出错")
|
|
session.execute.assert_awaited_once()
|
|
session.commit.assert_awaited_once()
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_update_status_without_summary():
|
|
db, session = _make_db()
|
|
session.execute = AsyncMock()
|
|
session.commit = AsyncMock()
|
|
|
|
await db.update_status("t1", status="running")
|
|
session.execute.assert_awaited_once()
|
|
session.commit.assert_awaited_once()
|