mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-04-27 01:11:40 +00:00
4b16341975
Rewrite all import statements, patch() targets, sys.modules keys, importlib.import_module() strings, and subprocess -m references to use hermes_agent.* paths. Strip sys.path.insert hacks from production code (rely on editable install). Update COMPONENT_PREFIXES for logger filtering. Fix 3 hardcoded getLogger() calls to use __name__. Update transport and tool registry discovery paths. Update plugin module path strings. Add legacy process-name patterns for gateway PID detection. Add main() to skills_sync for console_script entry point. Fix _get_bundled_dir() path traversal after move. Part of #14182, #14183
144 lines
3.6 KiB
Python
144 lines
3.6 KiB
Python
"""Tests for _ThreadedProcessHandle — the adapter for SDK backends."""
|
|
|
|
import threading
|
|
import time
|
|
|
|
from hermes_agent.backends.base import _ThreadedProcessHandle
|
|
|
|
|
|
class TestBasicExecution:
|
|
def test_successful_execution(self):
|
|
def exec_fn():
|
|
return ("hello world", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn)
|
|
handle.wait(timeout=5)
|
|
|
|
assert handle.returncode == 0
|
|
output = handle.stdout.read()
|
|
assert "hello world" in output
|
|
|
|
def test_nonzero_exit_code(self):
|
|
def exec_fn():
|
|
return ("error occurred", 42)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn)
|
|
handle.wait(timeout=5)
|
|
|
|
assert handle.returncode == 42
|
|
output = handle.stdout.read()
|
|
assert "error occurred" in output
|
|
|
|
def test_exception_in_exec_fn(self):
|
|
def exec_fn():
|
|
raise RuntimeError("boom")
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn)
|
|
handle.wait(timeout=5)
|
|
|
|
assert handle.returncode == 1
|
|
|
|
def test_empty_output(self):
|
|
def exec_fn():
|
|
return ("", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn)
|
|
handle.wait(timeout=5)
|
|
|
|
assert handle.returncode == 0
|
|
output = handle.stdout.read()
|
|
assert output == ""
|
|
|
|
|
|
class TestPolling:
|
|
def test_poll_returns_none_while_running(self):
|
|
event = threading.Event()
|
|
|
|
def exec_fn():
|
|
event.wait(timeout=5)
|
|
return ("done", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn)
|
|
assert handle.poll() is None
|
|
|
|
event.set()
|
|
handle.wait(timeout=5)
|
|
assert handle.poll() == 0
|
|
|
|
def test_poll_returns_returncode_when_done(self):
|
|
def exec_fn():
|
|
return ("ok", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn)
|
|
handle.wait(timeout=5)
|
|
assert handle.poll() == 0
|
|
|
|
|
|
class TestCancelFn:
|
|
def test_cancel_fn_called_on_kill(self):
|
|
called = threading.Event()
|
|
|
|
def cancel():
|
|
called.set()
|
|
|
|
def exec_fn():
|
|
time.sleep(10)
|
|
return ("", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn, cancel_fn=cancel)
|
|
handle.kill()
|
|
assert called.is_set()
|
|
|
|
def test_cancel_fn_none_is_safe(self):
|
|
def exec_fn():
|
|
return ("ok", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn, cancel_fn=None)
|
|
handle.kill() # should not raise
|
|
handle.wait(timeout=5)
|
|
assert handle.returncode == 0
|
|
|
|
def test_cancel_fn_exception_swallowed(self):
|
|
def cancel():
|
|
raise RuntimeError("cancel failed")
|
|
|
|
def exec_fn():
|
|
return ("ok", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn, cancel_fn=cancel)
|
|
handle.kill() # should not raise despite cancel raising
|
|
handle.wait(timeout=5)
|
|
|
|
|
|
class TestStdoutPipe:
|
|
def test_stdout_is_readable(self):
|
|
def exec_fn():
|
|
return ("line1\nline2\nline3\n", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn)
|
|
handle.wait(timeout=5)
|
|
|
|
lines = handle.stdout.readlines()
|
|
assert len(lines) == 3
|
|
assert lines[0] == "line1\n"
|
|
|
|
def test_stdout_iterable(self):
|
|
def exec_fn():
|
|
return ("a\nb\nc\n", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn)
|
|
handle.wait(timeout=5)
|
|
|
|
collected = list(handle.stdout)
|
|
assert len(collected) == 3
|
|
|
|
def test_unicode_output(self):
|
|
def exec_fn():
|
|
return ("hello 世界 🌍\n", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn)
|
|
handle.wait(timeout=5)
|
|
|
|
output = handle.stdout.read()
|
|
assert "世界" in output
|
|
assert "🌍" in output
|