mirror of
https://github.com/NCBM/plyngent.git
synced 2026-07-23 05:55:16 +08:00
NCBM
510dd67a54
Add classify_danger for delete/move/overwrite paths, get_path_denylist, and clearer path/command denial messages.
31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
from plyngent.tools.danger import classify_danger
|
|
|
|
|
|
def test_classify_delete_and_move() -> None:
|
|
assert classify_danger("delete_path", {"path": "a.txt"}) == "delete path 'a.txt'"
|
|
assert "recursively" in (classify_danger("delete_path", {"path": "d", "recursive": True}) or "")
|
|
assert "move" in (classify_danger("move_path", {"src": "a", "dst": "b"}) or "")
|
|
|
|
|
|
def test_classify_copy_overwrite_only() -> None:
|
|
assert classify_danger("copy_path", {"src": "a", "dst": "b"}) is None
|
|
assert "overwrite" in (
|
|
classify_danger("copy_path", {"src": "a", "dst": "b", "overwrite": True}) or ""
|
|
)
|
|
|
|
|
|
def test_classify_write_file_overwrite(workspace: object) -> None:
|
|
assert isinstance(workspace, Path)
|
|
_ = (workspace / "x.txt").write_text("old", encoding="utf-8")
|
|
assert "overwrite" in (classify_danger("write_file", {"path": "x.txt", "content": "n"}) or "")
|
|
assert classify_danger("write_file", {"path": "new.txt", "content": "n"}) is None
|
|
|
|
|
|
def test_classify_safe_tools() -> None:
|
|
assert classify_danger("read_file", {"path": "a"}) is None
|
|
assert classify_danger("run_command", {"command": ["echo", "hi"]}) is None
|