diff --git a/tests/unit/test_cover_extract_frames.py b/tests/unit/test_cover_extract_frames.py index 4813880d8..025fbbcb9 100644 --- a/tests/unit/test_cover_extract_frames.py +++ b/tests/unit/test_cover_extract_frames.py @@ -3,9 +3,11 @@ 1. Worker 渲染完成后提取封面帧写入 cover_url 2. API 创建预览任务时自动关联 source_edit_plan_id """ + import pytest -from packages.domain.generation_task import GenerationTask, GenerationTaskStatus + from packages.application.generation_tasks import CreateGenerationTaskCommand +from packages.domain.generation_task import GenerationTask, GenerationTaskStatus def _make_task(**kwargs): @@ -126,10 +128,7 @@ class TestCoverRouteStepB: cover_url="https://oss.example.com/cover.jpg", status=GenerationTaskStatus.FAILED, ) - is_valid = ( - task.status == GenerationTaskStatus.COMPLETED - and bool(task.cover_url) - ) + is_valid = task.status == GenerationTaskStatus.COMPLETED and bool(task.cover_url) assert is_valid is False def test_step_b_skips_tasks_without_cover_url(self): @@ -139,8 +138,5 @@ class TestCoverRouteStepB: cover_url="", status=GenerationTaskStatus.COMPLETED, ) - is_valid = ( - task.status == GenerationTaskStatus.COMPLETED - and bool(task.cover_url) - ) + is_valid = task.status == GenerationTaskStatus.COMPLETED and bool(task.cover_url) assert is_valid is False