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>
102 lines
3.4 KiB
Python
102 lines
3.4 KiB
Python
"""批次 3 回归：管理员重置密码走 JSON body。
|
|
|
|
此前后端把 new_password 声明为裸 str 参数（FastAPI 解析为 query param），
|
|
前端两个调用点均发送 JSON body，重置密码端到端断裂（必 422）。
|
|
现收敛为 Pydantic 请求模型 { new_password }，与 api-client.ts 结构一致。
|
|
"""
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
from fastapi import FastAPI
|
|
from httpx import AsyncClient, ASGITransport
|
|
from sqlalchemy.ext.asyncio import async_sessionmaker, AsyncSession
|
|
from sqlalchemy import select
|
|
|
|
from shared.database.database import get_db_session
|
|
from shared.models.identity import User
|
|
from shared.services.auth_service import (
|
|
get_current_active_user,
|
|
get_password_hash,
|
|
verify_password,
|
|
)
|
|
from shared.services.auth_routes import router as auth_router
|
|
|
|
|
|
@pytest.fixture
|
|
async def reset_env(async_engine, seeded_db):
|
|
"""播种被重置目标用户（id=2，已知旧密码）；返回 (client, app, session_factory)。"""
|
|
session_factory = async_sessionmaker(async_engine, class_=AsyncSession, expire_on_commit=False)
|
|
|
|
async with session_factory() as session:
|
|
target = User(
|
|
id=2, username="resetme", email="resetme@example.com",
|
|
hashed_password=get_password_hash("oldpass123"), is_active=True,
|
|
)
|
|
session.add(target)
|
|
await session.commit()
|
|
|
|
test_app = FastAPI()
|
|
test_app.include_router(auth_router) # router 自带 /api/auth 前缀
|
|
|
|
async def override_get_db_session():
|
|
async with session_factory() as session:
|
|
yield session
|
|
|
|
test_app.dependency_overrides[get_db_session] = override_get_db_session
|
|
|
|
transport = ASGITransport(app=test_app)
|
|
async with AsyncClient(transport=transport, base_url="http://test") as ac:
|
|
yield ac, test_app, session_factory
|
|
|
|
test_app.dependency_overrides.clear()
|
|
|
|
|
|
def _login_as(app: FastAPI, *, superuser: bool):
|
|
# User.is_superuser 为只读 hybrid property，覆写用户用 SimpleNamespace 承载
|
|
app.dependency_overrides[get_current_active_user] = lambda: SimpleNamespace(
|
|
id=500, username="admin", is_superuser=superuser
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_reset_password_with_json_body(reset_env):
|
|
"""JSON body { new_password } 生效：密码真实更新且可用新口令验证。"""
|
|
ac, app, session_factory = reset_env
|
|
_login_as(app, superuser=True)
|
|
|
|
resp = await ac.put(
|
|
"/api/auth/users/2/reset-password",
|
|
json={"new_password": "brandnew456"},
|
|
)
|
|
assert resp.status_code == 200
|
|
|
|
async with session_factory() as session:
|
|
user = (await session.execute(select(User).where(User.id == 2))).scalar_one()
|
|
assert verify_password("brandnew456", user.hashed_password)
|
|
assert not verify_password("oldpass123", user.hashed_password)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_reset_password_rejects_short_password(reset_env):
|
|
"""最短 6 位对齐前端校验，违约 422。"""
|
|
ac, app, _ = reset_env
|
|
_login_as(app, superuser=True)
|
|
|
|
resp = await ac.put(
|
|
"/api/auth/users/2/reset-password",
|
|
json={"new_password": "abc"},
|
|
)
|
|
assert resp.status_code == 422
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_reset_password_requires_admin(reset_env):
|
|
ac, app, _ = reset_env
|
|
_login_as(app, superuser=False)
|
|
|
|
resp = await ac.put(
|
|
"/api/auth/users/2/reset-password",
|
|
json={"new_password": "brandnew456"},
|
|
)
|
|
assert resp.status_code == 403
|