more timeouts

tests/test_task_guardrails.py

@@ -8,6 +8,7 @@ from crewai.task import Task
 from crewai.tasks.task_output import TaskOutput
 
 
+@pytest.mark.timeout(60)
 def test_task_without_guardrail():
     """Test that tasks work normally without guardrails (backward compatibility)."""
     agent = Mock()
@@ -15,18 +16,17 @@ def test_task_without_guardrail():
     agent.execute_task.return_value = "test result"
     agent.crew = None
 
-    task = Task(
-        description="Test task",
-        expected_output="Output"
-    )
+    task = Task(description="Test task", expected_output="Output")
 
     result = task.execute_sync(agent=agent)
     assert isinstance(result, TaskOutput)
     assert result.raw == "test result"
 
 
+@pytest.mark.timeout(60)
 def test_task_with_successful_guardrail():
     """Test that successful guardrail validation passes transformed result."""
+
     def guardrail(result: TaskOutput):
         return (True, result.raw.upper())
 
@@ -35,35 +35,30 @@ def test_task_with_successful_guardrail():
     agent.execute_task.return_value = "test result"
     agent.crew = None
 
-    task = Task(
-        description="Test task",
-        expected_output="Output",
-        guardrail=guardrail
-    )
+    task = Task(description="Test task", expected_output="Output", guardrail=guardrail)
 
     result = task.execute_sync(agent=agent)
     assert isinstance(result, TaskOutput)
     assert result.raw == "TEST RESULT"
 
 
+@pytest.mark.timeout(60)
 def test_task_with_failing_guardrail():
     """Test that failing guardrail triggers retry with error context."""
+
     def guardrail(result: TaskOutput):
         return (False, "Invalid format")
 
     agent = Mock()
     agent.role = "test_agent"
-    agent.execute_task.side_effect = [
-        "bad result",
-        "good result"
-    ]
+    agent.execute_task.side_effect = ["bad result", "good result"]
     agent.crew = None
 
     task = Task(
         description="Test task",
         expected_output="Output",
         guardrail=guardrail,
-        max_retries=1
+        max_retries=1,
     )
 
     # First execution fails guardrail, second succeeds
@@ -75,8 +70,10 @@ def test_task_with_failing_guardrail():
     assert task.retry_count == 1
 
 
+@pytest.mark.timeout(60)
 def test_task_with_guardrail_retries():
     """Test that guardrail respects max_retries configuration."""
+
     def guardrail(result: TaskOutput):
         return (False, "Invalid format")
 
@@ -89,7 +86,7 @@ def test_task_with_guardrail_retries():
         description="Test task",
         expected_output="Output",
         guardrail=guardrail,
-        max_retries=2
+        max_retries=2,
     )
 
     with pytest.raises(Exception) as exc_info:
@@ -100,8 +97,10 @@ def test_task_with_guardrail_retries():
     assert "Invalid format" in str(exc_info.value)
 
 
+@pytest.mark.timeout(60)
 def test_guardrail_error_in_context():
     """Test that guardrail error is passed in context for retry."""
+
     def guardrail(result: TaskOutput):
         return (False, "Expected JSON, got string")
 
@@ -113,11 +112,12 @@ def test_guardrail_error_in_context():
         description="Test task",
         expected_output="Output",
         guardrail=guardrail,
-        max_retries=1
+        max_retries=1,
     )
 
     # Mock execute_task to succeed on second attempt
     first_call = True
+
     def execute_task(task, context, tools):
         nonlocal first_call
         if first_call: