aljaz/goose
1
0
Fork 0
mirror of https://github.com/aljazceru/goose.git synced 2025-12-28 11:34:26 +01:00
Code Issues Packages Projects Releases Wiki Activity

refactor: improve safety rails speed and prompt (#45)

Browse Source
This commit is contained in:
Bradley Axen
2024-09-04 22:05:53 -07:00
committed by GitHub
parent 72d927f84f
commit a579e21037
6 changed files with 63 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
tests/utils/test_check_shell_command.py
View File

@@ -15,13 +15,26 @@ from goose.utils.check_shell_command import is_dangerous_command
"systemctl stop nginx",
"reboot",
"shutdown now",
"echo hello > ~/.bashrc",
"cat ~/.hello.txt",
"cat ~/.config/example.txt",
],
)
def test_dangerous_commands(command):
assert is_dangerous_command(command)
@pytest.mark.parametrize("command", ["ls -la", 'echo "Hello World"', "cp ~/folder/file.txt /tmp/"])
@pytest.mark.parametrize(
"command",
[
"ls -la",
'echo "Hello World"',
"cp ~/folder/file.txt /tmp/",
"echo hello > ~/toplevel/sublevel.txt",
"cat hello.txt",
"cat ~/config/example.txt",
"ls -la path/to/visible/file",
"echo 'file.with.dot.txt'",
],
)
def test_safe_commands(command):
assert not is_dangerous_command(command)