Files
odysseus/tests/test_history_db_fallback_hidden.py
T

39 lines
1.6 KiB
Python

"""Regression: the DB fallback in get_session_history must hide the same
messages the in-memory path hides.
The in-memory branch skips messages whose metadata has ``hidden`` (e.g.
compaction summaries that are kept for AI context but not shown to the user).
The DB fallback (taken when the in-memory history is empty, e.g. after a
restart) built the client response from every DB row with no such filter, so
hidden messages leaked to the client on DB-served sessions. Hydration of
``session.history`` belongs to ``get_session``; this fallback only shapes the
response, so only the response is filtered.
get_session_history depends on the DB, the session manager and a FastAPI
request, so this pins the regression at the source level (as other route tests
in this repo do).
"""
import ast
from pathlib import Path
SRC = Path(__file__).resolve().parent.parent / "routes" / "history" / "history_routes.py"
def _function_source(src_text, name):
tree = ast.parse(src_text)
for node in ast.walk(tree):
if isinstance(node, (ast.FunctionDef, ast.AsyncFunctionDef)) and node.name == name:
return ast.get_source_segment(src_text, node)
raise AssertionError(f"{name} not found in {SRC}")
def test_db_fallback_filters_hidden_from_response():
src = _function_source(SRC.read_text(), "get_session_history")
marker = "load from DB"
assert marker in src, "expected the DB fallback block in get_session_history"
db_section = src.split(marker, 1)[1]
assert "hidden" in db_section, (
"the DB-fallback path must filter `hidden` messages from the response "
"to match the in-memory path"
)