Maybe fix tests, fix safe_path function.

This commit is contained in:
BillSchumacher
2023-04-16 23:18:29 -05:00
parent 708374d95b
commit 23d3dafc51
3 changed files with 4 additions and 3 deletions

View File

@@ -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}'."
) )

View File

@@ -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)

View 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)