Files
wehub-resource-sync b4fbd6fe9f
Deploy Site / deploy-vercel (push) Has been skipped
Deploy Site / deploy-docs (push) Has been skipped
Build Skills Index / build-index (push) Has been skipped
Build Skills Index / trigger-deploy (push) Waiting to run
CI / Deny unrelated histories (push) Has been skipped
CI / CI timing report (push) Blocked by required conditions
CI / Detect affected areas (push) Successful in 27m35s
CI / OSV scan (push) Failing after 4s
CI / Build&Test Docker image (push) Successful in 9s
CI / Supply-chain scan (push) Has been skipped
CI / Lint Docker scripts (push) Failing after 5m13s
CI / Check contributors (push) Failing after 12m8s
CI / Docs Site (push) Failing after 12m8s
CI / TypeScript (push) Failing after 12m8s
CI / Python lints (push) Failing after 12m9s
CI / Python tests (push) Failing after 12m9s
CI / Check uv.lock (push) Failing after 23m22s
CI / All required checks pass (push) Waiting to run
chore: import upstream snapshot with attribution
2026-07-13 11:56:03 +08:00

625 lines
17 KiB
Python

import sys
def test_sessions_export_md_writes_single_session(monkeypatch, tmp_path, capsys):
import hermes_cli.main as main_mod
import hermes_state
captured = {}
class FakeDB:
def resolve_session_id(self, session_id):
captured["resolved_from"] = session_id
return "20260706_123456_abcd1234"
def export_session(self, session_id):
captured["exported"] = session_id
return {
"id": session_id,
"title": "Export CLI Test",
"source": "cli",
"message_count": 1,
"messages": [{"role": "user", "content": "hello"}],
}
def delete_session(self, *args, **kwargs):
raise AssertionError("markdown export must not delete sessions")
def prune_sessions(self, *args, **kwargs):
raise AssertionError("markdown export must not prune sessions")
def close(self):
captured["closed"] = True
monkeypatch.setattr(hermes_state, "SessionDB", lambda: FakeDB())
monkeypatch.setattr(
sys,
"argv",
[
"hermes",
"sessions",
"export",
"--format",
"md",
"--session-id",
"20260706_123456",
str(tmp_path),
],
)
main_mod.main()
output = capsys.readouterr().out
files = list(tmp_path.glob("*.md"))
assert len(files) == 1
text = files[0].read_text(encoding="utf-8")
assert "# Export CLI Test" in text
assert "hello" in text
assert captured == {
"resolved_from": "20260706_123456",
"exported": "20260706_123456_abcd1234",
"closed": True,
}
assert "Exported 1 session" in output
assert "1 message" in output
assert str(files[0]) in output
def test_sessions_export_md_reports_unknown_session(monkeypatch, tmp_path, capsys):
import hermes_cli.main as main_mod
import hermes_state
output_dir = tmp_path / "exports"
class FakeDB:
def resolve_session_id(self, session_id):
return None
def export_session(self, session_id):
raise AssertionError("export_session should not be called")
def close(self):
pass
monkeypatch.setattr(hermes_state, "SessionDB", lambda: FakeDB())
monkeypatch.setattr(
sys,
"argv",
[
"hermes",
"sessions",
"export",
"--format",
"md",
"--session-id",
"missing",
str(output_dir),
],
)
main_mod.main()
output = capsys.readouterr().out
assert "Session 'missing' not found." in output
assert not output_dir.exists()
def test_sessions_export_md_supports_qmd_format(monkeypatch, tmp_path, capsys):
import hermes_cli.main as main_mod
import hermes_state
class FakeDB:
def resolve_session_id(self, session_id):
return "s1"
def export_session(self, session_id):
return {"id": "s1", "title": "QMD", "messages": []}
def close(self):
pass
monkeypatch.setattr(hermes_state, "SessionDB", lambda: FakeDB())
monkeypatch.setattr(
sys,
"argv",
[
"hermes",
"sessions",
"export",
"--session-id",
"s1",
"--format",
"qmd",
str(tmp_path),
],
)
main_mod.main()
assert len(list(tmp_path.glob("*.qmd"))) == 1
assert "Exported 1 session" in capsys.readouterr().out
def test_sessions_export_md_rejects_stdout_target(monkeypatch, tmp_path, capsys):
import hermes_cli.main as main_mod
import hermes_state
class FakeDB:
def resolve_session_id(self, session_id):
raise AssertionError("md export to stdout must be refused before DB access")
def close(self):
pass
monkeypatch.setattr(hermes_state, "SessionDB", lambda: FakeDB())
monkeypatch.setattr(
sys,
"argv",
["hermes", "sessions", "export", "--format", "md", "--session-id", "s1", "-"],
)
main_mod.main()
assert "only supported with --format jsonl" in capsys.readouterr().out
def test_sessions_export_jsonl_requires_output_path(monkeypatch, capsys):
import hermes_cli.main as main_mod
import hermes_state
class FakeDB:
def export_all(self, **kwargs):
raise AssertionError("jsonl export without an output path must be refused")
def close(self):
pass
monkeypatch.setattr(hermes_state, "SessionDB", lambda: FakeDB())
monkeypatch.setattr(sys, "argv", ["hermes", "sessions", "export"])
main_mod.main()
assert "requires an output path" in capsys.readouterr().out
def test_sessions_export_md_bulk_dry_run_lists_candidates(monkeypatch, tmp_path, capsys):
import hermes_cli.main as main_mod
import hermes_state
class FakeDB:
def list_prune_candidates(self, **kwargs):
# Export flows through the shared prune-filter machinery:
# --older-than 30 becomes a started_before epoch bound, source
# passes through, and archived is tri-state None (export includes
# archived sessions).
assert kwargs.get("source") == "cron"
assert kwargs.get("started_before") is not None
assert kwargs.get("archived") is None
return [{"id": "s1", "source": "cron"}, {"id": "s2", "source": "cron"}]
def close(self):
pass
monkeypatch.setattr(hermes_state, "SessionDB", lambda: FakeDB())
monkeypatch.setattr(
sys,
"argv",
[
"hermes",
"sessions",
"export",
"--format",
"md",
"--older-than",
"30",
"--source",
"cron",
"--dry-run",
str(tmp_path),
],
)
main_mod.main()
output = capsys.readouterr().out
assert "Would export 2 session(s)" in output
assert "s1" in output
assert "s2" in output
assert not list(tmp_path.glob("*.md"))
def test_sessions_export_md_bulk_requires_filter(monkeypatch, tmp_path, capsys):
import hermes_cli.main as main_mod
import hermes_state
class FakeDB:
def list_prune_candidates(self, **kwargs):
raise AssertionError("bulk export without filters should refuse")
def close(self):
pass
monkeypatch.setattr(hermes_state, "SessionDB", lambda: FakeDB())
monkeypatch.setattr(
sys,
"argv",
["hermes", "sessions", "export", "--format", "md", str(tmp_path)],
)
main_mod.main()
assert "Refusing bulk export without a filter" in capsys.readouterr().out
def test_sessions_export_md_bulk_writes_manifest(monkeypatch, tmp_path, capsys):
import hermes_cli.main as main_mod
import hermes_state
class FakeDB:
def list_prune_candidates(self, **kwargs):
return [{"id": "s1"}, {"id": "s2"}]
def export_session_lineage(self, session_id):
return {"id": session_id, "title": session_id, "messages": [{"role": "user", "content": session_id}]}
def close(self):
pass
monkeypatch.setattr(hermes_state, "SessionDB", lambda: FakeDB())
monkeypatch.setattr(
sys,
"argv",
[
"hermes",
"sessions",
"export",
"--format",
"md",
"--older-than",
"90",
"--lineage",
"logical",
str(tmp_path),
],
)
main_mod.main()
assert len(list(tmp_path.glob("*.md"))) == 2
manifest = tmp_path / "manifest.jsonl"
assert manifest.exists()
lines = manifest.read_text(encoding="utf-8").splitlines()
assert len(lines) == 2
assert "Exported 2 session(s)" in capsys.readouterr().out
def test_sessions_export_md_delete_after_verified_requires_yes(monkeypatch, tmp_path, capsys):
import hermes_cli.main as main_mod
import hermes_state
class FakeDB:
def close(self):
pass
monkeypatch.setattr(hermes_state, "SessionDB", lambda: FakeDB())
monkeypatch.setattr(
sys,
"argv",
[
"hermes",
"sessions",
"export",
"--format",
"md",
"--session-id",
"s1",
"--delete-after-verified",
str(tmp_path),
],
)
main_mod.main()
assert "requires --yes" in capsys.readouterr().out
def test_sessions_export_md_delete_after_verified_deletes_after_file_check(monkeypatch, tmp_path, capsys):
import hermes_cli.main as main_mod
import hermes_state
captured = {}
class FakeDB:
def resolve_session_id(self, session_id):
return "s1"
def export_session(self, session_id):
return {"id": "s1", "title": "Delete", "message_count": 1, "messages": [{"role": "user", "content": "safe"}]}
def delete_session(self, session_id, **kwargs):
captured["deleted"] = session_id
return True
def close(self):
pass
monkeypatch.setattr(hermes_state, "SessionDB", lambda: FakeDB())
monkeypatch.setattr(
sys,
"argv",
[
"hermes",
"sessions",
"export",
"--format",
"md",
"--session-id",
"s1",
"--delete-after-verified",
"--yes",
str(tmp_path),
],
)
main_mod.main()
assert captured == {"deleted": "s1"}
assert len(list(tmp_path.glob("*.md"))) == 1
assert "Deleted exported session 's1'" in capsys.readouterr().out
def test_sessions_export_md_accepts_duration_age_grammar(monkeypatch, tmp_path, capsys):
"""--older-than accepts the same AGE grammar as prune ('2w', '5h', ISO)."""
import hermes_cli.main as main_mod
import hermes_state
class FakeDB:
def list_prune_candidates(self, **kwargs):
assert kwargs.get("started_before") is not None
return [{"id": "s1", "source": "cli"}]
def close(self):
pass
monkeypatch.setattr(hermes_state, "SessionDB", lambda: FakeDB())
monkeypatch.setattr(
sys,
"argv",
[
"hermes", "sessions", "export", "--format", "md",
"--older-than", "2w", "--dry-run", str(tmp_path),
],
)
main_mod.main()
assert "Would export 1 session(s)" in capsys.readouterr().out
def test_sessions_export_md_supports_extended_prune_filters(monkeypatch, tmp_path, capsys):
"""Filters like --model/--min-messages pass through the shared machinery."""
import hermes_cli.main as main_mod
import hermes_state
captured = {}
class FakeDB:
def list_prune_candidates(self, **kwargs):
captured.update(kwargs)
return []
def close(self):
pass
monkeypatch.setattr(hermes_state, "SessionDB", lambda: FakeDB())
monkeypatch.setattr(
sys,
"argv",
[
"hermes", "sessions", "export", "--format", "md",
"--model", "sonnet", "--min-messages", "5", "--dry-run",
str(tmp_path),
],
)
main_mod.main()
assert captured.get("model_like") == "sonnet"
assert captured.get("min_messages") == 5
assert "Would export 0 session(s)" in capsys.readouterr().out
def test_sessions_export_jsonl_honors_filters(monkeypatch, tmp_path, capsys):
"""JSONL bulk export uses the same filter machinery as md/qmd."""
import json
import hermes_cli.main as main_mod
import hermes_state
class FakeDB:
def list_prune_candidates(self, **kwargs):
assert kwargs.get("source") == "telegram"
return [{"id": "s1", "source": "telegram"}]
def export_session(self, session_id):
return {"id": session_id, "messages": [{"role": "user", "content": "hi"}]}
def export_all(self, **kwargs):
raise AssertionError("filtered jsonl export must not fall back to export_all")
def close(self):
pass
out = tmp_path / "out.jsonl"
monkeypatch.setattr(hermes_state, "SessionDB", lambda: FakeDB())
monkeypatch.setattr(
sys,
"argv",
["hermes", "sessions", "export", "--source", "telegram", str(out)],
)
main_mod.main()
lines = out.read_text(encoding="utf-8").splitlines()
assert len(lines) == 1
assert json.loads(lines[0])["id"] == "s1"
assert "Exported 1 sessions" in capsys.readouterr().out
def test_sessions_export_redact_scrubs_secrets(monkeypatch, tmp_path):
"""--redact runs exported content through force-mode secret redaction."""
import hermes_cli.main as main_mod
import hermes_state
secret = "sk-proj-Zz12345678901234567890123456789012345678"
class FakeDB:
def resolve_session_id(self, session_id):
return "s1"
def export_session(self, session_id):
return {
"id": "s1",
"title": "Redact",
"messages": [
{"role": "tool", "name": "terminal", "content": f"api key: {secret}"}
],
}
def close(self):
pass
monkeypatch.setattr(hermes_state, "SessionDB", lambda: FakeDB())
monkeypatch.setattr(
sys,
"argv",
[
"hermes", "sessions", "export", "--format", "md",
"--session-id", "s1", "--redact", str(tmp_path),
],
)
main_mod.main()
text = next(tmp_path.glob("*.md")).read_text(encoding="utf-8")
assert secret not in text
assert "api key:" in text
def _trace_fake_db(captured):
class FakeDB:
def resolve_session_id(self, session_id):
return "s1"
def get_session(self, session_id):
return {"id": session_id, "model": "test-model"}
def get_messages_as_conversation(self, session_id):
captured["conv"] = session_id
return [
{"role": "user", "content": "hello trace"},
{"role": "assistant", "content": "hi"},
]
def close(self):
captured["closed"] = True
return FakeDB()
def test_sessions_export_trace_writes_claude_jsonl(monkeypatch, tmp_path, capsys):
import json
import hermes_cli.main as main_mod
import hermes_state
captured = {}
out = tmp_path / "trace.jsonl"
monkeypatch.setattr(hermes_state, "SessionDB", lambda: _trace_fake_db(captured))
monkeypatch.setattr(
sys,
"argv",
["hermes", "sessions", "export", "--format", "trace", "--session-id", "s1", str(out)],
)
main_mod.main()
assert "Exported 1 session trace" in capsys.readouterr().out
lines = [json.loads(line) for line in out.read_text(encoding="utf-8").splitlines()]
assert {rec["type"] for rec in lines} == {"user", "assistant"}
assert all("uuid" in rec for rec in lines)
assert captured["conv"] == "s1"
assert captured["closed"] is True
def test_sessions_export_trace_stdout(monkeypatch, capsys):
import json
import hermes_cli.main as main_mod
import hermes_state
captured = {}
monkeypatch.setattr(hermes_state, "SessionDB", lambda: _trace_fake_db(captured))
monkeypatch.setattr(
sys,
"argv",
["hermes", "sessions", "export", "--format", "trace", "--session-id", "s1", "-"],
)
main_mod.main()
lines = [json.loads(line) for line in capsys.readouterr().out.splitlines()]
assert len(lines) == 2
assert lines[0]["type"] == "user"
def test_sessions_export_trace_upload_routes_to_uploader(monkeypatch, capsys):
import hermes_cli.main as main_mod
import hermes_state
from agent import trace_upload as trace_mod
captured = {}
monkeypatch.setattr(hermes_state, "SessionDB", lambda: _trace_fake_db(captured))
def fake_upload(session_id, **kwargs):
captured["uploaded"] = session_id
captured["kwargs"] = kwargs
return "Uploaded -> https://example.test/dataset"
monkeypatch.setattr(trace_mod, "upload_session_trace", fake_upload)
monkeypatch.setattr(
sys,
"argv",
[
"hermes", "sessions", "export", "--format", "trace",
"--session-id", "s1", "--upload", "--public",
],
)
main_mod.main()
assert captured["uploaded"] == "s1"
assert captured["kwargs"]["private"] is False
assert captured["kwargs"]["redact"] is True
assert "Uploaded ->" in capsys.readouterr().out
def test_sessions_export_trace_only_flag_rejected(monkeypatch, capsys):
import hermes_cli.main as main_mod
import hermes_state
captured = {}
monkeypatch.setattr(hermes_state, "SessionDB", lambda: _trace_fake_db(captured))
monkeypatch.setattr(
sys,
"argv",
[
"hermes", "sessions", "export", "--format", "trace",
"--session-id", "s1", "--only", "user-prompts", "-",
],
)
main_mod.main()
assert "--only user-prompts supports --format jsonl or md." in capsys.readouterr().out