kshitijk4poor
b94581315e
The salvaged #88217 test replaced Path.stat on the class with a lambda taking one positional arg; pathlib.exists() passes follow_symlinks= and pytest's own tmp_path teardown crashed with INTERNALERROR TypeError. Delegate to the real stat for every path other than state.db.
53 lines
1.8 KiB
Python
53 lines
1.8 KiB
Python
"""Regression coverage for FTS storage upgrade discoverability."""
|
|
|
|
import sqlite3
|
|
from types import SimpleNamespace
|
|
|
|
|
|
def test_update_notice_offers_v1_trigram_tool_calls_rebuild(tmp_path, monkeypatch, capsys):
|
|
"""A deployed v1 trigram projection still receives the opt-in notice."""
|
|
from hermes_cli import update_cmd
|
|
import hermes_constants
|
|
import hermes_state
|
|
|
|
db_path = tmp_path / "state.db"
|
|
db_path.touch()
|
|
conn = sqlite3.connect(db_path)
|
|
conn.executescript(
|
|
"""
|
|
CREATE TABLE state_meta (key TEXT PRIMARY KEY, value TEXT);
|
|
CREATE TABLE messages_fts (content TEXT, tool_name TEXT, tool_calls TEXT);
|
|
CREATE TABLE messages_fts_trigram (content TEXT, tool_name TEXT, tool_calls TEXT);
|
|
"""
|
|
)
|
|
|
|
class FakeSessionDB:
|
|
def __init__(self, **_kwargs):
|
|
self._conn = conn
|
|
|
|
def close(self):
|
|
pass
|
|
|
|
_db_needs_fts_storage_upgrade = staticmethod(
|
|
hermes_state.SessionDB._db_needs_fts_storage_upgrade
|
|
)
|
|
|
|
monkeypatch.setattr(hermes_constants, "get_hermes_home", lambda: tmp_path)
|
|
monkeypatch.setattr(hermes_state, "SessionDB", FakeSessionDB)
|
|
# Report a large state.db without patching Path.stat globally: a
|
|
# 1-arg lambda on the class breaks pathlib.exists(follow_symlinks=...)
|
|
# for every caller in the process (pytest's own teardown included).
|
|
real_stat = update_cmd.Path.stat
|
|
|
|
def _stat(path, *args, **kwargs):
|
|
if path.name == "state.db":
|
|
return SimpleNamespace(st_size=512 * 1024 ** 2)
|
|
return real_stat(path, *args, **kwargs)
|
|
|
|
monkeypatch.setattr(update_cmd.Path, "stat", _stat)
|
|
|
|
update_cmd._print_fts_optimize_available_notice()
|
|
|
|
assert "hermes sessions optimize-storage" in capsys.readouterr().out
|
|
conn.close()
|