feat(ai-studio): separate image pool from task states
This commit is contained in:
@@ -256,6 +256,9 @@ class ImageStudioGenerationTests(TempDirMixin, unittest.TestCase):
|
||||
path=cfg["db_path"],
|
||||
)
|
||||
|
||||
failed_job = image_studio.get_job(job.id, path=cfg["db_path"])
|
||||
self.assertEqual(image_studio.JOB_RECOVERY_RESUME, failed_job.recovery_action)
|
||||
|
||||
resumable = image_studio.list_resumable_jobs(
|
||||
path=cfg["db_path"],
|
||||
project_id=project.id,
|
||||
@@ -291,6 +294,7 @@ class ImageStudioGenerationTests(TempDirMixin, unittest.TestCase):
|
||||
self.assertEqual("succeeded", updated.status)
|
||||
self.assertEqual("cmhub-task-download", updated.task_id)
|
||||
self.assertEqual("call-download", updated.call_id)
|
||||
self.assertEqual(image_studio.JOB_RECOVERY_NONE, updated.recovery_action)
|
||||
assets = image_studio.list_assets(project.id, kind="generated_main", path=cfg["db_path"])
|
||||
self.assertEqual(1, len(assets))
|
||||
|
||||
@@ -339,6 +343,16 @@ class ImageStudioGenerationTests(TempDirMixin, unittest.TestCase):
|
||||
self.assertEqual(1, summary["failed"])
|
||||
assets = image_studio.list_assets(project.id, kind="generated_main", path=cfg["db_path"])
|
||||
self.assertEqual(1, len(assets))
|
||||
failed_jobs = [
|
||||
job
|
||||
for job in image_studio.list_resumable_jobs(
|
||||
path=cfg["db_path"],
|
||||
project_id=project.id,
|
||||
include_failed_downloads=True,
|
||||
)
|
||||
if job.status == "failed"
|
||||
]
|
||||
self.assertEqual([], failed_jobs)
|
||||
|
||||
self.assert_removed(temp_dir)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user