feat: 执行并提交 PDD 采集任务 (#32)

2026-08-07 17:38:29 +08:00
parent 34b4b54b63
commit 2d1fdbea3f
17 changed files with 2156 additions and 108 deletions
@@ -0,0 +1,164 @@
+"""采集任务应用流程测试，不连接真机和网络。"""
+
+import tempfile
+import unittest
+from pathlib import Path
+
+from src.admin_gateway import AdminTask, ClaimCapabilities, ClientInfo
+from src.collect_task_service import CollectTaskService
+from src.mock_admin_gateway import MockAdminGateway
+from src.pdd_collect_service import PddCollectError
+from src.task_models import NewClaimedTask, TaskStatus, TaskType
+from src.task_repository import TaskRepository
+
+
+class FakeResult:
+    def to_pdd_data(self):
+        return {
+            "schema_version": 1,
+            "goods_id": "737116531267",
+            "title": "测试商品",
+            "shop_name": "测试店铺",
+            "price_granularity": "color",
+            "dimensions": [
+                {"key": "color", "name": "颜色分类"},
+                {"key": "size", "name": "尺码"},
+            ],
+            "skus": [
+                {
+                    "options": {"color": "黑色", "size": "M"},
+                    "price_cent": 990,
+                    "price_observed_at": {"color": "黑色", "size": "M"},
+                    "available": True,
+                }
+            ],
+        }
+
+
+class FakeCollector:
+    def __init__(self, calls, error=None):
+        self.calls = calls
+        self.error = error
+
+    def collect(self, task):
+        self.calls.append(task.remote_task_id)
+        if self.error is not None:
+            raise self.error
+        return FakeResult()
+
+
+class CollectTaskServiceTest(unittest.TestCase):
+    def setUp(self):
+        self.temporary = tempfile.TemporaryDirectory()
+        self.repository = TaskRepository(Path(self.temporary.name) / "client.db")
+        self.gateway = MockAdminGateway()
+        self.client = ClientInfo("CLIENT-001", "测试电脑")
+        self.task = AdminTask(
+            task_id="COL-001",
+            task_type=TaskType.COLLECT,
+            version=1,
+            priority=0,
+            payload={
+                "goods_id": "737116531267",
+                "goods_url": "https://mobile.yangkeduo.com/goods.html?goods_id=737116531267",
+            },
+            created_at="2026-08-07T08:00:00Z",
+            updated_at="2026-08-07T08:00:00Z",
+        )
+        self.gateway.enqueue_task(self.task, self.client.client_id)
+
+    def tearDown(self):
+        self.temporary.cleanup()
+
+    def _service(self, calls, error=None):
+        return CollectTaskService(
+            self.gateway,
+            self.repository,
+            self.client,
+            "USB-001",
+            collect_service_factory=lambda *_args: FakeCollector(calls, error),
+        )
+
+    def test_claim_collect_persist_submit_completes_one_task(self):
+        calls = []
+
+        outcome = self._service(calls).execute_one()
+
+        self.assertEqual(outcome.kind, "succeeded")
+        self.assertEqual(calls, ["COL-001"])
+        detail = self.repository.get_task("COL-001")
+        self.assertEqual(detail.status, TaskStatus.SUCCEEDED)
+        self.assertEqual(detail.pdd_data["price_granularity"], "color")
+        self.assertEqual(self.gateway.submission_count, 1)
+
+    def test_submit_timeout_retries_stored_outbox_without_recollecting(self):
+        remote = self.gateway.claim_next(
+            self.client,
+            ClaimCapabilities(supported_types=(TaskType.COLLECT,)),
+        )
+        self.repository.add_claimed_task(
+            NewClaimedTask(
+                remote_task_id=remote.task_id,
+                task_type=remote.task_type,
+                goods_url=remote.payload["goods_url"],
+                goods_id=remote.payload["goods_id"],
+                version=remote.version,
+                admin_payload={"payload": dict(remote.payload)},
+            )
+        )
+        calls = []
+        self.gateway.timeout_next_call()
+
+        first = self._service(calls).execute_one()
+        second = self._service(calls).execute_one()
+
+        self.assertEqual(first.kind, "result_pending")
+        self.assertEqual(second.kind, "succeeded")
+        self.assertEqual(calls, ["COL-001"])
+        self.assertEqual(self.gateway.submission_count, 1)
+
+    def test_pending_outbox_can_submit_without_android_device(self):
+        remote = self.gateway.claim_next(
+            self.client,
+            ClaimCapabilities(supported_types=(TaskType.COLLECT,)),
+        )
+        self.repository.add_claimed_task(
+            NewClaimedTask(
+                remote_task_id=remote.task_id,
+                task_type=remote.task_type,
+                goods_url=remote.payload["goods_url"],
+                goods_id=remote.payload["goods_id"],
+                version=remote.version,
+            )
+        )
+        started = self.repository.start_collect_run(remote.task_id, "USB-001")
+        self.repository.save_collect_result(
+            remote.task_id, started.attempt_id, FakeResult().to_pdd_data()
+        )
+
+        outcome = CollectTaskService(
+            self.gateway, self.repository, self.client, ""
+        ).execute_one()
+
+        self.assertEqual(outcome.kind, "succeeded")
+        self.assertEqual(
+            self.repository.get_task(remote.task_id).status,
+            TaskStatus.SUCCEEDED,
+        )
+
+    def test_captcha_becomes_manual_review_and_is_reported(self):
+        calls = []
+        outcome = self._service(
+            calls, PddCollectError("PDD_PAGE_CAPTCHA", "需要验证")
+        ).execute_one()
+
+        self.assertEqual(outcome.kind, "failed")
+        self.assertEqual(
+            self.repository.get_task("COL-001").status,
+            TaskStatus.MANUAL_REVIEW,
+        )
+        self.assertEqual(self.gateway.submission_count, 1)
+
+
+if __name__ == "__main__":
+    unittest.main()