Enhance run_crew.py with improved logging, error handling, and tests

Co-Authored-By: Joe Moura <joao@crewai.com>

tests/cli/test_run_crew.py

@@ -10,7 +10,14 @@ from crewai.cli.run_crew import CrewType, execute_command
 
 
 def test_execute_command_adds_src_to_path():
-    """Test that execute_command adds the src directory to sys.path."""
+    """
+    Test that execute_command correctly modifies sys.path.
+
+    Ensures:
+    1. src directory is added to sys.path when it exists.
+    2. Original sys.path is preserved for other entries.
+    3. Command execution proceeds correctly.
+    """
     # Create a temporary directory with a src subdirectory
     with tempfile.TemporaryDirectory() as temp_dir:
         temp_path = Path(temp_dir)

tests/cli/test_run_crew_edge_cases.py

@@ -0,0 +1,115 @@
+import os
+import subprocess
+import sys
+import tempfile
+from pathlib import Path
+from unittest.mock import patch
+
+import pytest
+
+from crewai.cli.run_crew import CrewType, execute_command
+
+
+@pytest.mark.parametrize("crew_type", [CrewType.STANDARD, CrewType.FLOW])
+def test_execute_command_with_different_crew_types(crew_type):
+    """
+    Test that execute_command works with different crew types.
+    
+    Verifies that the correct command is executed based on the crew type.
+    """
+    # Create a temporary directory with a src subdirectory
+    with tempfile.TemporaryDirectory() as temp_dir:
+        temp_path = Path(temp_dir)
+        src_path = temp_path / "src"
+        src_path.mkdir()
+        
+        # Change to the temporary directory
+        original_dir = os.getcwd()
+        os.chdir(temp_dir)
+        
+        try:
+            # Save the original sys.path
+            original_sys_path = sys.path.copy()
+            
+            # Mock subprocess.run to avoid actually running the command
+            with patch("subprocess.run") as mock_run:
+                # Call execute_command
+                execute_command(crew_type)
+                
+                # Check that the correct command was called based on crew_type
+                expected_command = ["uv", "run", "kickoff" if crew_type == CrewType.FLOW else "run_crew"]
+                mock_run.assert_called_once_with(expected_command, capture_output=False, text=True, check=True)
+        finally:
+            # Restore the original directory and sys.path
+            os.chdir(original_dir)
+            sys.path = original_sys_path
+
+
+def test_execute_command_handles_missing_src_directory():
+    """
+    Test that execute_command handles missing src directory gracefully.
+    
+    Verifies that the command executes even when the src directory doesn't exist.
+    """
+    # Create a temporary directory without a src subdirectory
+    with tempfile.TemporaryDirectory() as temp_dir:
+        temp_path = Path(temp_dir)
+        
+        # Change to the temporary directory
+        original_dir = os.getcwd()
+        os.chdir(temp_dir)
+        
+        try:
+            # Save the original sys.path
+            original_sys_path = sys.path.copy()
+            
+            # Mock subprocess.run to avoid actually running the command
+            with patch("subprocess.run") as mock_run:
+                # Call execute_command
+                execute_command(CrewType.STANDARD)
+                
+                # Check that sys.path wasn't modified (since src doesn't exist)
+                assert sys.path == original_sys_path
+                
+                # Check that the command was still called
+                mock_run.assert_called_once_with(["uv", "run", "run_crew"], capture_output=False, text=True, check=True)
+        finally:
+            # Restore the original directory and sys.path
+            os.chdir(original_dir)
+            sys.path = original_sys_path
+
+
+def test_execute_command_handles_subprocess_error():
+    """
+    Test that execute_command properly handles subprocess errors.
+    
+    Verifies that exceptions from subprocess.run are propagated correctly.
+    """
+    with tempfile.TemporaryDirectory() as temp_dir:
+        temp_path = Path(temp_dir)
+        src_path = temp_path / "src"
+        src_path.mkdir()
+        
+        # Create a dummy pyproject.toml file
+        with open(temp_path / "pyproject.toml", "w") as f:
+            f.write("[project]\nname = \"test\"\n")
+        
+        # Change to the temporary directory
+        original_dir = os.getcwd()
+        os.chdir(temp_dir)
+        
+        try:
+            # Save the original sys.path
+            original_sys_path = sys.path.copy()
+            
+            # Mock subprocess.run to raise an exception
+            with patch("subprocess.run", side_effect=subprocess.CalledProcessError(1, [])):
+                # Call execute_command and verify it handles the error
+                execute_command(CrewType.STANDARD)
+                
+                # Verify that sys.path was modified correctly
+                assert str(src_path) in sys.path
+        finally:
+            # Restore the original directory and sys.path
+            os.chdir(original_dir)
+            sys.path = original_sys_path