cjw
0e6b3b1811
按 ROADMAP §3.1 治理批次推进的后端设计审查整改：
- 批次 0（安全）：/api/status/{task_id} 补 JWT 鉴权与任务归属校验；
pythonocc_available 真实探测；bcrypt 超 72 字节显式拒绝；
SECRET_KEY/RUSTFS_* 惰性校验，代码侧弱默认移除
- 批次 1（部署正确性）：主处理链路改走 RustFS（分派入参 stp_file_id 化，
worker 按 object_key 下载）；AUTO_MIGRATE 开关 + 迁移目录 alembic/→migrations/
修复包遮蔽（自动迁移此前从未真正生效）；OCC 镜像改 conda 原生执行 +
基础镜像 tag 锁定；compose 关键项改 ${VAR:?} 强制显式配置
- 批次 2（任务一致性）：删除 Redis 进程内存回退，PG 为任务状态单一事实源；
批量元数据入库（processing_tasks.batch_id，迁移 a3f8c2d91e47）；
型腔失败任务标 failed 不再静默 completed；事务边界收口
（数据本体写 flush-only、失败先回滚再置 failed、进度更新保留即时 commit）
- 批次 3（API 与代码结构）：592 行 advanced_router 拆为 design/cost/machining/
export 四子路由，请求体全量 Pydantic 化；ROUTE_MODULES + route_registry
（/api/health 呈现 degraded，DEBUG fail fast）；纯计算端点统一 to_thread；
StorageIntegrationService 按职责三拆；MAX_FILE_SIZE 接线生效、
celery 复用 Settings.redis_url；管理员重置密码改 JSON body（端到端断裂修复）；
openapi.json 重导出（76 paths）+ 前端 gen:api
- 批次 4（架构演进）：共享 ORM 按模块拆分（shared/models/base.py + identity.py、
moldinsight/models/、inventory/models/，删除三条无使用方的跨模块
relationship，跨模块桥接收敛为裸 FK 硬规则，无兼容 facade）；
OCC executor 重建补 cancel_futures=True（消除旧队列被慢恢复线程
并行消化的数据竞争）；OCC 吞吐方案设计先行
（docs/topics/performance/OCC_THROUGHPUT.md）；顺手清偿 D15
（vite.config.ts 未用参数致 npm run build 失败）
测试基线：125 passed, 2 skipped（pytest + sqlite+aiosqlite；归属边界、
路由契约、配置治理、鉴权回归等随批新增）
文档同步：STATUS / TECH_DEBT / ROADMAP / ARCHITECTURE / API_CONTRACT /
OPERATIONS / AGENTS
Co-Authored-By: Claude Code <noreply@anthropic.com>
58 lines
2.3 KiB
Python
58 lines
2.3 KiB
Python
"""批次 2（D7 / D9）回归测试。
|
||
|
||
- D7：Redis 任务管理器不再有进程内存回退——Redis 不可用时写 no-op、读返回 None，
|
||
状态查询路径落到 PG（PG 为单一事实源）
|
||
- D9：存储服务的数据本体写方法只 flush 不 commit，事务由编排层收口
|
||
"""
|
||
import pytest
|
||
from unittest.mock import AsyncMock, MagicMock
|
||
|
||
from shared.services.redis_task_manager import RedisTaskManager
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_no_memory_fallback_when_disconnected():
|
||
"""Redis 未连接：写 no-op、读 None——不得再出现进程内可见的副本。"""
|
||
mgr = RedisTaskManager() # 不 connect
|
||
assert not mgr.is_connected
|
||
|
||
await mgr.set_task("t1", {"status": "processing"})
|
||
assert await mgr.get_task("t1") is None
|
||
|
||
await mgr.update_task("t1", {"status": "completed"}) # no-op，不得抛异常
|
||
|
||
assert await mgr.get_all_tasks() == {}
|
||
assert await mgr.get_task_count() == 0
|
||
await mgr.delete_task("t1") # no-op，不得抛异常
|
||
|
||
assert await mgr.get_task("t1") is None
|
||
|
||
|
||
def test_fallback_storage_removed():
|
||
"""防回归：内存回退存储必须已删除，防止静默回归。"""
|
||
assert not hasattr(RedisTaskManager, "_fallback_tasks")
|
||
assert not hasattr(RedisTaskManager, "_fallback_set")
|
||
assert not hasattr(RedisTaskManager, "cleanup_old_tasks")
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_storage_writes_flush_but_never_commit():
|
||
"""D9：数据本体写方法仅 flush；commit 由编排层/请求侧负责。"""
|
||
pytest.importorskip("minio")
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
from moldinsight.services.task_storage_service import TaskStorageService
|
||
|
||
svc = TaskStorageService()
|
||
session = AsyncMock(spec=AsyncSession)
|
||
# update_task_parameters：select 返回 None（任务不存在）→ 直接 return
|
||
result_mock = MagicMock()
|
||
result_mock.scalar_one_or_none.return_value = None
|
||
session.execute.return_value = result_mock
|
||
|
||
await svc.update_task_parameters(session, "task-x", {"a": 1})
|
||
await svc.create_processing_task(session, "task-y", stp_file_id=1, batch_id="b-1")
|
||
|
||
assert session.flush.await_count >= 1
|
||
# 注意：不能用 .awaited（AsyncMock 上访问会自动创建 truthy 子 mock），用 await_count
|
||
assert session.commit.await_count == 0, "存储写方法不得自行 commit（D9 事务收口）"
|