From 58c335829fceaeafa2a9fc2ec3b8f843fa97d1a4 Mon Sep 17 00:00:00 2001 From: CI Bot Date: Tue, 14 Jul 2026 10:20:00 +0800 Subject: [PATCH] =?UTF-8?q?style:=20black=E6=A0=BC=E5=BC=8F=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/integration/test_generation_api.py | 2 +- tests/integration/test_task_center_api.py | 2 +- tests/unit/test_generation_presigned_url.py | 10 +++++----- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/tests/integration/test_generation_api.py b/tests/integration/test_generation_api.py index 4160b8ca0..c4192d439 100755 --- a/tests/integration/test_generation_api.py +++ b/tests/integration/test_generation_api.py @@ -138,7 +138,6 @@ class StubGenerationTaskRepository: def list_by_source_edit_plan(self, plan_id: str) -> list[GenerationTask]: return [t for t in self._tasks.values() if getattr(t, "source_edit_plan_id", "") == plan_id] - def list_by_user_filtered( self, user_id: str, @@ -171,6 +170,7 @@ class StubGenerationTaskRepository: items = [t for t in items if str(t.status) == status] return len(items) + class StubGeneratedVideoRepository: def __init__(self, videos: dict[str, GeneratedVideo] | None = None): self._videos = videos or {} diff --git a/tests/integration/test_task_center_api.py b/tests/integration/test_task_center_api.py index 1fb37c5b2..7a94f5122 100755 --- a/tests/integration/test_task_center_api.py +++ b/tests/integration/test_task_center_api.py @@ -103,7 +103,6 @@ class StubGenerationTaskRepository: def list_by_source_edit_plan(self, plan_id: str) -> list[GenerationTask]: return [t for t in self._tasks.values() if getattr(t, "source_edit_plan_id", "") == plan_id] - def list_by_user_filtered( self, user_id: str, @@ -136,6 +135,7 @@ class StubGenerationTaskRepository: items = [t for t in items if str(t.status) == status] return len(items) + class StubIngestJobRepository: def __init__(self, jobs: dict[str, IngestJob] | None = None): self._jobs = jobs or {} diff --git a/tests/unit/test_generation_presigned_url.py b/tests/unit/test_generation_presigned_url.py index 61eea45a9..725fc4da2 100755 --- a/tests/unit/test_generation_presigned_url.py +++ b/tests/unit/test_generation_presigned_url.py @@ -64,11 +64,10 @@ class StubGenerationTaskRepository: def count_pending_total(self): return 0 - def list_by_user_filtered(self, user_id, *, status=None, limit=None, offset=0): - items = [t for t in self._tasks.values() if getattr(t, 'created_by_user_id', None) == user_id] + items = [t for t in self._tasks.values() if getattr(t, "created_by_user_id", None) == user_id] if status: - items = [t for t in items if getattr(t, 'status', None) == status] + items = [t for t in items if getattr(t, "status", None) == status] if offset: items = items[offset:] if limit is not None: @@ -76,11 +75,12 @@ class StubGenerationTaskRepository: return items def count_by_user_filtered(self, user_id, *, status=None): - items = [t for t in self._tasks.values() if getattr(t, 'created_by_user_id', None) == user_id] + items = [t for t in self._tasks.values() if getattr(t, "created_by_user_id", None) == user_id] if status: - items = [t for t in items if getattr(t, 'status', None) == status] + items = [t for t in items if getattr(t, "status", None) == status] return len(items) + class StubGeneratedVideoRepository: def __init__(self, videos=None): self._videos = videos or {}