fix: 修改一下生图的 bug

2026-08-31 21:56:48 +08:00
parent 2a667b4f3b
commit 8b54f7ed1e
14 changed files with 636 additions and 47 deletions
@@ -0,0 +1,63 @@
+"""后台生图 Worker 的取消竞态回归测试。"""
+from __future__ import annotations
+
+import sys
+import unittest
+from pathlib import Path
+from types import SimpleNamespace
+from unittest.mock import AsyncMock, patch
+
+SERVER_DIR = Path(__file__).resolve().parents[1]
+if str(SERVER_DIR) not in sys.path:
+    sys.path.insert(0, str(SERVER_DIR))
+
+from services.generator import GenerationAdvance
+from services.job_worker import process_item
+from services.jobs import ClaimedItem
+
+
+class WorkerCancellationTests(unittest.IsolatedAsyncioTestCase):
+    async def test_cancelled_task_is_not_returned_to_provider_waiting_queue(self) -> None:
+        claim = ClaimedItem(
+            job_id="00000000-0000-0000-0000-000000000001",
+            item_id="00000000-0000-0000-0000-000000000002",
+            job_kind="image_edit",
+            product_id=None,
+            provider="pcore",
+            model="gpt-image-2",
+            platform="ozon",
+            lang="ru",
+            ratio="3:4",
+            style_set=5,
+            request_payload={"context": {"title": "test"}, "ref_images": []},
+            item_payload={"kind": "custom", "title": "AI生图"},
+            provider_task_id="provider-task",
+            provider_task_kind="edits",
+        )
+        settings = SimpleNamespace(
+            image_provider="pcore",
+            dashscope_model="wan2.7-image-pro",
+            rightapi_image_model="gpt-image-2",
+            ark_image_model="doubao-seedream-4-5-251128",
+        )
+
+        with (
+            patch("services.job_worker.get_settings", return_value=settings),
+            patch("services.job_worker.build_context", return_value={}),
+            patch("services.job_worker.build_prompt", return_value="prompt"),
+            patch("services.job_worker._refs_for_job", return_value=[]),
+            patch("services.job_worker.advance_generation", new=AsyncMock(return_value=GenerationAdvance(
+                provider_task_id="provider-task", provider_task_kind="edits",
+            ))),
+            patch("services.job_worker.job_can_write", new=AsyncMock(return_value=False)),
+            patch("services.job_worker.mark_item_cancelled", new=AsyncMock()) as cancelled,
+            patch("services.job_worker.mark_waiting_provider", new=AsyncMock()) as waiting,
+        ):
+            await process_item(claim)
+
+        cancelled.assert_awaited_once_with(claim.item_id)
+        waiting.assert_not_awaited()
+
+
+if __name__ == "__main__":
+    unittest.main()