feat: 批量重跑实时刷新任务状态 (#143)
This commit is contained in:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user