From 8f20de3c5366acdddadd8d7ebe71e0d516a912e4 Mon Sep 17 00:00:00 2001 From: CI Bot Date: Mon, 24 Aug 2026 15:20:14 +0000 Subject: [PATCH] style: auto-format with black + isort + prettier [skip ci-format-check] --- .../test_create_reuse_preview_fallback.py | 42 ++++++++++++------- 1 file changed, 27 insertions(+), 15 deletions(-) diff --git a/tests/unit/test_create_reuse_preview_fallback.py b/tests/unit/test_create_reuse_preview_fallback.py index 572132167..8649442e2 100644 --- a/tests/unit/test_create_reuse_preview_fallback.py +++ b/tests/unit/test_create_reuse_preview_fallback.py @@ -19,7 +19,6 @@ sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "..", "apps", " from packages.domain.generation_task import GenerationTask, GenerationTaskStatus - # ── Stubs ──────────────────────────────────────────────────────────────────── @@ -42,10 +41,13 @@ class StubGenerationTaskRepository: return [t for t in self._store.values() if t.created_by_user_id == user_id] def count_pending_by_user(self, user_id: str) -> int: - return len([ - t for t in self._store.values() - if t.created_by_user_id == user_id and t.status == GenerationTaskStatus.PENDING - ]) + return len( + [ + t + for t in self._store.values() + if t.created_by_user_id == user_id and t.status == GenerationTaskStatus.PENDING + ] + ) def count_pending_total(self) -> int: return len([t for t in self._store.values() if t.status == GenerationTaskStatus.PENDING]) @@ -194,6 +196,7 @@ def app( def _make_client(app: FastAPI, db: MagicMock) -> TestClient: from app.dependencies import get_db_session + app.dependency_overrides[get_db_session] = lambda: db return TestClient(app) @@ -230,11 +233,14 @@ class TestCreateEndpointFallback: gen_task_repo.create(preview) db, _ = _make_db_with_preview(preview) - with patch( - "packages.adapters.sqlalchemy_impl.generation_task_repository._to_domain", - return_value=preview, - ), patch( - "app.api.routes.generation_tasks._writeback_edit_plan_config", + with ( + patch( + "packages.adapters.sqlalchemy_impl.generation_task_repository._to_domain", + return_value=preview, + ), + patch( + "app.api.routes.generation_tasks._writeback_edit_plan_config", + ), ): client = _make_client(app, db) resp = client.post( @@ -307,8 +313,11 @@ class TestCreateEndpointFallback: assert resp.json()["items"][0]["is_preview"] is True # 兜底查询 GenerationTaskModel 不应被调用(只可能查 EditPlanModel 做自动关联) from packages.adapters.sqlalchemy_impl.models import GenerationTaskModel + for call_args in db.query.call_args_list: - assert call_args[0][0] is not GenerationTaskModel, "fallback should not query GenerationTaskModel for preview requests" + assert ( + call_args[0][0] is not GenerationTaskModel + ), "fallback should not query GenerationTaskModel for preview requests" def test_fallback_resolution_mismatch_creates_new( self, @@ -320,10 +329,13 @@ class TestCreateEndpointFallback: gen_task_repo.create(preview) db, _ = _make_db_with_preview(preview) - with patch( - "packages.adapters.sqlalchemy_impl.generation_task_repository._to_domain", - return_value=preview, - ), patch("app.api.routes.generation_tasks.safe_enqueue_generation_task", return_value=True): + with ( + patch( + "packages.adapters.sqlalchemy_impl.generation_task_repository._to_domain", + return_value=preview, + ), + patch("app.api.routes.generation_tasks.safe_enqueue_generation_task", return_value=True), + ): client = _make_client(app, db) resp = client.post( "/api/v1/generation/tasks",