From 9427b5873decd74684ea3c5a53faf608badcebba Mon Sep 17 00:00:00 2001 From: CI Bot Date: Sat, 11 Jul 2026 20:36:52 +0800 Subject: [PATCH] =?UTF-8?q?test:=20=E8=A1=A5=E5=85=A8StubGenerationTaskRep?= =?UTF-8?q?ository=E7=9A=84=E9=99=90=E6=B5=81=E8=AE=A1=E6=95=B0=E6=96=B9?= =?UTF-8?q?=E6=B3=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/integration/test_generation_api.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/integration/test_generation_api.py b/tests/integration/test_generation_api.py index 867985de4..88865f5e7 100644 --- a/tests/integration/test_generation_api.py +++ b/tests/integration/test_generation_api.py @@ -118,6 +118,16 @@ class StubGenerationTaskRepository: def count_by_user(self, user_id: str) -> int: return len([t for t in self._tasks.values() if t.created_by_user_id == user_id]) + def count_pending_by_user(self, user_id: str) -> int: + from app.models.generation_task import TaskStatus + pending_statuses = {TaskStatus.PENDING, TaskStatus.PROCESSING, TaskStatus.QUEUED} + return len([t for t in self._tasks.values() if t.created_by_user_id == user_id and t.status in pending_statuses]) + + def count_pending_total(self) -> int: + from app.models.generation_task import TaskStatus + pending_statuses = {TaskStatus.PENDING, TaskStatus.PROCESSING, TaskStatus.QUEUED} + return len([t for t in self._tasks.values() if t.status in pending_statuses]) + def list_recent_by_user(self, user_id: str, limit: int = 5) -> list[GenerationTask]: items = [t for t in self._tasks.values() if t.created_by_user_id == user_id] items.sort(key=lambda t: t.created_at, reverse=True)