Fix failing task test

Author: LucaButBoring

tests/server/fastmcp/test_integration.py

@@ -733,9 +733,11 @@ async def on_task_status(task_result: GetTaskResult) -> None:
                 task_status_updates.append(task_result.status)
 
             # Test begin_call_tool for task-based execution
+            # Use a longer delay (10s) to ensure the polling path has time to get status updates
+            # before the direct result completes (default poll interval is 5s)
             pending_request = session.begin_call_tool(
                 "long_running_computation",
-                arguments={"data": "test_data", "delay_seconds": 1},
+                arguments={"data": "test_data", "delay_seconds": 10},
             )
 
             # Wait for the result with callbacks
@@ -747,25 +749,23 @@ async def on_task_status(task_result: GetTaskResult) -> None:
             assert len(tool_result.content) == 1
             assert isinstance(tool_result.content[0], TextContent)
             assert "Processed: TEST_DATA" in tool_result.content[0].text
-            assert "1s" in tool_result.content[0].text or "1.0s" in tool_result.content[0].text
+            assert "10" in tool_result.content[0].text or "10.0s" in tool_result.content[0].text
 
             # Verify callbacks were invoked
             assert task_created_called, "on_task_created callback was not invoked"
             assert len(task_status_updates) > 0, "on_task_status callback was never invoked"
 
-            # Due to the race between direct result and task polling:
-            # - With 1s delay and 5s default polling interval, direct result usually wins
-            # - We should see at least one status update (typically "submitted")
-            # - We may or may not see "completed" depending on timing
+            # With 10s delay and 5s default polling interval, polling should get at least one update
+            # before the task completes
+            # We should see at least "submitted" and possibly "working" or "completed"
 
             # Verify we got at least one valid status
             valid_statuses = ["submitted", "working", "completed"]
             assert all(status in valid_statuses for status in task_status_updates), (
                 f"Got invalid status in updates: {task_status_updates}"
             )
 
-            # If direct result won the race, we may only see submitted/working; if polling won, we'll see completed
-            last_status = task_status_updates[-1]
-            assert last_status in ["submitted", "working", "completed"], (
-                f"Unexpected last status: {last_status} from {task_status_updates}"
+            # Verify we got at least the initial status
+            assert "submitted" in task_status_updates or "working" in task_status_updates, (
+                f"Expected to see submitted or working status, got: {task_status_updates}"
             )