mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-07-30 19:09:28 +00:00
64beb25a35
Streamed responses no longer insert real newlines at terminal width — logical lines are emitted whole and the terminal soft-wraps them, so highlight-copy rejoins the full line (emulators only keep linebreaks the app actually printed). This is the CLI equivalent of the TUI's selection copy, which reads logical source lines from its screen buffer. TTFT perception is preserved by mirroring the unfinished line's tail into the spinner status text instead of chunk-printing. /copy now prefers OSC 52 when running over SSH (SSH_CONNECTION / SSH_TTY / SSH_CLIENT) — native tools there write the REMOTE clipboard, which is never what the user wants. The CLI's OSC 52 writer also gains tmux/screen DCS passthrough wrapping, mirroring the TUI's wrapForMultiplexer. Fixes #31528 for the CLI surface. Sabotage-verified: restoring the old chunk emitter fails 3 of the new tests (hard-wrap detection, spinner mirror, unbreakable-run split).
124 lines
4.7 KiB
Python
124 lines
4.7 KiB
Python
"""Tests for native clipboard text write (hermes_cli/clipboard.py).
|
|
|
|
Mirrors the TUI's writeClipboardText fallback chain: pbcopy /
|
|
PowerShell Set-Clipboard / wl-copy / xclip / xsel, with OSC 52 left to
|
|
the caller when every backend fails.
|
|
"""
|
|
import base64
|
|
import subprocess
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from hermes_cli import clipboard as clip
|
|
|
|
|
|
def _completed(returncode=0):
|
|
return subprocess.CompletedProcess(args=[], returncode=returncode)
|
|
|
|
|
|
def test_darwin_uses_pbcopy():
|
|
with patch.object(clip.sys, "platform", "darwin"), \
|
|
patch.object(clip.subprocess, "run", return_value=_completed()) as run:
|
|
assert clip.write_clipboard_text("hello") is True
|
|
argv = run.call_args[0][0]
|
|
assert argv == ["pbcopy"]
|
|
assert run.call_args[1]["input"] == b"hello"
|
|
|
|
|
|
def test_windows_uses_powershell_base64():
|
|
with patch.object(clip.sys, "platform", "win32"), \
|
|
patch.object(clip.subprocess, "run", return_value=_completed()) as run:
|
|
assert clip.write_clipboard_text("héllo 🎉") is True
|
|
argv = run.call_args[0][0]
|
|
assert argv[0] == "powershell"
|
|
script = argv[-1]
|
|
b64 = base64.b64encode("héllo 🎉".encode("utf-8")).decode("ascii")
|
|
assert b64 in script
|
|
assert "Set-Clipboard" in script
|
|
|
|
|
|
def test_linux_falls_through_backends_until_success():
|
|
calls = []
|
|
|
|
def fake_run(argv, **kwargs):
|
|
calls.append(argv[0])
|
|
# xclip fails, xsel succeeds
|
|
return _completed(returncode=0 if argv[0] == "xsel" else 1)
|
|
|
|
with patch.object(clip.sys, "platform", "linux"), \
|
|
patch.object(clip, "_is_wsl", return_value=False), \
|
|
patch.dict(clip.os.environ, {}, clear=False), \
|
|
patch.object(clip.os.environ, "get", lambda k, d=None: None), \
|
|
patch.object(clip.subprocess, "run", side_effect=fake_run):
|
|
assert clip.write_clipboard_text("x") is True
|
|
assert calls == ["xclip", "xsel"]
|
|
|
|
|
|
def test_returns_false_when_all_backends_fail():
|
|
with patch.object(clip.sys, "platform", "linux"), \
|
|
patch.object(clip, "_is_wsl", return_value=False), \
|
|
patch.object(clip.os.environ, "get", lambda k, d=None: None), \
|
|
patch.object(clip.subprocess, "run", side_effect=FileNotFoundError):
|
|
assert clip.write_clipboard_text("x") is False
|
|
|
|
|
|
def test_wayland_prefers_wl_copy():
|
|
with patch.object(clip.sys, "platform", "linux"), \
|
|
patch.object(clip, "_is_wsl", return_value=False), \
|
|
patch.object(clip.os.environ, "get",
|
|
lambda k, d=None: ":0" if k == "WAYLAND_DISPLAY" else None), \
|
|
patch.object(clip.subprocess, "run", return_value=_completed()) as run:
|
|
assert clip.write_clipboard_text("x") is True
|
|
assert run.call_args[0][0][0] == "wl-copy"
|
|
|
|
|
|
def test_is_remote_shell_session_detects_ssh_env():
|
|
assert clip.is_remote_shell_session({"SSH_CONNECTION": "1.2.3.4 5 6.7.8.9 22"})
|
|
assert clip.is_remote_shell_session({"SSH_TTY": "/dev/pts/0"})
|
|
assert clip.is_remote_shell_session({"SSH_CLIENT": "1.2.3.4 5 22"})
|
|
assert not clip.is_remote_shell_session({})
|
|
assert not clip.is_remote_shell_session({"TERM": "xterm-256color"})
|
|
|
|
|
|
class TestOsc52MultiplexerWrapping:
|
|
"""CLI _write_osc52_clipboard must wrap for tmux/screen passthrough
|
|
(mirrors ui-tui/src/lib/osc52.ts wrapForMultiplexer)."""
|
|
|
|
def _capture_seq(self, env):
|
|
import io
|
|
from unittest.mock import patch as _patch
|
|
from cli import HermesCLI
|
|
|
|
cli_obj = HermesCLI.__new__(HermesCLI)
|
|
cli_obj._app = None
|
|
buf = io.StringIO()
|
|
with _patch.dict(clip.os.environ, env, clear=False), \
|
|
_patch("cli.sys.stdout", buf):
|
|
for var in ("TMUX", "STY"):
|
|
if var not in env:
|
|
clip.os.environ.pop(var, None)
|
|
cli_obj._write_osc52_clipboard("hello")
|
|
return buf.getvalue()
|
|
|
|
def test_tmux_wraps_in_dcs_passthrough(self, monkeypatch):
|
|
monkeypatch.setenv("TMUX", "/tmp/tmux-123/default,1,0")
|
|
monkeypatch.delenv("STY", raising=False)
|
|
seq = self._capture_seq({"TMUX": "/tmp/tmux-123/default,1,0"})
|
|
assert seq.startswith("\x1bPtmux;")
|
|
assert "]52;c;" in seq
|
|
assert seq.endswith("\x1b\\")
|
|
|
|
def test_raw_osc52_outside_multiplexers(self, monkeypatch):
|
|
monkeypatch.delenv("TMUX", raising=False)
|
|
monkeypatch.delenv("STY", raising=False)
|
|
seq = self._capture_seq({})
|
|
assert seq.startswith("\x1b]52;c;")
|
|
assert seq.endswith("\x07")
|
|
|
|
def test_screen_wraps_in_dcs(self, monkeypatch):
|
|
monkeypatch.delenv("TMUX", raising=False)
|
|
monkeypatch.setenv("STY", "12345.pts-0.host")
|
|
seq = self._capture_seq({"STY": "12345.pts-0.host"})
|
|
assert seq.startswith("\x1bP\x1b]52;c;")
|
|
assert seq.endswith("\x1b\\")
|