admin
afcddfaeb9
- 新增基于 JWT 当前用户的登录恢复、角色权限、用户管理、审计日志和演示出厂重置后台接口与前端管理页。 - 重串 GT_label 导出和 GT Mask 导入逻辑：导出保留类别真实 maskid，导入仅接受灰度或 RGB 等通道 maskid 图，支持未知 maskid 策略、尺寸最近邻拉伸和导入预览。 - 统一分割结果导出体验：默认当前帧，按项目抽帧顺序和 XhXXmXXsXXXms 时间戳命名 ZIP 与图片，补齐 GT/Pro/Mix/分开 Mask 输出和映射 JSON。 - 调整工作区左侧工具栏：移除创建点/线段入口，新增画笔、橡皮擦及尺寸控制，并按绘制、布尔、导入/AI 工具分组分隔。 - 扩展 Canvas 编辑能力：画笔按语义分类绘制并可自动并入连通选中 mask，橡皮擦对选中区域扣除，优化布尔操作、选区、撤销重做和保存状态联动。 - 优化自动传播时间轴显示：同一蓝色系按传播新旧递进变暗，老传播记录达到阈值后统一旧记录色，并维护范围选择与清空后的历史显示。 - 将 AI 智能分割入口替换为更明确的 AI 元素图标，并同步侧栏、工作区和 AI 页面入口表现。 - 完善模板分类、maskid 工具函数、分类树联动、遮罩透明度、边缘平滑和传播链同步相关前端状态。 - 扩展后端项目、媒体、任务、Dashboard、模板和传播 runner 的用户隔离、任务控制、进度事件与兼容处理。 - 补充前后端测试，覆盖用户管理、GT_label 往返导入导出、GT Mask 校验和预览、画笔/橡皮擦、时间轴传播历史、导出范围、WebSocket 与 API 封装。 - 更新 AGENTS、README 和 doc 文档，记录当前接口契约、实现状态、测试计划、安装说明和 maskid/GT_label 规则。
79 lines
2.3 KiB
Python
79 lines
2.3 KiB
Python
"""Shared pytest fixtures for backend API tests."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
from typing import Iterator
|
|
|
|
import pytest
|
|
from fastapi import FastAPI
|
|
from fastapi.testclient import TestClient
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import Session, sessionmaker
|
|
from sqlalchemy.pool import StaticPool
|
|
|
|
BACKEND_DIR = Path(__file__).resolve().parents[1]
|
|
if str(BACKEND_DIR) not in sys.path:
|
|
sys.path.insert(0, str(BACKEND_DIR))
|
|
|
|
from database import Base, get_db # noqa: E402
|
|
from main import websocket_progress # noqa: E402
|
|
from routers import admin, ai, auth, dashboard, export, media, projects, tasks, templates # noqa: E402
|
|
|
|
|
|
@pytest.fixture()
|
|
def db_session() -> Iterator[Session]:
|
|
engine = create_engine(
|
|
"sqlite://",
|
|
connect_args={"check_same_thread": False},
|
|
poolclass=StaticPool,
|
|
)
|
|
TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
|
Base.metadata.create_all(bind=engine)
|
|
session = TestingSessionLocal()
|
|
auth.ensure_default_admin(session)
|
|
try:
|
|
yield session
|
|
finally:
|
|
session.close()
|
|
Base.metadata.drop_all(bind=engine)
|
|
engine.dispose()
|
|
|
|
|
|
@pytest.fixture()
|
|
def app(db_session: Session) -> FastAPI:
|
|
test_app = FastAPI()
|
|
|
|
def override_get_db() -> Iterator[Session]:
|
|
yield db_session
|
|
|
|
test_app.dependency_overrides[get_db] = override_get_db
|
|
test_app.include_router(auth.router)
|
|
test_app.include_router(projects.router)
|
|
test_app.include_router(templates.router)
|
|
test_app.include_router(media.router)
|
|
test_app.include_router(ai.router)
|
|
test_app.include_router(export.router)
|
|
test_app.include_router(dashboard.router)
|
|
test_app.include_router(tasks.router)
|
|
test_app.include_router(admin.router)
|
|
|
|
@test_app.get("/health")
|
|
def health_check() -> dict[str, str]:
|
|
return {"status": "ok", "service": "SegServer"}
|
|
|
|
test_app.add_api_websocket_route("/ws/progress", websocket_progress)
|
|
|
|
return test_app
|
|
|
|
|
|
@pytest.fixture()
|
|
def client(app: FastAPI, db_session: Session) -> Iterator[TestClient]:
|
|
with TestClient(app) as test_client:
|
|
admin = auth.ensure_default_admin(db_session)
|
|
test_client.headers.update({
|
|
"Authorization": f"Bearer {auth.create_access_token(admin)}"
|
|
})
|
|
yield test_client
|