mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-18 14:34:23 +01:00
Maybe fix tests, fix safe_path function.
This commit is contained in:
@@ -35,7 +35,7 @@ def safe_path_join(base: Path, *paths: str | Path) -> Path:
|
|||||||
"""
|
"""
|
||||||
joined_path = base.joinpath(*paths).resolve()
|
joined_path = base.joinpath(*paths).resolve()
|
||||||
|
|
||||||
if not joined_path.is_relative_to(base):
|
if not str(joined_path.absolute()).startswith(str(base.absolute())):
|
||||||
raise ValueError(
|
raise ValueError(
|
||||||
f"Attempted to access path '{joined_path}' outside of working directory '{base}'."
|
f"Attempted to access path '{joined_path}' outside of working directory '{base}'."
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -139,7 +139,7 @@ class TestCommandRegistry:
|
|||||||
def test_import_mock_commands_module(self):
|
def test_import_mock_commands_module(self):
|
||||||
"""Test that the registry can import a module with mock command plugins."""
|
"""Test that the registry can import a module with mock command plugins."""
|
||||||
registry = CommandRegistry()
|
registry = CommandRegistry()
|
||||||
mock_commands_module = "auto_gpt.tests.mocks.mock_commands"
|
mock_commands_module = "tests.mocks.mock_commands"
|
||||||
|
|
||||||
registry.import_commands(mock_commands_module)
|
registry.import_commands(mock_commands_module)
|
||||||
|
|
||||||
@@ -155,7 +155,7 @@ class TestCommandRegistry:
|
|||||||
registry = CommandRegistry()
|
registry = CommandRegistry()
|
||||||
|
|
||||||
# Create a temp command file
|
# Create a temp command file
|
||||||
src = Path("/app/auto_gpt/tests/mocks/mock_commands.py")
|
src = Path("mocks/mock_commands.py")
|
||||||
temp_commands_file = tmp_path / "mock_commands.py"
|
temp_commands_file = tmp_path / "mock_commands.py"
|
||||||
shutil.copyfile(src, temp_commands_file)
|
shutil.copyfile(src, temp_commands_file)
|
||||||
|
|
||||||
|
|||||||
@@ -38,6 +38,7 @@ class TestPromptGenerator(TestCase):
|
|||||||
"label": command_label,
|
"label": command_label,
|
||||||
"name": command_name,
|
"name": command_name,
|
||||||
"args": args,
|
"args": args,
|
||||||
|
"function": None,
|
||||||
}
|
}
|
||||||
self.assertIn(command, self.generator.commands)
|
self.assertIn(command, self.generator.commands)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user