mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-05-30 06:41:51 +00:00
66827f8947
Remove unused imports (F401) and duplicate/shadowed import redefinitions (F811) across the codebase using ruff's safe autofixes. No behavioral changes -- imports only. - ~1400 safe autofixes applied across 644 files (net -1072 lines) - __init__.py re-exports preserved (excluded from F401 removal so public re-export surfaces stay intact) - Re-exports that are imported or monkeypatched by tests but look unused in their defining module are kept with explicit # noqa: F401 (gateway/run.py load_dotenv; run_agent re-exports from agent.message_sanitization, agent.context_compressor, agent.retry_utils, agent.prompt_builder, agent.process_bootstrap, agent.codex_responses_adapter) - Unsafe F841 (unused-variable) fixes deliberately skipped -- those can change behavior when the RHS has side effects - ruff lints remain disabled in pyproject.toml (only PLW1514 is selected); this is a one-time cleanup, not a config change Verification: - python -m compileall: clean - pytest --collect-only: all 27161 tests collect (zero import errors) - core entry points import clean (run_agent, model_tools, cli, toolsets, hermes_state, batch_runner, gateway) - static scan: every name any test imports directly from an edited module still resolves
128 lines
4.9 KiB
Python
128 lines
4.9 KiB
Python
"""Tests for _is_write_denied() — verifies deny list blocks sensitive paths on all platforms."""
|
|
|
|
import os
|
|
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
from tools.file_operations import _is_write_denied
|
|
|
|
|
|
class TestWriteDenyExactPaths:
|
|
def test_etc_shadow(self):
|
|
assert _is_write_denied("/etc/shadow") is True
|
|
|
|
def test_etc_passwd(self):
|
|
assert _is_write_denied("/etc/passwd") is True
|
|
|
|
def test_etc_sudoers(self):
|
|
assert _is_write_denied("/etc/sudoers") is True
|
|
|
|
def test_ssh_authorized_keys(self):
|
|
assert _is_write_denied("~/.ssh/authorized_keys") is True
|
|
|
|
def test_ssh_id_rsa(self):
|
|
path = os.path.join(str(Path.home()), ".ssh", "id_rsa")
|
|
assert _is_write_denied(path) is True
|
|
|
|
def test_ssh_id_ed25519(self):
|
|
path = os.path.join(str(Path.home()), ".ssh", "id_ed25519")
|
|
assert _is_write_denied(path) is True
|
|
|
|
def test_netrc(self):
|
|
path = os.path.join(str(Path.home()), ".netrc")
|
|
assert _is_write_denied(path) is True
|
|
|
|
def test_hermes_env(self):
|
|
# ``.env`` under the active HERMES_HOME (profile-aware, not just
|
|
# ``~/.hermes``) must be write-denied. The hermetic test conftest
|
|
# points HERMES_HOME at a tempdir — resolve via get_hermes_home()
|
|
# to match the denylist.
|
|
from hermes_constants import get_hermes_home
|
|
path = str(get_hermes_home() / ".env")
|
|
assert _is_write_denied(path) is True
|
|
|
|
def test_hermes_root_env_when_running_under_profile(self, tmp_path, monkeypatch):
|
|
"""Top-level ``<root>/.env`` stays write-denied even when running under
|
|
a profile (#15981).
|
|
|
|
Before the fix, ``build_write_denied_paths`` only added
|
|
``<active_profile>/.env`` to the deny list, so the global
|
|
``~/.hermes/.env`` (whose credentials are inherited by every profile)
|
|
could be silently overwritten by ``write_file`` while a profile was
|
|
active.
|
|
"""
|
|
root = tmp_path / "hermes_root"
|
|
profile_home = root / "profiles" / "coder"
|
|
profile_home.mkdir(parents=True)
|
|
global_env = root / ".env"
|
|
global_env.write_text("OPENAI_API_KEY=sk-real\n")
|
|
|
|
monkeypatch.setenv("HERMES_HOME", str(profile_home))
|
|
|
|
# Sanity check: HERMES_HOME does point to the profile dir, not the root.
|
|
from hermes_constants import get_hermes_home, get_default_hermes_root
|
|
assert get_hermes_home() == profile_home
|
|
assert get_default_hermes_root() == root
|
|
|
|
assert _is_write_denied(str(global_env)) is True
|
|
|
|
def test_shell_profiles(self):
|
|
home = str(Path.home())
|
|
for name in [".bashrc", ".zshrc", ".profile", ".bash_profile", ".zprofile"]:
|
|
assert _is_write_denied(os.path.join(home, name)) is True, f"{name} should be denied"
|
|
|
|
def test_package_manager_configs(self):
|
|
home = str(Path.home())
|
|
for name in [".npmrc", ".pypirc", ".pgpass"]:
|
|
assert _is_write_denied(os.path.join(home, name)) is True, f"{name} should be denied"
|
|
|
|
|
|
class TestWriteDenyPrefixes:
|
|
def test_ssh_prefix(self):
|
|
path = os.path.join(str(Path.home()), ".ssh", "some_key")
|
|
assert _is_write_denied(path) is True
|
|
|
|
def test_aws_prefix(self):
|
|
path = os.path.join(str(Path.home()), ".aws", "credentials")
|
|
assert _is_write_denied(path) is True
|
|
|
|
def test_gnupg_prefix(self):
|
|
path = os.path.join(str(Path.home()), ".gnupg", "secring.gpg")
|
|
assert _is_write_denied(path) is True
|
|
|
|
def test_kube_prefix(self):
|
|
path = os.path.join(str(Path.home()), ".kube", "config")
|
|
assert _is_write_denied(path) is True
|
|
|
|
def test_sudoers_d_prefix(self):
|
|
assert _is_write_denied("/etc/sudoers.d/custom") is True
|
|
|
|
def test_systemd_prefix(self, tmp_path):
|
|
# On NixOS, /etc/systemd is a symlink into /nix/store, so
|
|
# realpath() resolves it to a store path that doesn't match
|
|
# the /etc/systemd/ prefix. Build a real directory tree so
|
|
# realpath is a no-op and prefix matching works.
|
|
fake_etc = tmp_path / "etc" / "systemd" / "system"
|
|
fake_etc.mkdir(parents=True)
|
|
target = str(fake_etc / "evil.service")
|
|
# Patch the prefix builder to include our tmp_path prefix
|
|
import agent.file_safety as _fs
|
|
_orig = _fs.build_write_denied_prefixes
|
|
_extra_prefix = str(tmp_path / "etc" / "systemd") + os.sep
|
|
def _patched(home):
|
|
return _orig(home) + [_extra_prefix]
|
|
with patch.object(_fs, "build_write_denied_prefixes", _patched):
|
|
assert _is_write_denied(target) is True
|
|
|
|
|
|
class TestWriteAllowed:
|
|
def test_tmp_file(self):
|
|
assert _is_write_denied("/tmp/safe_file.txt") is False
|
|
|
|
def test_project_file(self):
|
|
assert _is_write_denied("/home/user/project/main.py") is False
|
|
|
|
def test_hermes_config_not_env(self):
|
|
path = os.path.join(str(Path.home()), ".hermes", "config.yaml")
|
|
assert _is_write_denied(path) is False
|