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()
if not joined_path.is_relative_to(base):
if not str(joined_path.absolute()).startswith(str(base.absolute())):
raise ValueError(
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):
"""Test that the registry can import a module with mock command plugins."""
registry = CommandRegistry()
mock_commands_module = "auto_gpt.tests.mocks.mock_commands"
mock_commands_module = "tests.mocks.mock_commands"
registry.import_commands(mock_commands_module)
@@ -155,7 +155,7 @@ class TestCommandRegistry:
registry = CommandRegistry()
# 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"
shutil.copyfile(src, temp_commands_file)

View File

@@ -38,6 +38,7 @@ class TestPromptGenerator(TestCase):
"label": command_label,
"name": command_name,
"args": args,
"function": None,
}
self.assertIn(command, self.generator.commands)