fix: add working directory safety checks to prevent test interference

Co-Authored-By: João <joao@crewai.com>
This commit is contained in:
Devin AI
2025-06-24 19:56:09 +00:00
parent cabdfafb4a
commit babcd1d712

View File

@@ -24,17 +24,23 @@ def temp_dir():
def test_create_folder_structure_strips_single_trailing_slash():
with tempfile.TemporaryDirectory() as temp_dir:
original_cwd = os.getcwd()
os.chdir(temp_dir)
try:
folder_path, folder_name, class_name = create_folder_structure("hello/")
assert folder_name == "hello"
assert class_name == "Hello"
assert folder_path.name == "hello"
assert folder_path.exists()
finally:
os.chdir(original_cwd)
def test_create_folder_structure_strips_multiple_trailing_slashes():
with tempfile.TemporaryDirectory() as temp_dir:
old_cwd = os.getcwd()
try:
os.chdir(temp_dir)
folder_path, folder_name, class_name = create_folder_structure("hello///")
@@ -42,10 +48,15 @@ def test_create_folder_structure_strips_multiple_trailing_slashes():
assert class_name == "Hello"
assert folder_path.name == "hello"
assert folder_path.exists()
finally:
if os.path.exists(old_cwd):
os.chdir(old_cwd)
def test_create_folder_structure_handles_complex_name_with_trailing_slash():
with tempfile.TemporaryDirectory() as temp_dir:
old_cwd = os.getcwd()
try:
os.chdir(temp_dir)
folder_path, folder_name, class_name = create_folder_structure("my-awesome_project/")
@@ -53,10 +64,15 @@ def test_create_folder_structure_handles_complex_name_with_trailing_slash():
assert class_name == "MyAwesomeProject"
assert folder_path.name == "my_awesome_project"
assert folder_path.exists()
finally:
if os.path.exists(old_cwd):
os.chdir(old_cwd)
def test_create_folder_structure_normal_name_unchanged():
with tempfile.TemporaryDirectory() as temp_dir:
old_cwd = os.getcwd()
try:
os.chdir(temp_dir)
folder_path, folder_name, class_name = create_folder_structure("hello")
@@ -64,6 +80,9 @@ def test_create_folder_structure_normal_name_unchanged():
assert class_name == "Hello"
assert folder_path.name == "hello"
assert folder_path.exists()
finally:
if os.path.exists(old_cwd):
os.chdir(old_cwd)
@@ -71,6 +90,8 @@ def test_create_folder_structure_normal_name_unchanged():
def test_create_folder_structure_with_parent_folder():
with tempfile.TemporaryDirectory() as temp_dir:
old_cwd = os.getcwd()
try:
os.chdir(temp_dir)
parent_path = Path(temp_dir) / "parent"
parent_path.mkdir()
@@ -82,6 +103,9 @@ def test_create_folder_structure_with_parent_folder():
assert folder_path.name == "child"
assert folder_path.parent == parent_path
assert folder_path.exists()
finally:
if os.path.exists(old_cwd):
os.chdir(old_cwd)
@mock.patch("crewai.cli.create_crew.copy_template")
@@ -140,6 +164,8 @@ def test_create_crew_normal_name_still_works(mock_load_env, mock_write_env, mock
def test_create_folder_structure_handles_spaces_and_dashes_with_slash():
with tempfile.TemporaryDirectory() as temp_dir:
old_cwd = os.getcwd()
try:
os.chdir(temp_dir)
folder_path, folder_name, class_name = create_folder_structure("My Cool-Project/")
@@ -147,10 +173,15 @@ def test_create_folder_structure_handles_spaces_and_dashes_with_slash():
assert class_name == "MyCoolProject"
assert folder_path.name == "my_cool_project"
assert folder_path.exists()
finally:
if os.path.exists(old_cwd):
os.chdir(old_cwd)
def test_create_folder_structure_raises_error_for_invalid_class_names():
with tempfile.TemporaryDirectory() as temp_dir:
old_cwd = os.getcwd()
try:
os.chdir(temp_dir)
invalid_cases = [
@@ -168,12 +199,17 @@ def test_create_folder_structure_raises_error_for_invalid_class_names():
for invalid_name, expected_error in invalid_cases:
with pytest.raises(ValueError, match=expected_error):
create_folder_structure(invalid_name)
finally:
if os.path.exists(old_cwd):
os.chdir(old_cwd)
def test_create_folder_structure_validates_class_names():
import keyword
with tempfile.TemporaryDirectory() as temp_dir:
old_cwd = os.getcwd()
try:
os.chdir(temp_dir)
valid_cases = [
@@ -194,6 +230,9 @@ def test_create_folder_structure_validates_class_names():
if folder_path.exists():
shutil.rmtree(folder_path)
finally:
if os.path.exists(old_cwd):
os.chdir(old_cwd)
@mock.patch("crewai.cli.create_crew.copy_template")