more timeouts

tests/cli/authentication/test_auth_main.py

@@ -1,6 +1,7 @@
 import unittest
 from unittest.mock import MagicMock, patch
 
+import pytest
 import requests
 
 from crewai.cli.authentication.main import AuthenticationCommand
@@ -10,6 +11,7 @@ class TestAuthenticationCommand(unittest.TestCase):
     def setUp(self):
         self.auth_command = AuthenticationCommand()
 
+    @pytest.mark.timeout(60)
     @patch("crewai.cli.authentication.main.requests.post")
     def test_get_device_code(self, mock_post):
         mock_response = MagicMock()
@@ -30,6 +32,7 @@ class TestAuthenticationCommand(unittest.TestCase):
         )
         self.assertEqual(device_code_data["interval"], 5)
 
+    @pytest.mark.timeout(60)
     @patch("crewai.cli.authentication.main.console.print")
     @patch("crewai.cli.authentication.main.webbrowser.open")
     def test_display_auth_instructions(self, mock_open, mock_print):
@@ -44,6 +47,7 @@ class TestAuthenticationCommand(unittest.TestCase):
         mock_print.assert_any_call("2. Enter the following code: ", "ABCDEF")
         mock_open.assert_called_once_with("https://example.com")
 
+    @pytest.mark.timeout(60)
     @patch("crewai.cli.authentication.main.ToolCommand")
     @patch("crewai.cli.authentication.main.requests.post")
     @patch("crewai.cli.authentication.main.validate_token")
@@ -69,6 +73,7 @@ class TestAuthenticationCommand(unittest.TestCase):
             "\n[bold green]Welcome to CrewAI Enterprise![/bold green]\n"
         )
 
+    @pytest.mark.timeout(60)
     @patch("crewai.cli.authentication.main.requests.post")
     @patch("crewai.cli.authentication.main.console.print")
     def test_poll_for_token_error(self, mock_print, mock_post):
@@ -85,6 +90,7 @@ class TestAuthenticationCommand(unittest.TestCase):
 
         mock_print.assert_not_called()
 
+    @pytest.mark.timeout(60)
     @patch("crewai.cli.authentication.main.requests.post")
     @patch("crewai.cli.authentication.main.console.print")
     def test_poll_for_token_timeout(self, mock_print, mock_post):

tests/cli/authentication/test_utils.py

@@ -3,12 +3,14 @@ import unittest
 from datetime import datetime, timedelta
 from unittest.mock import MagicMock, patch
 
+import pytest
 from cryptography.fernet import Fernet
 
 from crewai.cli.authentication.utils import TokenManager, validate_token
 
 
 class TestValidateToken(unittest.TestCase):
+    @pytest.mark.timeout(60)
     @patch("crewai.cli.authentication.utils.AsymmetricSignatureVerifier")
     @patch("crewai.cli.authentication.utils.TokenVerifier")
     def test_validate_token(self, mock_token_verifier, mock_asymmetric_verifier):
@@ -32,6 +34,7 @@ class TestTokenManager(unittest.TestCase):
     def setUp(self):
         self.token_manager = TokenManager()
 
+    @pytest.mark.timeout(60)
     @patch("crewai.cli.authentication.utils.TokenManager.read_secure_file")
     @patch("crewai.cli.authentication.utils.TokenManager.save_secure_file")
     @patch("crewai.cli.authentication.utils.TokenManager._get_or_create_key")
@@ -44,6 +47,7 @@ class TestTokenManager(unittest.TestCase):
 
         self.assertEqual(result, mock_key)
 
+    @pytest.mark.timeout(60)
     @patch("crewai.cli.authentication.utils.Fernet.generate_key")
     @patch("crewai.cli.authentication.utils.TokenManager.read_secure_file")
     @patch("crewai.cli.authentication.utils.TokenManager.save_secure_file")
@@ -59,6 +63,7 @@ class TestTokenManager(unittest.TestCase):
         mock_generate.assert_called_once()
         mock_save.assert_called_once_with("secret.key", mock_key)
 
+    @pytest.mark.timeout(60)
     @patch("crewai.cli.authentication.utils.TokenManager.save_secure_file")
     def test_save_tokens(self, mock_save):
         access_token = "test_token"
@@ -79,6 +84,7 @@ class TestTokenManager(unittest.TestCase):
             delta=timedelta(seconds=1),
         )
 
+    @pytest.mark.timeout(60)
     @patch("crewai.cli.authentication.utils.TokenManager.read_secure_file")
     def test_get_token_valid(self, mock_read):
         access_token = "test_token"
