1685ffcb15
The export_all assertion already lives in tests/hermes_cli/test_session_export_batch.py. Replace it with the lineage test (timings over merged messages, red on the previous head) which also proves the derived block is excluded from the import size budget.
115 lines
4.4 KiB
Python
115 lines
4.4 KiB
Python
"""Session export timing evidence."""
|
|
|
|
from hermes_state import SessionDB
|
|
|
|
|
|
def test_export_session_includes_text_free_timing_evidence(tmp_path):
|
|
db = SessionDB(db_path=tmp_path / "state.db")
|
|
try:
|
|
db.create_session(session_id="s1", source="cli", model="test-model")
|
|
db.append_message("s1", "user", "secret prompt", timestamp=1000.0)
|
|
db.append_message(
|
|
"s1",
|
|
"assistant",
|
|
"",
|
|
tool_calls=[{"id": "call-1", "function": {"name": "terminal"}}],
|
|
timestamp=1001.25,
|
|
)
|
|
db.append_message(
|
|
"s1",
|
|
"tool",
|
|
"secret tool output",
|
|
tool_name="terminal",
|
|
tool_call_id="call-1",
|
|
timestamp=1003.0,
|
|
)
|
|
db.append_message("s1", "assistant", "done", timestamp=1003.5)
|
|
|
|
exported = db.export_session("s1")
|
|
finally:
|
|
db.close()
|
|
|
|
timings = exported["timings"]
|
|
assert timings["source"] == "message_timestamps"
|
|
assert timings["available"] is True
|
|
assert timings["complete"] is False
|
|
assert timings["wall_clock_ms"] == 3500
|
|
assert timings["largest_gap_ms"] == 1750
|
|
assert timings["message_timestamps"] == {"available": 4, "missing": 0}
|
|
assert timings["role_counts"] == {"user": 1, "assistant": 2, "tool": 1}
|
|
assert timings["tool_calls_emitted"] == 1
|
|
assert timings["tool_result_count"] == 1
|
|
assert timings["intervals"] == [
|
|
{
|
|
"from_message_id": 1,
|
|
"to_message_id": 2,
|
|
"from_role": "user",
|
|
"to_role": "assistant",
|
|
"gap_ms": 1250,
|
|
},
|
|
{
|
|
"from_message_id": 2,
|
|
"to_message_id": 3,
|
|
"from_role": "assistant",
|
|
"to_role": "tool",
|
|
"gap_ms": 1750,
|
|
},
|
|
{
|
|
"from_message_id": 3,
|
|
"to_message_id": 4,
|
|
"from_role": "tool",
|
|
"to_role": "assistant",
|
|
"gap_ms": 500,
|
|
},
|
|
]
|
|
assert "secret prompt" not in str(timings)
|
|
assert "secret tool output" not in str(timings)
|
|
|
|
|
|
def test_lineage_export_timings_span_the_merged_messages_and_import_ignores_their_size(tmp_path):
|
|
"""A compression lineage exports as one logical session: its timings must cover every
|
|
segment's messages, not just the last segment's. On import the derived block is stripped
|
|
from the per-session size measurement so a long lineage's intervals cannot trip the limit."""
|
|
db = SessionDB(db_path=tmp_path / "state.db")
|
|
try:
|
|
db.create_session(session_id="root", source="cli", model="m")
|
|
db.append_message("root", "user", "first", timestamp=100.0)
|
|
db.append_message("root", "assistant", "ok", timestamp=101.0)
|
|
db.end_session("root", end_reason="compression")
|
|
db.create_session(session_id="child", source="cli", model="m", parent_session_id="root")
|
|
db.append_message("child", "user", "second", timestamp=200.0)
|
|
db.append_message("child", "assistant", "done", timestamp=200.5)
|
|
|
|
exported = db.export_session_lineage("child")
|
|
assert exported["lineage_session_ids"] == ["root", "child"]
|
|
assert exported["timings"]["wall_clock_ms"] == 100_500
|
|
assert exported["timings"]["message_timestamps"] == {"available": 4, "missing": 0}
|
|
assert exported["segments"][-1]["timings"]["wall_clock_ms"] == 500
|
|
|
|
exported["timings"]["intervals"] = [{"pad": "x" * 100} for _ in range(60_000)] # ~6 MiB
|
|
target = SessionDB(db_path=tmp_path / "target.db")
|
|
try:
|
|
report = target.import_sessions([exported])
|
|
finally:
|
|
target.close()
|
|
finally:
|
|
db.close()
|
|
assert report["errors"] == [] and report["imported"] == 1
|
|
|
|
|
|
def test_corrupt_timestamp_rows_count_as_missing_instead_of_aborting_export(tmp_path):
|
|
db = SessionDB(db_path=tmp_path / "state.db")
|
|
try:
|
|
db.create_session(session_id="s1", source="cli", model="test-model")
|
|
db.append_message("s1", "user", "hello", timestamp=10.0)
|
|
db.append_message("s1", "assistant", "hi", timestamp=11.0)
|
|
# Writers refuse bad stamps; emulate a pre-existing corrupt row directly.
|
|
db._conn.execute("UPDATE messages SET timestamp = 8.4e252 WHERE id = 2")
|
|
db._conn.commit()
|
|
timings = db.export_session("s1")["timings"]
|
|
finally:
|
|
db.close()
|
|
|
|
assert timings["message_timestamps"] == {"available": 1, "missing": 1}
|
|
assert timings["wall_clock_ms"] == 0 and timings["intervals"] == []
|