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

This commit is contained in:
chengma
2026-08-11 11:20:22 +08:00
parent e3dc48e1e6
commit 6553ae9367
7 changed files with 134 additions and 5 deletions
+24
View File
@@ -207,6 +207,30 @@ class CollectTaskServiceTest(unittest.TestCase):
self.repository.get_task("COL-OTHER").status, TaskStatus.CLAIMED
)
def test_started_callback_runs_after_database_enters_running(self):
self.repository.add_claimed_task(
NewClaimedTask(
remote_task_id="COL-STARTED",
task_type=TaskType.COLLECT,
goods_url="https://example.test/COL-STARTED",
)
)
observed = []
service = CollectTaskService(
self.gateway,
self.repository,
self.client,
"USB-001",
started=lambda task_id: observed.append(
(task_id, self.repository.get_task(task_id).status)
),
collect_service_factory=lambda *_args: FakeCollector([]),
)
service.execute_selected("COL-STARTED")
self.assertEqual(observed, [("COL-STARTED", TaskStatus.RUNNING)])
def test_selected_task_cancelled_before_collector_starts_is_persisted(self):
self.repository.add_claimed_task(
NewClaimedTask(