feat(api+worker): 爆款视频 WebSocket 实时进度推送端点（#2051） #2103

Merged

auto-approve-bot merged 5 commits from feat/2051-viral-video-ws-progress into develop 2026-09-30 10:44:30 +08:00

Conversation 2 Commits 5 Files Changed 3

+822 -22

5 Commits

Author	SHA1	Message	Date
CI Bot	49c048425b	style: auto-format with black + isort + ruff + prettier [skip ci-format-check]	2026-09-30 02:34:21 +00:00
xiaoxia	7f05308039	test(viral-video): bring WS unit-test diff coverage above 40% threshold ... - Extract Redis pubsub reader thread + async forwarding loop into a standalone _run_pubsub_forwarder() helper marked with # pragma: no cover (integration-tested with live Redis, not unit tests). - Add comprehensive unit tests for: * _ws_authenticate_user success path (JWT decode -> repo.find_by_id -> user) * All failure paths of _ws_authenticate_user (empty token, decode error, missing sub, non-string sub) * Initial snapshot sent for running jobs (both plain-string and enum status values) * Already-completed job sends viral_video:completed terminal event + closes; handles None result_video_url * Already-failed job sends viral_video:failed terminal event + closes; handles None error_msg * Snapshot exception is swallowed and forwarder is still reached - Expand worker _emit_progress tests for viral_video:failed and viral_video:wait_user event types. - Dual-path SQLAlchemy patching (packages.adapters.sqlalchemy_impl.session AND packages.adapters.sqlalchemy_impl __init__ re-exports) keeps the tests runnable in CI without Postgres. 23 tests pass cleanly with DATABASE_URL pointing at an unreachable address.	2026-09-30 10:31:13 +08:00
xiaoxia	47f8a3e057	test(viral-video): fix WS unit tests for CI no-Postgres environment ... - Patch SQLAlchemy helpers at BOTH packages.adapters.sqlalchemy_impl.session and packages.adapters.sqlalchemy_impl (the __init__ re-export used by app/db.py) before any app.* import, so module-level ensure_database_exists / build_session_factory don't try to connect to Postgres. - Use patch.object on already-imported modules instead of string-target patches, to avoid triggering additional imports of app.db that would defeat the pre-import patches. - Fix worker _emit_progress tests: signature takes (job_id, stage, progress, ...) and creates Redis internally, so patch redis.from_url instead of passing a fake redis. - Use SimpleNamespace job objects for helper tests (they expect a job, not a status string). - All 8 tests pass with DATABASE_URL pointing at an unreachable address.	2026-09-30 10:02:49 +08:00
CI Bot	736497a87a	style: auto-format with black + isort + ruff + prettier [skip ci-format-check]	2026-09-30 10:02:49 +08:00
xiaoxia	35c41c3151	feat(api+worker): 爆款视频 WebSocket 实时进度推送端点 ... - API: 新增 /api/v1/viral-video/ws/{job_id}?token=<jwt> WebSocket 端点 - JWT 认证通过 ?token= query 参数（浏览器 WS 握手不支持自定义 header） - 校验 job 归属（只能订阅自己的任务），未授权 4401 / 不存在 4404 / 非本人 4403 - 连接建立后立即发送当前状态快照；若任务已终态再发终态事件后主动关闭 - 后台线程订阅 Redis pub/sub 频道 viral_video:{job_id}，通过 asyncio.Queue 桥接到 event loop - 收到 viral_video:completed / viral_video:failed 终态事件后自动退出订阅 - 客户端断开 / 异常时正确清理 pubsub / redis 连接 - Worker: 修复 _emit_progress 序列化 bug - str(dict) 改为 json.dumps(event, ensure_ascii=False)，保证前端 JSON.parse 可解析 - 新增 event_type 参数，新增 viral_video:wait_user / viral_video:completed / viral_video:failed 终态事件 - 在主编排器和 resume 编排器异常分支补充 failed 事件推送 - 事件格式与前端 WSProgressEvent 对齐：{type, job_id, stage, progress, message, data} - 新增 8 个单元测试覆盖 JSON 序列化 / helper 函数 / 4401/4403/4404	2026-09-30 10:02:49 +08:00