feat: 批量重跑实时刷新任务状态 (#143)

2026-08-11 11:20:22 +08:00
parent e3dc48e1e6
commit 6553ae9367
7 changed files with 134 additions and 5 deletions
@@ -217,6 +217,24 @@ class PurchaseTaskServiceTest(unittest.TestCase):
        self.assertFalse(purchase["order_submitted"])
        self.assertFalse(purchase["payment_attempted"])

+    def test_started_callback_runs_after_database_enters_running(self):
+        self._prepare_task(task_id="PUR-STARTED")
+        observed = []
+        service = PurchaseTaskService(
+            self.gateway,
+            self.repository,
+            self.client,
+            "USB-001",
+            lambda _address, _cancelled: RecordingDryRunAdapter(),
+            started=lambda task_id: observed.append(
+                (task_id, self.repository.get_task(task_id).status)
+            ),
+        )
+
+        service.execute_selected("PUR-STARTED")
+
+        self.assertEqual(observed, [("PUR-STARTED", TaskStatus.RUNNING)])
+
    def test_price_above_limit_stops_before_confirmation(self):
        self._prepare_task()
        adapter = RecordingDryRunAdapter(price_cent=5001)