style: fix whitespace in test files

Co-Authored-By: Rushil Srivastava <[email protected]>

Author: devin-ai-integration[bot]

tests/conftest.py

@@ -31,7 +31,7 @@ def scheduled_route():
     """Create a ScheduledRouteBuilder configured for testing."""
     return ScheduledRouteBuilder(
         base_url="http://localhost:8000",
-        location_path=f"projects/test-project/locations/us-central1",
+        location_path="projects/test-project/locations/us-central1",
     )
 
 

tests/test_delayed_route.py

@@ -1,76 +1,75 @@
 """Tests for DelayedRouteBuilder functionality."""
-import pytest
+
 from fastapi import APIRouter, Depends
+from google.protobuf import duration_pb2
 from pydantic import BaseModel
 
 from fastapi_gcp_tasks.dependencies import max_retries
 from fastapi_gcp_tasks.hooks import deadline_delayed_hook
-from google.protobuf import duration_pb2
 
 
 class TestPayload(BaseModel):
     """Test payload for task endpoints."""
+
     message: str
 
 
 def test_delayed_task_creation(app, delayed_route, test_client):
     """Test basic task creation and execution."""
     router = APIRouter(route_class=delayed_route)
-    
+
     @router.post("/test")
     async def test_task(payload: TestPayload):
         return {"received": payload.message}
-    
+
     app.include_router(router)
-    
+
     response = test_client.post("/test", json={"message": "test"})
     assert response.status_code == 200
     assert response.json() == {"received": "test"}
 
 
 def test_delayed_task_with_hook(app, delayed_route, test_client):
     """Test task creation with deadline hook."""
-    delayed_route.pre_create_hook = deadline_delayed_hook(
-        duration=duration_pb2.Duration(seconds=300)
-    )
+    delayed_route.pre_create_hook = deadline_delayed_hook(duration=duration_pb2.Duration(seconds=300))
     router = APIRouter(route_class=delayed_route)
-    
+
     @router.post("/test-hook")
     async def test_task(payload: TestPayload):
         return {"received": payload.message}
-    
+
     app.include_router(router)
-    
+
     response = test_client.post("/test-hook", json={"message": "test"})
     assert response.status_code == 200
 
 
 def test_delayed_task_with_retries(app, delayed_route, test_client):
     """Test task with max retries dependency."""
     router = APIRouter(route_class=delayed_route)
-    
+
     @router.post("/test-retries", dependencies=[Depends(max_retries(2))])
     async def test_task():
         raise Exception("Test failure")
-    
+
     app.include_router(router)
-    
+
     response = test_client.post("/test-retries")
     assert response.status_code == 500  # Should fail after 2 retries
 
 
 def test_delayed_task_with_countdown(app, delayed_route, test_client):
     """Test task creation with countdown."""
     router = APIRouter(route_class=delayed_route)
-    
+
     @router.post("/test-countdown")
     async def test_task(payload: TestPayload):
         return {"received": payload.message}
-    
+
     app.include_router(router)
-    
+
     # Test with countdown option
     test_task.options(countdown=60).delay(payload=TestPayload(message="delayed"))
-    
+
     response = test_client.post("/test-countdown", json={"message": "test"})
     assert response.status_code == 200

tests/test_scheduled_route.py

@@ -1,57 +1,52 @@
 """Tests for ScheduledRouteBuilder functionality."""
-import pytest
+
 from fastapi import APIRouter
+from google.protobuf import duration_pb2
 from pydantic import BaseModel
 
 from fastapi_gcp_tasks.hooks import deadline_scheduled_hook
-from google.protobuf import duration_pb2
 
 
 class TestPayload(BaseModel):
     """Test payload for scheduled task endpoints."""
+
     message: str
 
 
 def test_scheduled_task_creation(app, scheduled_route, test_client):
     """Test basic scheduled task creation."""
     router = APIRouter(route_class=scheduled_route)
-    
+
     @router.post("/test-scheduled")
     async def test_task(payload: TestPayload):
         return {"received": payload.message}
-    
+
     app.include_router(router)
-    
+
     # Schedule a task
-    test_task.scheduler(
-        name="test-scheduled-task",
-        schedule="*/5 * * * *",
-        time_zone="UTC"
-    ).schedule(payload=TestPayload(message="scheduled"))
-    
+    test_task.scheduler(name="test-scheduled-task", schedule="*/5 * * * *", time_zone="UTC").schedule(
+        payload=TestPayload(message="scheduled")
+    )
+
     response = test_client.post("/test-scheduled", json={"message": "test"})
     assert response.status_code == 200
 
 
 def test_scheduled_task_with_hook(app, scheduled_route, test_client):
     """Test scheduled task with deadline hook."""
-    scheduled_route.pre_create_hook = deadline_scheduled_hook(
-        duration=duration_pb2.Duration(seconds=300)
-    )
+    scheduled_route.pre_create_hook = deadline_scheduled_hook(duration=duration_pb2.Duration(seconds=300))
     router = APIRouter(route_class=scheduled_route)
-    
+
     @router.post("/test-scheduled-hook")
     async def test_task(payload: TestPayload):
         return {"received": payload.message}
-    
+
     app.include_router(router)
-    
+
     # Schedule a task with hook
-    test_task.scheduler(
-        name="test-scheduled-task-hook",
-        schedule="0 * * * *",
-        time_zone="UTC"
-    ).schedule(payload=TestPayload(message="scheduled with hook"))
-    
+    test_task.scheduler(name="test-scheduled-task-hook", schedule="0 * * * *", time_zone="UTC").schedule(
+        payload=TestPayload(message="scheduled with hook")
+    )
+
     response = test_client.post("/test-scheduled-hook", json={"message": "test"})
     assert response.status_code == 200