@@ -91,6 +97,7 @@ class TestTokenManager(unittest.TestCase):
 
         self.assertEqual(result, access_token)
 
+    @pytest.mark.timeout(60)
     @patch("crewai.cli.authentication.utils.TokenManager.read_secure_file")
     def test_get_token_expired(self, mock_read):
         access_token = "test_token"
@@ -103,6 +110,7 @@ class TestTokenManager(unittest.TestCase):
 
         self.assertIsNone(result)
 
+    @pytest.mark.timeout(60)
     @patch("crewai.cli.authentication.utils.TokenManager.get_secure_storage_path")
     @patch("builtins.open", new_callable=unittest.mock.mock_open)
     @patch("crewai.cli.authentication.utils.os.chmod")
@@ -119,6 +127,7 @@ class TestTokenManager(unittest.TestCase):
         mock_open().write.assert_called_once_with(content)
         mock_chmod.assert_called_once_with(mock_path.__truediv__.return_value, 0o600)
 
+    @pytest.mark.timeout(60)
     @patch("crewai.cli.authentication.utils.TokenManager.get_secure_storage_path")
     @patch(
         "builtins.open", new_callable=unittest.mock.mock_open, read_data=b"test_content"
@@ -135,6 +144,7 @@ class TestTokenManager(unittest.TestCase):
         mock_path.__truediv__.assert_called_once_with(filename)
         mock_open.assert_called_once_with(mock_path.__truediv__.return_value, "rb")
 
+    @pytest.mark.timeout(60)
     @patch("crewai.cli.authentication.utils.TokenManager.get_secure_storage_path")
     def test_read_secure_file_not_exists(self, mock_get_path):
         mock_path = MagicMock()

tests/cli/deploy/test_deploy_main.py

@@ -12,6 +12,7 @@ from crewai.cli.utils import parse_toml
 
 
 class TestDeployCommand(unittest.TestCase):
+    @pytest.mark.timeout(60)
     @patch("crewai.cli.command.get_auth_token")
     @patch("crewai.cli.deploy.main.get_project_name")
     @patch("crewai.cli.command.PlusAPI")
@@ -26,10 +27,12 @@ class TestDeployCommand(unittest.TestCase):
         self.deploy_command = DeployCommand()
         self.mock_client = self.deploy_command.plus_api_client
 
+    @pytest.mark.timeout(60)
     def test_init_success(self):
         self.assertEqual(self.deploy_command.project_name, "test_project")
         self.mock_plus_api.assert_called_once_with(api_key="test_token")
 
+    @pytest.mark.timeout(60)
     @patch("crewai.cli.command.get_auth_token")
     def test_init_failure(self, mock_get_auth_token):
         mock_get_auth_token.side_effect = Exception("Auth failed")
@@ -37,6 +40,7 @@ class TestDeployCommand(unittest.TestCase):
         with self.assertRaises(SystemExit):
             DeployCommand()
 
+    @pytest.mark.timeout(60)
     def test_validate_response_successful_response(self):
         mock_response = Mock(spec=requests.Response)
         mock_response.json.return_value = {"message": "Success"}
@@ -47,6 +51,7 @@ class TestDeployCommand(unittest.TestCase):
             self.deploy_command._validate_response(mock_response)
             assert fake_out.getvalue() == ""
 
+    @pytest.mark.timeout(60)
     def test_validate_response_json_decode_error(self):
         mock_response = Mock(spec=requests.Response)
         mock_response.json.side_effect = JSONDecodeError("Decode error", "", 0)
@@ -64,6 +69,7 @@ class TestDeployCommand(unittest.TestCase):
             assert "Status Code: 500" in output
             assert "Response:\nb'Invalid JSON'" in output
 
+    @pytest.mark.timeout(60)
     def test_validate_response_422_error(self):
         mock_response = Mock(spec=requests.Response)
         mock_response.json.return_value = {
@@ -84,6 +90,7 @@ class TestDeployCommand(unittest.TestCase):
             assert "Field1 Error message 1" in output
             assert "Field2 Error message 2" in output
 
+    @pytest.mark.timeout(60)
     def test_validate_response_other_error(self):
         mock_response = Mock(spec=requests.Response)
         mock_response.json.return_value = {"error": "Something went wrong"}
@@ -97,11 +104,13 @@ class TestDeployCommand(unittest.TestCase):
             assert "Request to Enterprise API failed. Details:" in output
             assert "Details:\nSomething went wrong" in output
 
+    @pytest.mark.timeout(60)
     def test_standard_no_param_error_message(self):
         with patch("sys.stdout", new=StringIO()) as fake_out:
             self.deploy_command._standard_no_param_error_message()
             self.assertIn("No UUID provided", fake_out.getvalue())
 
+    @pytest.mark.timeout(60)
     def test_display_deployment_info(self):
         with patch("sys.stdout", new=StringIO()) as fake_out:
             self.deploy_command._display_deployment_info(
@@ -111,6 +120,7 @@ class TestDeployCommand(unittest.TestCase):
             self.assertIn("test-uuid", fake_out.getvalue())
             self.assertIn("deployed", fake_out.getvalue())
 
+    @pytest.mark.timeout(60)
     def test_display_logs(self):
         with patch("sys.stdout", new=StringIO()) as fake_out:
             self.deploy_command._display_logs(
@@ -118,6 +128,7 @@ class TestDeployCommand(unittest.TestCase):
             )
             self.assertIn("2023-01-01 - INFO: Test log", fake_out.getvalue())
 
+    @pytest.mark.timeout(60)
     @patch("crewai.cli.deploy.main.DeployCommand._display_deployment_info")
     def test_deploy_with_uuid(self, mock_display):
         mock_response = MagicMock()
@@ -130,6 +141,7 @@ class TestDeployCommand(unittest.TestCase):
         self.mock_client.deploy_by_uuid.assert_called_once_with("test-uuid")
         mock_display.assert_called_once_with({"uuid": "test-uuid"})
 
+    @pytest.mark.timeout(60)
     @patch("crewai.cli.deploy.main.DeployCommand._display_deployment_info")
     def test_deploy_with_project_name(self, mock_display):
         mock_response = MagicMock()
@@ -142,6 +154,7 @@ class TestDeployCommand(unittest.TestCase):
         self.mock_client.deploy_by_name.assert_called_once_with("test_project")
         mock_display.assert_called_once_with({"uuid": "test-uuid"})
 
+    @pytest.mark.timeout(60)
     @patch("crewai.cli.deploy.main.fetch_and_json_env_file")
     @patch("crewai.cli.deploy.main.git.Repository.origin_url")
     @patch("builtins.input")
@@ -160,6 +173,7 @@ class TestDeployCommand(unittest.TestCase):
             self.assertIn("Deployment created successfully!", fake_out.getvalue())
             self.assertIn("new-uuid", fake_out.getvalue())
 
+    @pytest.mark.timeout(60)
     def test_list_crews(self):
         mock_response = MagicMock()
         mock_response.status_code = 200
@@ -174,6 +188,7 @@ class TestDeployCommand(unittest.TestCase):
             self.assertIn("Crew1 (uuid1) active", fake_out.getvalue())
             self.assertIn("Crew2 (uuid2) inactive", fake_out.getvalue())
 
+    @pytest.mark.timeout(60)
     def test_get_crew_status(self):
         mock_response = MagicMock()
         mock_response.status_code = 200
@@ -185,6 +200,7 @@ class TestDeployCommand(unittest.TestCase):
             self.assertIn("TestCrew", fake_out.getvalue())
             self.assertIn("active", fake_out.getvalue())
 
+    @pytest.mark.timeout(60)
     def test_get_crew_logs(self):
         mock_response = MagicMock()
         mock_response.status_code = 200
@@ -199,6 +215,7 @@ class TestDeployCommand(unittest.TestCase):
             self.assertIn("2023-01-01 - INFO: Log1", fake_out.getvalue())
             self.assertIn("2023-01-02 - ERROR: Log2", fake_out.getvalue())
 
+    @pytest.mark.timeout(60)
     def test_remove_crew(self):
         mock_response = MagicMock()
         mock_response.status_code = 204
@@ -210,6 +227,7 @@ class TestDeployCommand(unittest.TestCase):
                 "Crew 'test_project' removed successfully", fake_out.getvalue()
             )
 
+    @pytest.mark.timeout(60)
     @unittest.skipIf(sys.version_info < (3, 11), "Requires Python 3.11+")
     def test_parse_toml_python_311_plus(self):
         toml_content = """
@@ -224,6 +242,7 @@ class TestDeployCommand(unittest.TestCase):
         parsed = parse_toml(toml_content)
         self.assertEqual(parsed["tool"]["poetry"]["name"], "test_project")
 
+    @pytest.mark.timeout(60)
     @patch(
         "builtins.open",
         new_callable=unittest.mock.mock_open,
@@ -242,6 +261,7 @@ class TestDeployCommand(unittest.TestCase):
         print("project_name", project_name)
         self.assertEqual(project_name, "test_project")
 
+    @pytest.mark.timeout(60)
     @unittest.skipIf(sys.version_info < (3, 11), "Requires Python 3.11+")
     @patch(
         "builtins.open",
@@ -260,6 +280,7 @@ class TestDeployCommand(unittest.TestCase):
         project_name = get_project_name()
         self.assertEqual(project_name, "test_project")
 
+    @pytest.mark.timeout(60)
     def test_get_crewai_version(self):
         from crewai.cli.version import get_crewai_version
 

tests/cli/tools/test_main.py

@@ -7,6 +7,7 @@ from io import StringIO
 from unittest import mock
 from unittest.mock import MagicMock, patch
 
+import pytest
 from pytest import raises
 
 from crewai.cli.tools.main import ToolCommand
@@ -23,14 +24,16 @@ def in_temp_dir():
             os.chdir(original_dir)
 
 
+@pytest.mark.timeout(60)
 @patch("crewai.cli.tools.main.subprocess.run")
 def test_create_success(mock_subprocess):
     with in_temp_dir():
         tool_command = ToolCommand()
 
-        with patch.object(tool_command, "login") as mock_login, patch(
-            "sys.stdout", new=StringIO()
-        ) as fake_out:
+        with (
+            patch.object(tool_command, "login") as mock_login,
+            patch("sys.stdout", new=StringIO()) as fake_out,
+        ):
             tool_command.create("test-tool")
             output = fake_out.getvalue()
 
@@ -52,6 +55,7 @@ def test_create_success(mock_subprocess):
         assert "Creating custom tool test_tool..." in output
 
 
+@pytest.mark.timeout(60)
 @patch("crewai.cli.tools.main.subprocess.run")
 @patch("crewai.cli.plus_api.PlusAPI.get_tool")
 def test_install_success(mock_get, mock_subprocess_run):
@@ -82,12 +86,13 @@ def test_install_success(mock_get, mock_subprocess_run):
         capture_output=False,
         text=True,
         check=True,
-        env=unittest.mock.ANY
+        env=unittest.mock.ANY,
     )
 
     assert "Successfully installed sample-tool" in output
 
 
+@pytest.mark.timeout(60)
 @patch("crewai.cli.plus_api.PlusAPI.get_tool")
 def test_install_tool_not_found(mock_get):
     mock_get_response = MagicMock()
@@ -107,6 +112,7 @@ def test_install_tool_not_found(mock_get):
     assert "No tool found with this name" in output
 
 
+@pytest.mark.timeout(60)
 @patch("crewai.cli.plus_api.PlusAPI.get_tool")
 def test_install_api_error(mock_get):
     mock_get_response = MagicMock()
@@ -126,6 +132,7 @@ def test_install_api_error(mock_get):
     assert "Failed to get tool details" in output
 
 
+@pytest.mark.timeout(60)
 @patch("crewai.cli.tools.main.git.Repository.is_synced", return_value=False)
 def test_publish_when_not_in_sync(mock_is_synced):
     with patch("sys.stdout", new=StringIO()) as fake_out, raises(SystemExit):
@@ -135,6 +142,7 @@ def test_publish_when_not_in_sync(mock_is_synced):
     assert "Local changes need to be resolved before publishing" in fake_out.getvalue()
 
 
+@pytest.mark.timeout(60)
 @patch("crewai.cli.tools.main.get_project_name", return_value="sample-tool")
 @patch("crewai.cli.tools.main.get_project_version", return_value="1.0.0")
 @patch("crewai.cli.tools.main.get_project_description", return_value="A sample tool")
@@ -183,6 +191,7 @@ def test_publish_when_not_in_sync_and_force(
     )
 
 
+@pytest.mark.timeout(60)
 @patch("crewai.cli.tools.main.get_project_name", return_value="sample-tool")
 @patch("crewai.cli.tools.main.get_project_version", return_value="1.0.0")
 @patch("crewai.cli.tools.main.get_project_description", return_value="A sample tool")
@@ -231,6 +240,7 @@ def test_publish_success(
     )
 
 
+@pytest.mark.timeout(60)
 @patch("crewai.cli.tools.main.get_project_name", return_value="sample-tool")
 @patch("crewai.cli.tools.main.get_project_version", return_value="1.0.0")
 @patch("crewai.cli.tools.main.get_project_description", return_value="A sample tool")
@@ -270,6 +280,7 @@ def test_publish_failure(
     assert "Name is already taken" in output
 
 
+@pytest.mark.timeout(60)
 @patch("crewai.cli.tools.main.get_project_name", return_value="sample-tool")
 @patch("crewai.cli.tools.main.get_project_version", return_value="1.0.0")
 @patch("crewai.cli.tools.main.get_project_description", return_value="A sample tool")