- Keep (a) list/stats/pinned open SessionDB(read_only=True) — one parametrized test — and (b) a missing store prints empty results and is never created. - Drop the insights read-only test (already covered on main), the status test, the mutating-action/live-writer/doctor-isolation tests, and the two doctor factory tests. - Replace _EmptyObservationalStore + error-string sniffing with a plain '_default_db_path() does not exist' branch printing each action's empty output; the fake-store tests in test_sessions_pin keep working because the branch only runs when the open fails.
97 lines
3.3 KiB
Python
97 lines
3.3 KiB
Python
from unittest.mock import MagicMock, patch
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
from cli import HermesCLI
|
|
from hermes_cli.main_agent_cmds import cmd_insights
|
|
from hermes_state import _default_db_path
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _state_db_exists():
|
|
# insights short-circuits before opening when state.db is absent (fresh install).
|
|
_default_db_path().parent.mkdir(parents=True, exist_ok=True)
|
|
_default_db_path().touch()
|
|
|
|
|
|
class _InsightsEngineStub:
|
|
calls = []
|
|
|
|
def __init__(self, db):
|
|
self.db = db
|
|
|
|
def generate(self, *, days=30, source=None):
|
|
self.calls.append({"days": days, "source": source})
|
|
return {"days": days, "source": source}
|
|
|
|
def format_terminal(self, report):
|
|
return f"days={report['days']} source={report['source']}"
|
|
|
|
|
|
def _run_show_insights(command: str):
|
|
cli_obj = HermesCLI.__new__(HermesCLI)
|
|
db = MagicMock()
|
|
_InsightsEngineStub.calls = []
|
|
with patch("hermes_state.SessionDB", return_value=db), \
|
|
patch("agent.insights.InsightsEngine", _InsightsEngineStub):
|
|
cli_obj._show_insights(command)
|
|
return _InsightsEngineStub.calls, db
|
|
|
|
|
|
def test_cli_insights_accepts_positional_days(capsys):
|
|
calls, db = _run_show_insights("/insights 7")
|
|
|
|
assert calls == [{"days": 7, "source": None}]
|
|
db.close.assert_called_once()
|
|
assert "days=7 source=None" in capsys.readouterr().out
|
|
|
|
|
|
def test_cli_insights_keeps_days_flag_and_source(capsys):
|
|
calls, db = _run_show_insights("/insights --days 14 --source discord")
|
|
|
|
assert calls == [{"days": 14, "source": "discord"}]
|
|
db.close.assert_called_once()
|
|
assert "days=14 source=discord" in capsys.readouterr().out
|
|
|
|
|
|
def test_insights_skips_open_when_store_missing(capsys):
|
|
# Fresh install: no state.db yet → no open at all (a read-only open needs an existing file).
|
|
_default_db_path().unlink()
|
|
for run in (lambda: HermesCLI.__new__(HermesCLI)._show_insights("/insights 7"),
|
|
lambda: cmd_insights(SimpleNamespace(days=30, source=None))):
|
|
with patch("hermes_state.SessionDB") as ctor:
|
|
run()
|
|
ctor.assert_not_called()
|
|
assert "No session data yet." in capsys.readouterr().out
|
|
|
|
|
|
def test_show_insights_opens_read_only():
|
|
cli_obj = HermesCLI.__new__(HermesCLI)
|
|
db = MagicMock()
|
|
with patch("hermes_state.SessionDB", return_value=db) as ctor, \
|
|
patch("agent.insights.InsightsEngine", _InsightsEngineStub):
|
|
_InsightsEngineStub.calls = []
|
|
cli_obj._show_insights("/insights 7")
|
|
ctor.assert_called_once_with(read_only=True)
|
|
db.close.assert_called_once()
|
|
|
|
|
|
def test_subcommand_insights_opens_read_only():
|
|
db = MagicMock()
|
|
with patch("hermes_state.SessionDB", return_value=db) as ctor, \
|
|
patch("agent.insights.InsightsEngine", _InsightsEngineStub):
|
|
cmd_insights(SimpleNamespace(days=30, source=None))
|
|
ctor.assert_called_once_with(read_only=True)
|
|
db.close.assert_called_once()
|
|
|
|
|
|
def test_subcommand_insights_closes_database_when_generation_fails(capsys):
|
|
db = MagicMock()
|
|
with patch("hermes_state.SessionDB", return_value=db), \
|
|
patch("agent.insights.InsightsEngine", side_effect=RuntimeError("boom")):
|
|
cmd_insights(SimpleNamespace(days=30, source=None))
|
|
|
|
db.close.assert_called_once()
|
|
assert "Error generating insights: boom" in capsys.readouterr().out
|