mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-09-17 05:32:20 +02:00
39 lines
1.6 KiB
Python
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"
|
|
)
|