aljaz/goose
1
0
Fork 0
mirror of https://github.com/aljazceru/goose.git synced 2026-01-03 22:44:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

feat: allow plugins to set custom patterns for dangerous commands (#281)

Browse Source
This commit is contained in:
Alistair Gray
2024-12-10 12:30:19 -05:00
committed by GitHub
parent 64a1ee4a7c
commit db11a22a6f
5 changed files with 62 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/utils/test_check_shell_command.py
View File

@@ -1,5 +1,5 @@
import pytest
from goose.utils.shell import is_dangerous_command
from goose.utils.command_checker import add_dangerous_command_patterns, is_dangerous_command
@pytest.mark.parametrize(
@@ -40,3 +40,11 @@ def test_dangerous_commands(command):
)
def test_safe_commands(command):
assert not is_dangerous_command(command)
def test_add_dangerous_patterns():
add_dangerous_command_patterns(["echo hello"])
assert is_dangerous_command("echo hello")
# and that the original commands are still flagged
assert is_dangerous_command("rm -rf /")