feat: support voice asset jobs in generation controls

backend/tests/test_generation_jobs.py

@@ -243,6 +243,8 @@ async def test_queue_story_asset_generation_dispatches_background_job(
     assert summary["input_type"] == "image"
     assert summary["status"] == "running"
     assert summary["current_step"] == "request_accepted"
+    assert summary["can_cancel"] is True
+    assert summary["can_retry"] is False
     mock_delay.assert_called_once_with(summary["id"])
 
     job = (
@@ -305,6 +307,92 @@ async def test_asset_generation_job_worker_completes_cover_image(
     ]
 
 
+async def test_cancel_queued_asset_generation_job_marks_it_canceled(
+    db_session,
+    auth_token,
+    degraded_story_with_text,
+):
+    async def override_get_db():
+        yield db_session
+
+    app.dependency_overrides[get_db] = override_get_db
+    transport = ASGITransport(app=app)
+
+    job = await create_generation_job(
+        db_session,
+        user_id=degraded_story_with_text.user_id,
+        output_mode="asset_generation",
+        input_type="image",
+        request_payload={"story_id": degraded_story_with_text.id, "assets": ["image"]},
+        story_id=degraded_story_with_text.id,
+    )
+
+    try:
+        async with AsyncClient(transport=transport, base_url="http://test") as client:
+            client.cookies.set("access_token", auth_token)
+
+            response = await client.post(f"/api/generations/jobs/{job.id}/cancel")
+
+        assert response.status_code == 200
+        data = response.json()
+        assert data["status"] == "canceled"
+        assert data["current_step"] == "generation_canceled"
+        assert data["can_cancel"] is False
+        assert data["can_retry"] is True
+    finally:
+        app.dependency_overrides.clear()
+
+
+async def test_retry_failed_asset_generation_job_requeues_new_worker_job(
+    db_session,
+    auth_token,
+    degraded_story_with_text,
+):
+    async def override_get_db():
+        yield db_session
+
+    app.dependency_overrides[get_db] = override_get_db
+    transport = ASGITransport(app=app)
+    task_delay_path = "app.tasks.generation_workflow.run_generation_workflow_task.delay"
+
+    failed_job = await create_generation_job(
+        db_session,
+        user_id=degraded_story_with_text.user_id,
+        output_mode="asset_generation",
+        input_type="image",
+        request_payload={"story_id": degraded_story_with_text.id, "assets": ["image"]},
+        story_id=degraded_story_with_text.id,
+    )
+    await finish_generation_job(
+        db_session,
+        job=failed_job,
+        story=degraded_story_with_text,
+        status="failed",
+        current_step="asset_generation_failed",
+        error_message="cover timeout",
+        message="Cover image generation failed.",
+    )
+
+    try:
+        with patch(task_delay_path) as mock_delay:
+            async with AsyncClient(transport=transport, base_url="http://test") as client:
+                client.cookies.set("access_token", auth_token)
+
+                response = await client.post(f"/api/generations/jobs/{failed_job.id}/retry")
+
+            assert response.status_code == 200
+            data = response.json()
+            assert data["id"] != failed_job.id
+            assert data["output_mode"] == "asset_generation"
+            assert data["status"] == "running"
+            assert data["current_step"] == "retry_queued"
+            assert data["can_cancel"] is True
+            assert data["can_retry"] is False
+            mock_delay.assert_called_once_with(data["id"])
+    finally:
+        app.dependency_overrides.clear()
+
+
 async def test_storybook_generation_is_queued_then_worker_records_page_image_events(
     db_session,
     auth_token,