mirror of
https://github.com/nesquena/hermes-webui.git
synced 2026-07-12 18:50:35 +00:00
Rod Boev
90 lines
2.7 KiB
Python
90 lines
2.7 KiB
Python
import api.routes as routes
|
|
from api.request_diagnostics import RequestDiagnostics
|
|
|
|
|
|
def test_request_diagnostics_maybe_start_still_covers_sessions_route():
|
|
assert RequestDiagnostics.maybe_start("GET", "/api/sessions") is not None
|
|
|
|
|
|
def _session_cache_diag_stage_names(diag: RequestDiagnostics) -> list[str]:
|
|
return [stage["name"] for stage in diag._stages] + [diag._current_stage]
|
|
|
|
|
|
def _session_cache_key() -> tuple:
|
|
return routes._session_list_cache_key(
|
|
active_profile="default",
|
|
all_profiles=False,
|
|
show_cli_sessions=False,
|
|
show_previous_messaging_sessions=False,
|
|
show_cron_sessions=False,
|
|
)
|
|
|
|
|
|
def test_sessions_route_emits_cache_hit_diagnostic_stages():
|
|
key = _session_cache_key()
|
|
routes._session_list_cache_clear()
|
|
routes._session_list_cache_set(key, {"sessions": [], "cli_count": 0})
|
|
|
|
diag = RequestDiagnostics("GET", "/api/sessions", auto_start=False)
|
|
payload = routes._get_cached_session_list_payload(
|
|
key=key,
|
|
builder=lambda: {"sessions": ["unexpected"]},
|
|
diag=diag,
|
|
)
|
|
|
|
assert payload == {"sessions": [], "cli_count": 0}
|
|
assert _session_cache_diag_stage_names(diag) == [
|
|
"start",
|
|
"session_list_cache_lookup",
|
|
"session_list_cache_hit",
|
|
]
|
|
|
|
|
|
def test_sessions_route_emits_cache_store_diagnostic_stages():
|
|
key = _session_cache_key()
|
|
routes._session_list_cache_clear()
|
|
|
|
diag = RequestDiagnostics("GET", "/api/sessions", auto_start=False)
|
|
payload = routes._get_cached_session_list_payload(
|
|
key=key,
|
|
builder=lambda: {"sessions": ["rebuilt"], "cli_count": 0},
|
|
diag=diag,
|
|
)
|
|
|
|
assert payload == {"sessions": ["rebuilt"], "cli_count": 0}
|
|
assert _session_cache_diag_stage_names(diag) == [
|
|
"start",
|
|
"session_list_cache_lookup",
|
|
"session_list_cache_rebuild_owner",
|
|
"session_list_cache_stored",
|
|
]
|
|
|
|
|
|
def test_sessions_route_emits_invalidation_retry_diagnostic_stage():
|
|
key = _session_cache_key()
|
|
routes._session_list_cache_clear()
|
|
calls = {"count": 0}
|
|
|
|
def _builder():
|
|
calls["count"] += 1
|
|
if calls["count"] == 1:
|
|
routes._session_list_cache_clear()
|
|
return {"sessions": [calls["count"]], "cli_count": 0}
|
|
|
|
diag = RequestDiagnostics("GET", "/api/sessions", auto_start=False)
|
|
payload = routes._get_cached_session_list_payload(
|
|
key=key,
|
|
builder=_builder,
|
|
diag=diag,
|
|
)
|
|
|
|
assert payload == {"sessions": [2], "cli_count": 0}
|
|
assert calls["count"] == 2
|
|
assert _session_cache_diag_stage_names(diag) == [
|
|
"start",
|
|
"session_list_cache_lookup",
|
|
"session_list_cache_rebuild_owner",
|
|
"session_list_cache_invalidated_during_rebuild",
|
|
"session_list_cache_stored",
|
|
]
|