mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-06-16 09:31:37 +00:00
384 lines
15 KiB
Python
384 lines
15 KiB
Python
"""Tests for subprocess HOME handling in profile mode.
|
|
|
|
Hermes state stays profile-scoped through HERMES_HOME. Host subprocesses should
|
|
keep the user's real HOME by default so external CLIs find existing credentials.
|
|
Containers still use the profile home for persistence, and users can explicitly
|
|
opt into profile HOME isolation on the host.
|
|
|
|
See: https://github.com/NousResearch/hermes-agent/issues/25114
|
|
See: https://github.com/NousResearch/hermes-agent/issues/36144
|
|
See: https://github.com/NousResearch/hermes-agent/issues/29015
|
|
"""
|
|
|
|
import os
|
|
import threading
|
|
from pathlib import Path
|
|
|
|
import hermes_constants
|
|
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# get_subprocess_home()
|
|
# ---------------------------------------------------------------------------
|
|
|
|
class TestGetSubprocessHome:
|
|
"""Unit tests for hermes_constants.get_subprocess_home()."""
|
|
|
|
def _host_mode(self, monkeypatch):
|
|
monkeypatch.setattr(hermes_constants, "is_container", lambda: False)
|
|
monkeypatch.delenv("TERMINAL_HOME_MODE", raising=False)
|
|
monkeypatch.delenv("HERMES_REAL_HOME", raising=False)
|
|
|
|
def _container_mode(self, monkeypatch):
|
|
monkeypatch.setattr(hermes_constants, "is_container", lambda: True)
|
|
monkeypatch.delenv("TERMINAL_HOME_MODE", raising=False)
|
|
monkeypatch.delenv("HERMES_REAL_HOME", raising=False)
|
|
|
|
def test_returns_none_when_hermes_home_unset(self, monkeypatch):
|
|
monkeypatch.delenv("HERMES_HOME", raising=False)
|
|
from hermes_constants import get_subprocess_home
|
|
assert get_subprocess_home() is None
|
|
|
|
def test_returns_none_when_home_dir_missing(self, tmp_path, monkeypatch):
|
|
hermes_home = tmp_path / ".hermes"
|
|
hermes_home.mkdir()
|
|
monkeypatch.setenv("HERMES_HOME", str(hermes_home))
|
|
# No home/ subdirectory created
|
|
from hermes_constants import get_subprocess_home
|
|
assert get_subprocess_home() is None
|
|
|
|
def test_host_auto_keeps_real_home_when_profile_home_exists(self, tmp_path, monkeypatch):
|
|
"""Host installs should not hide real ~/.ssh, ~/.gitconfig, ~/.azure, etc."""
|
|
self._host_mode(monkeypatch)
|
|
real_home = tmp_path / "real-home"
|
|
hermes_home = real_home / ".hermes" / "profiles" / "coder"
|
|
profile_home = hermes_home / "home"
|
|
profile_home.mkdir(parents=True)
|
|
monkeypatch.setenv("HOME", str(real_home))
|
|
monkeypatch.setenv("HERMES_HOME", str(hermes_home))
|
|
from hermes_constants import get_subprocess_home
|
|
assert get_subprocess_home() is None
|
|
|
|
def test_container_auto_uses_profile_home_when_home_dir_exists(self, tmp_path, monkeypatch):
|
|
self._container_mode(monkeypatch)
|
|
hermes_home = tmp_path / ".hermes"
|
|
profile_home = hermes_home / "home"
|
|
profile_home.mkdir(parents=True)
|
|
monkeypatch.setenv("HERMES_HOME", str(hermes_home))
|
|
from hermes_constants import get_subprocess_home
|
|
assert get_subprocess_home() == str(profile_home)
|
|
|
|
def test_returns_profile_specific_path(self, tmp_path, monkeypatch):
|
|
"""Explicit profile mode keeps the old per-profile HOME behavior."""
|
|
self._host_mode(monkeypatch)
|
|
profile_dir = tmp_path / ".hermes" / "profiles" / "coder"
|
|
profile_dir.mkdir(parents=True)
|
|
profile_home = profile_dir / "home"
|
|
profile_home.mkdir()
|
|
monkeypatch.setenv("TERMINAL_HOME_MODE", "profile")
|
|
monkeypatch.setenv("HERMES_HOME", str(profile_dir))
|
|
from hermes_constants import get_subprocess_home
|
|
assert get_subprocess_home() == str(profile_home)
|
|
|
|
def test_real_mode_repairs_parent_home_already_pointing_at_profile(self, tmp_path, monkeypatch):
|
|
self._host_mode(monkeypatch)
|
|
profile_dir = tmp_path / ".hermes" / "profiles" / "coder"
|
|
profile_home = profile_dir / "home"
|
|
profile_home.mkdir(parents=True)
|
|
real_home = tmp_path / "real-home"
|
|
real_home.mkdir()
|
|
monkeypatch.setenv("TERMINAL_HOME_MODE", "real")
|
|
monkeypatch.setenv("HERMES_HOME", str(profile_dir))
|
|
monkeypatch.setenv("HOME", str(profile_home))
|
|
monkeypatch.setenv("HERMES_REAL_HOME", str(real_home))
|
|
|
|
from hermes_constants import get_subprocess_home, get_real_home
|
|
|
|
assert get_real_home() == str(real_home)
|
|
assert get_subprocess_home() == str(real_home)
|
|
|
|
def test_real_home_falls_back_to_os_account_when_home_is_profile(self, tmp_path, monkeypatch):
|
|
self._host_mode(monkeypatch)
|
|
profile_dir = tmp_path / ".hermes" / "profiles" / "coder"
|
|
profile_home = profile_dir / "home"
|
|
profile_home.mkdir(parents=True)
|
|
monkeypatch.setenv("HERMES_HOME", str(profile_dir))
|
|
monkeypatch.setenv("HOME", str(profile_home))
|
|
|
|
from hermes_constants import get_real_home
|
|
|
|
assert get_real_home() != str(profile_home)
|
|
|
|
def test_two_profiles_get_different_homes(self, tmp_path, monkeypatch):
|
|
self._container_mode(monkeypatch)
|
|
base = tmp_path / ".hermes" / "profiles"
|
|
for name in ("alpha", "beta"):
|
|
p = base / name
|
|
p.mkdir(parents=True)
|
|
(p / "home").mkdir()
|
|
|
|
from hermes_constants import get_subprocess_home
|
|
|
|
monkeypatch.setenv("HERMES_HOME", str(base / "alpha"))
|
|
home_a = get_subprocess_home()
|
|
|
|
monkeypatch.setenv("HERMES_HOME", str(base / "beta"))
|
|
home_b = get_subprocess_home()
|
|
|
|
assert home_a is not None
|
|
assert home_b is not None
|
|
assert home_a != home_b
|
|
assert home_a.endswith("alpha/home")
|
|
assert home_b.endswith("beta/home")
|
|
|
|
def test_context_override_is_thread_local(self, tmp_path, monkeypatch):
|
|
root = tmp_path / "root"
|
|
profile = tmp_path / "profile"
|
|
root.mkdir()
|
|
profile.mkdir()
|
|
monkeypatch.setenv("HERMES_HOME", str(root))
|
|
|
|
from hermes_constants import (
|
|
get_hermes_home,
|
|
reset_hermes_home_override,
|
|
set_hermes_home_override,
|
|
)
|
|
|
|
ready = threading.Event()
|
|
release = threading.Event()
|
|
seen: list[str] = []
|
|
|
|
def read_from_other_thread():
|
|
ready.set()
|
|
release.wait(timeout=5)
|
|
seen.append(str(get_hermes_home()))
|
|
|
|
thread = threading.Thread(target=read_from_other_thread)
|
|
thread.start()
|
|
assert ready.wait(timeout=5)
|
|
|
|
token = set_hermes_home_override(profile)
|
|
try:
|
|
assert get_hermes_home() == profile
|
|
release.set()
|
|
thread.join(timeout=5)
|
|
finally:
|
|
reset_hermes_home_override(token)
|
|
release.set()
|
|
|
|
assert seen == [str(root)]
|
|
assert get_hermes_home() == root
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# _make_run_env() injection
|
|
# ---------------------------------------------------------------------------
|
|
|
|
class TestMakeRunEnvHomeInjection:
|
|
"""Verify _make_run_env() applies the subprocess HOME policy."""
|
|
|
|
def test_host_auto_preserves_real_home_when_profile_home_exists(self, tmp_path, monkeypatch):
|
|
hermes_home = tmp_path / "hermes"
|
|
hermes_home.mkdir()
|
|
(hermes_home / "home").mkdir()
|
|
real_home = tmp_path / "real-home"
|
|
real_home.mkdir()
|
|
monkeypatch.setattr(hermes_constants, "is_container", lambda: False)
|
|
monkeypatch.setenv("HERMES_HOME", str(hermes_home))
|
|
monkeypatch.setenv("HOME", str(real_home))
|
|
monkeypatch.setenv("PATH", "/usr/bin:/bin")
|
|
|
|
from tools.environments.local import _make_run_env
|
|
result = _make_run_env({})
|
|
|
|
assert result["HOME"] == str(real_home)
|
|
assert result["HERMES_REAL_HOME"] == str(real_home)
|
|
|
|
def test_profile_mode_injects_profile_home_when_profile_home_exists(self, tmp_path, monkeypatch):
|
|
hermes_home = tmp_path / "hermes"
|
|
hermes_home.mkdir()
|
|
(hermes_home / "home").mkdir()
|
|
real_home = tmp_path / "real-home"
|
|
real_home.mkdir()
|
|
monkeypatch.setattr(hermes_constants, "is_container", lambda: False)
|
|
monkeypatch.setenv("TERMINAL_HOME_MODE", "profile")
|
|
monkeypatch.setenv("HERMES_HOME", str(hermes_home))
|
|
monkeypatch.setenv("HOME", str(real_home))
|
|
monkeypatch.setenv("PATH", "/usr/bin:/bin")
|
|
|
|
from tools.environments.local import _make_run_env
|
|
result = _make_run_env({})
|
|
|
|
assert result["HOME"] == str(hermes_home / "home")
|
|
assert result["HERMES_REAL_HOME"] == str(real_home)
|
|
|
|
def test_no_injection_when_home_dir_missing(self, tmp_path, monkeypatch):
|
|
hermes_home = tmp_path / "hermes"
|
|
hermes_home.mkdir()
|
|
# No home/ subdirectory
|
|
monkeypatch.setenv("HERMES_HOME", str(hermes_home))
|
|
monkeypatch.setenv("HOME", "/root")
|
|
monkeypatch.setenv("PATH", "/usr/bin:/bin")
|
|
|
|
from tools.environments.local import _make_run_env
|
|
result = _make_run_env({})
|
|
|
|
assert result["HOME"] == "/root"
|
|
|
|
def test_no_injection_when_hermes_home_unset(self, monkeypatch):
|
|
monkeypatch.delenv("HERMES_HOME", raising=False)
|
|
monkeypatch.setenv("HOME", "/home/user")
|
|
monkeypatch.setenv("PATH", "/usr/bin:/bin")
|
|
|
|
from tools.environments.local import _make_run_env
|
|
result = _make_run_env({})
|
|
|
|
assert result["HOME"] == "/home/user"
|
|
|
|
def test_context_override_bridges_to_subprocess_env(self, tmp_path, monkeypatch):
|
|
monkeypatch.setattr(hermes_constants, "is_container", lambda: True)
|
|
root = tmp_path / "root"
|
|
profile = tmp_path / "profile"
|
|
root.mkdir()
|
|
profile.mkdir()
|
|
(profile / "home").mkdir()
|
|
monkeypatch.setenv("HERMES_HOME", str(root))
|
|
monkeypatch.setenv("HOME", "/root")
|
|
monkeypatch.setenv("PATH", "/usr/bin:/bin")
|
|
|
|
from hermes_constants import reset_hermes_home_override, set_hermes_home_override
|
|
from tools.environments.local import _make_run_env
|
|
|
|
token = set_hermes_home_override(profile)
|
|
try:
|
|
result = _make_run_env({})
|
|
finally:
|
|
reset_hermes_home_override(token)
|
|
|
|
assert result["HERMES_HOME"] == str(profile)
|
|
assert result["HOME"] == str(profile / "home")
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# _sanitize_subprocess_env() injection
|
|
# ---------------------------------------------------------------------------
|
|
|
|
class TestSanitizeSubprocessEnvHomeInjection:
|
|
"""Verify _sanitize_subprocess_env() applies the subprocess HOME policy."""
|
|
|
|
def test_host_auto_preserves_real_home_when_profile_home_exists(self, tmp_path, monkeypatch):
|
|
hermes_home = tmp_path / "hermes"
|
|
hermes_home.mkdir()
|
|
(hermes_home / "home").mkdir()
|
|
real_home = tmp_path / "real-home"
|
|
real_home.mkdir()
|
|
monkeypatch.setattr(hermes_constants, "is_container", lambda: False)
|
|
monkeypatch.setenv("HERMES_HOME", str(hermes_home))
|
|
|
|
base_env = {"HOME": str(real_home), "PATH": "/usr/bin", "USER": "root"}
|
|
from tools.environments.local import _sanitize_subprocess_env
|
|
result = _sanitize_subprocess_env(base_env)
|
|
|
|
assert result["HOME"] == str(real_home)
|
|
assert result["HERMES_REAL_HOME"] == str(real_home)
|
|
|
|
def test_profile_mode_injects_profile_home_when_profile_home_exists(self, tmp_path, monkeypatch):
|
|
hermes_home = tmp_path / "hermes"
|
|
hermes_home.mkdir()
|
|
(hermes_home / "home").mkdir()
|
|
real_home = tmp_path / "real-home"
|
|
real_home.mkdir()
|
|
monkeypatch.setattr(hermes_constants, "is_container", lambda: False)
|
|
monkeypatch.setenv("TERMINAL_HOME_MODE", "profile")
|
|
monkeypatch.setenv("HERMES_HOME", str(hermes_home))
|
|
|
|
base_env = {"HOME": str(real_home), "PATH": "/usr/bin", "USER": "root"}
|
|
from tools.environments.local import _sanitize_subprocess_env
|
|
result = _sanitize_subprocess_env(base_env)
|
|
|
|
assert result["HOME"] == str(hermes_home / "home")
|
|
assert result["HERMES_REAL_HOME"] == str(real_home)
|
|
|
|
def test_no_injection_when_home_dir_missing(self, tmp_path, monkeypatch):
|
|
hermes_home = tmp_path / "hermes"
|
|
hermes_home.mkdir()
|
|
monkeypatch.setenv("HERMES_HOME", str(hermes_home))
|
|
|
|
base_env = {"HOME": "/root", "PATH": "/usr/bin"}
|
|
from tools.environments.local import _sanitize_subprocess_env
|
|
result = _sanitize_subprocess_env(base_env)
|
|
|
|
assert result["HOME"] == "/root"
|
|
|
|
def test_context_override_bridges_to_background_env(self, tmp_path, monkeypatch):
|
|
monkeypatch.setattr(hermes_constants, "is_container", lambda: True)
|
|
root = tmp_path / "root"
|
|
profile = tmp_path / "profile"
|
|
root.mkdir()
|
|
profile.mkdir()
|
|
(profile / "home").mkdir()
|
|
monkeypatch.setenv("HERMES_HOME", str(root))
|
|
|
|
base_env = {"HOME": "/root", "PATH": "/usr/bin"}
|
|
from hermes_constants import reset_hermes_home_override, set_hermes_home_override
|
|
from tools.environments.local import _sanitize_subprocess_env
|
|
|
|
token = set_hermes_home_override(profile)
|
|
try:
|
|
result = _sanitize_subprocess_env(base_env)
|
|
finally:
|
|
reset_hermes_home_override(token)
|
|
|
|
assert result["HERMES_HOME"] == str(profile)
|
|
assert result["HOME"] == str(profile / "home")
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Profile bootstrap
|
|
# ---------------------------------------------------------------------------
|
|
|
|
class TestProfileBootstrap:
|
|
"""Verify new profiles get a home/ subdirectory."""
|
|
|
|
def test_profile_dirs_includes_home(self):
|
|
from hermes_cli.profiles import _PROFILE_DIRS
|
|
assert "home" in _PROFILE_DIRS
|
|
|
|
def test_create_profile_bootstraps_home_dir(self, tmp_path, monkeypatch):
|
|
"""create_profile() should create home/ inside the profile dir."""
|
|
home = tmp_path / ".hermes"
|
|
home.mkdir()
|
|
monkeypatch.setattr(Path, "home", lambda: tmp_path)
|
|
monkeypatch.setenv("HERMES_HOME", str(home))
|
|
|
|
from hermes_cli.profiles import create_profile
|
|
profile_dir = create_profile("testbot", no_alias=True)
|
|
assert (profile_dir / "home").is_dir()
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Python process HOME unchanged
|
|
# ---------------------------------------------------------------------------
|
|
|
|
class TestPythonProcessUnchanged:
|
|
"""Confirm the Python process's own HOME is never modified."""
|
|
|
|
def test_path_home_unchanged_after_subprocess_home_resolved(
|
|
self, tmp_path, monkeypatch
|
|
):
|
|
hermes_home = tmp_path / "hermes"
|
|
hermes_home.mkdir()
|
|
(hermes_home / "home").mkdir()
|
|
monkeypatch.setenv("HERMES_HOME", str(hermes_home))
|
|
|
|
original_home = os.environ.get("HOME")
|
|
original_path_home = str(Path.home())
|
|
|
|
from hermes_constants import get_subprocess_home
|
|
sub_home = get_subprocess_home()
|
|
|
|
# Resolving subprocess HOME must not mutate the Python process env.
|
|
assert sub_home in (None, str(hermes_home / "home"), original_home)
|
|
assert os.environ.get("HOME") == original_home
|
|
assert str(Path.home()) == original_path_home
|