dac4c60fbb
Widen the container-backend refusal salvaged from #113530 to the sibling surfaces that render the same host checkpoints: the messaging gateway's /rollback (restore refused, bare listing prefixed with the reason) and /diff session, and the CLI's /diff session. The gateway arm follows the CLI's "default" classification, i.e. the configured terminal backend. Drop the thin _checkpoint_container_backend wrapper in favour of the container_backend_for_task predicate it wrapped, trim the salvaged suite to two invariant tests (one per class: no host store touched by a container task; every surface refuses a host restore/diff from a container session, with a local control), and update the docs. Co-authored-by: fangliquan <fangliquan@qq.com>
116 lines
3.2 KiB
Python
116 lines
3.2 KiB
Python
"""Tests for the CLI ``/diff`` command handler.
|
|
|
|
``/diff`` shows git changes in the working directory (unstaged + untracked by
|
|
default; ``staged``/``all`` modes) and ``/diff session`` shows the cumulative
|
|
checkpoint-baseline diff of everything Hermes changed. These drive the mixin
|
|
handler against real git repos (default modes) and a stubbed checkpoint
|
|
manager (session mode), asserting rendering, ``--stat``, and graceful
|
|
degradation.
|
|
"""
|
|
|
|
import contextlib
|
|
import io
|
|
import shutil
|
|
import subprocess
|
|
|
|
import pytest
|
|
|
|
from hermes_cli.cli_commands_mixin import CLICommandsMixin
|
|
|
|
requires_git = pytest.mark.skipif(
|
|
shutil.which("git") is None, reason="git required"
|
|
)
|
|
|
|
|
|
class _Console:
|
|
def __init__(self, sink):
|
|
self._sink = sink
|
|
|
|
def print(self, obj, **kwargs):
|
|
self._sink.write(getattr(obj, "plain", str(obj)) + "\n")
|
|
|
|
|
|
class _Mgr:
|
|
def __init__(self, result, enabled=True):
|
|
self.enabled = enabled
|
|
self._result = result
|
|
self.calls = []
|
|
|
|
def unsupported_backend_reason(self, task_id="default"):
|
|
return None
|
|
|
|
def session_diff(self, cwd):
|
|
self.calls.append(cwd)
|
|
return self._result
|
|
|
|
|
|
class _Agent:
|
|
def __init__(self, mgr):
|
|
self._checkpoint_mgr = mgr
|
|
|
|
|
|
class _Stub(CLICommandsMixin):
|
|
def __init__(self, agent=None):
|
|
self.agent = agent
|
|
|
|
|
|
def _run(stub, command):
|
|
buf = io.StringIO()
|
|
stub.console = _Console(buf)
|
|
with contextlib.redirect_stdout(buf):
|
|
stub._handle_diff_command(command)
|
|
return buf.getvalue()
|
|
|
|
|
|
def _git(repo, *args):
|
|
subprocess.run(["git", *args], cwd=repo, check=True, capture_output=True,
|
|
env={"GIT_AUTHOR_NAME": "t", "GIT_AUTHOR_EMAIL": "t@t",
|
|
"GIT_COMMITTER_NAME": "t", "GIT_COMMITTER_EMAIL": "t@t",
|
|
"HOME": str(repo),
|
|
"PATH": __import__("os").environ["PATH"]})
|
|
|
|
|
|
@pytest.fixture()
|
|
def repo(tmp_path, monkeypatch):
|
|
d = tmp_path / "repo"
|
|
d.mkdir()
|
|
_git(d, "init", "-q")
|
|
(d / "main.py").write_text("print('hello')\n")
|
|
_git(d, "add", "-A")
|
|
_git(d, "commit", "-q", "-m", "init")
|
|
monkeypatch.setenv("TERMINAL_CWD", str(d))
|
|
return d
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Default (working-tree) mode — real git
|
|
# ---------------------------------------------------------------------------
|
|
|
|
@requires_git
|
|
def test_diff_clean_repo_reports_no_changes(repo):
|
|
out = _run(_Stub(), "/diff")
|
|
assert "No changes" in out
|
|
|
|
|
|
@requires_git
|
|
def test_diff_non_git_directory_is_graceful(tmp_path, monkeypatch):
|
|
plain = tmp_path / "plain"
|
|
plain.mkdir()
|
|
monkeypatch.setenv("TERMINAL_CWD", str(plain))
|
|
out = _run(_Stub(), "/diff")
|
|
assert "not a git repository" in out.lower()
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Session mode — stubbed checkpoint manager
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
def test_diff_session_empty_reports_no_changes(tmp_path, monkeypatch):
|
|
monkeypatch.setenv("TERMINAL_CWD", str(tmp_path))
|
|
mgr = _Mgr({"success": True, "stat": "", "diff": "", "empty": True})
|
|
out = _run(_Stub(_Agent(mgr)), "/diff session")
|
|
assert "No changes" in out
|
|
|
|
|