mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-07-21 16:18:55 +00:00
82 lines
2.4 KiB
Python
82 lines
2.4 KiB
Python
"""Tests for explicit ownership by a wrapped external gateway supervisor."""
|
|
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
import hermes_cli.gateway as gateway
|
|
|
|
|
|
def _clear_native_supervisor_markers(monkeypatch):
|
|
monkeypatch.delenv("INVOCATION_ID", raising=False)
|
|
monkeypatch.delenv("HERMES_S6_SUPERVISED_CHILD", raising=False)
|
|
monkeypatch.setenv("XPC_SERVICE_NAME", "0")
|
|
|
|
|
|
def test_external_marker_identifies_supervisor_process(monkeypatch):
|
|
_clear_native_supervisor_markers(monkeypatch)
|
|
monkeypatch.setenv(gateway.EXTERNAL_GATEWAY_SUPERVISOR_ENV, "1")
|
|
|
|
assert gateway._running_under_gateway_supervisor() is True
|
|
|
|
|
|
def test_gateway_run_external_supervisor_flag_marks_process(monkeypatch):
|
|
monkeypatch.delenv(gateway.EXTERNAL_GATEWAY_SUPERVISOR_ENV, raising=False)
|
|
monkeypatch.setattr(
|
|
gateway, "_maybe_redirect_run_to_s6_supervision", lambda _args: False
|
|
)
|
|
observed = []
|
|
monkeypatch.setattr(
|
|
gateway,
|
|
"run_gateway",
|
|
lambda *_args, **_kwargs: observed.append(
|
|
gateway.os.environ.get(gateway.EXTERNAL_GATEWAY_SUPERVISOR_ENV)
|
|
),
|
|
)
|
|
|
|
gateway._gateway_command_inner(
|
|
SimpleNamespace(gateway_command="run", external_supervisor=True)
|
|
)
|
|
|
|
assert observed == ["1"]
|
|
|
|
|
|
def test_update_hands_external_supervisor_gateway_back_without_watcher(monkeypatch):
|
|
monkeypatch.setattr(
|
|
gateway,
|
|
"_capture_gateway_argv",
|
|
lambda _pid: [
|
|
"python",
|
|
"-m",
|
|
"hermes_cli.main",
|
|
"gateway",
|
|
"run",
|
|
"--external-supervisor",
|
|
],
|
|
)
|
|
monkeypatch.setattr(
|
|
gateway,
|
|
"launch_detached_profile_gateway_restart",
|
|
lambda *_args: pytest.fail("detached watcher must not be launched"),
|
|
)
|
|
|
|
assert gateway._prepare_profile_gateway_update_restart("work", 1234) == (
|
|
"external-supervisor"
|
|
)
|
|
|
|
|
|
def test_update_keeps_detached_restart_for_ordinary_foreground_gateway(monkeypatch):
|
|
monkeypatch.setattr(
|
|
gateway,
|
|
"_capture_gateway_argv",
|
|
lambda _pid: ["python", "-m", "hermes_cli.main", "gateway", "run"],
|
|
)
|
|
calls = []
|
|
monkeypatch.setattr(
|
|
gateway,
|
|
"launch_detached_profile_gateway_restart",
|
|
lambda profile, pid: calls.append((profile, pid)) or True,
|
|
)
|
|
|
|
assert gateway._prepare_profile_gateway_update_restart("work", 1234) == "detached"
|
|
assert calls == [("work", 1234)]
|