Files
hermes-webui/tests/test_issue660.py
T
nesquena-hermes 19080f73b1 Release v0.51.348 — Release LL (Phase 0 hotfix: timeout regression + data-loss + leaks) (#3917)
* stage v0.51.348: Phase 0 hotfix — approval/clarify timeout regression (#3913), queue/draft durability (#3906), settings auto-reopen (#3909), kanban FD leak (#3904)

* stage v0.51.348: re-anchor 4 SSE frontend tests to poll-only design (#3913); apply Opus SHOULD-FIX — immediate first poll tick so pending approval/clarify cards show instantly

---------

Co-authored-by: nesquena-hermes <nesquena-hermes@users.noreply.github.com>
2026-06-10 01:12:48 -07:00

116 lines
6.1 KiB
Python

"""
Tests for session queue persistence across page refresh and tab restore.
#660 introduced sessionStorage persistence. #3108 hardens it by mirroring queue
state to localStorage and restoring from the durable copy when sessionStorage is
missing after browser tab/process restore.
"""
import pathlib
UI_JS = pathlib.Path(__file__).parent.parent / 'static' / 'ui.js'
SESSIONS_JS = pathlib.Path(__file__).parent.parent / 'static' / 'sessions.js'
ui_src = UI_JS.read_text(encoding='utf-8')
sess_src = SESSIONS_JS.read_text(encoding='utf-8')
class TestQueuePersistence:
"""queueSessionMessage persists through the shared dual-storage helper."""
def test_queue_storage_helpers_exist(self):
"""Queue persistence must be centralized so write/delete paths stay symmetric."""
assert "function _queueStorageKey(sid)" in ui_src
assert "function _persistSessionQueueStorage(sid, queue)" in ui_src
assert "function _readPersistedSessionQueue(sid)" in ui_src
assert "function _clearPersistedSessionQueue(sid)" in ui_src
def test_queue_writes_to_session_and_local_storage(self):
"""queueSessionMessage must mirror queue state to sessionStorage and localStorage."""
helper_start = ui_src.find("function _persistSessionQueueStorage(sid, queue)")
helper_end = ui_src.find("function _readPersistedSessionQueue(sid)", helper_start)
assert helper_start != -1 and helper_end != -1, "_persistSessionQueueStorage helper not found"
helper = ui_src[helper_start:helper_end]
assert "sessionStorage.setItem(key,payload)" in helper
assert "localStorage.setItem(key,payload)" in helper
def test_queue_stamps_queued_at_timestamp(self):
"""Each queue entry must have a _queued_at timestamp for stale-entry detection."""
assert '_queued_at' in ui_src
def test_shift_uses_shared_persist_and_clear_helpers(self):
"""shiftQueuedSessionMessage must update/remove both storage layers through helpers."""
start = ui_src.find("function shiftQueuedSessionMessage(sid)")
end = ui_src.find("function getQueuedSessionCount(sid)", start)
assert start != -1 and end != -1, "shiftQueuedSessionMessage block not found"
body = ui_src[start:end]
assert "_clearPersistedSessionQueue(sid)" in body
assert "_persistSessionQueueStorage(sid,q)" in body
def test_queue_card_edit_paths_use_shared_helpers(self):
"""Queue edit/combine/delete paths must not leave localStorage stale."""
assert "_saveAndRefresh()" in ui_src
assert "_persistSessionQueueStorage(sid,liveQ)" in ui_src
assert "_clearPersistedSessionQueue(sid)" in ui_src
class TestQueueRestore:
"""Queue is restored from the shared storage helper on idle session load."""
def test_restore_reads_shared_helper(self):
"""sessions.js must use the shared helper so localStorage fallback is reachable."""
assert "_readPersistedSessionQueue(sid)" in sess_src
def test_read_helper_falls_back_to_local_storage(self):
"""The helper must fall back to localStorage and re-mirror sessionStorage."""
start = ui_src.find("function _readPersistedSessionQueue(sid)")
end = ui_src.find("function queueSessionMessage(sid", start)
assert start != -1 and end != -1, "_readPersistedSessionQueue block not found"
body = ui_src[start:end]
assert "const sessionValue=read(sessionStorage)" in body
assert "if(sessionValue&&sessionValue.length) return sessionValue;" in body
assert "const localValue=read(localStorage)" in body
assert "if(localValue&&localValue.length)" in body
assert "sessionStorage.setItem(key,JSON.stringify(localValue))" in body
def test_restore_uses_timestamp_guard(self):
"""Stale entries (created before last assistant response) must be dropped."""
assert '_queued_at' in sess_src
assert '_lastAsst' in sess_src
def test_restore_shows_toast(self):
"""User must see a toast notification when a queue is restored."""
assert 'queued message' in sess_src.lower() and 'restored' in sess_src.lower()
def test_restore_puts_text_in_composer(self):
"""First queued message goes into the composer input, not auto-sent."""
assert "_msg.value=_first.text" in sess_src
def test_restore_clears_stale_storage(self):
"""On timestamp mismatch, stale queue state is removed from both storage layers."""
assert "_clearPersistedSessionQueue(sid)" in sess_src
def test_restore_wrapped_in_try_catch(self):
"""Storage access must be wrapped in try/catch (private browsing may block it)."""
assert "catch(_){if(typeof _clearPersistedSessionQueue==='function') _clearPersistedSessionQueue(sid);}" in sess_src
def test_delete_session_clears_persisted_queue_after_success(self):
"""Deleting a session must clear localStorage-backed queue state after the API succeeds."""
start = sess_src.find("async function deleteSession(sid, beforeDelete=null)")
end = sess_src.find("// ── Project helpers", start)
assert start != -1 and end != -1, "deleteSession block not found"
body = sess_src[start:end]
clear_pos = body.find("if(typeof _clearPersistedSessionQueue==='function') _clearPersistedSessionQueue(sid);")
error_pos = body.find("if(deleteResult&&deleteResult.error){")
success_pos = body.find("const response=deleteResult&&deleteResult.response;")
assert error_pos != -1 and success_pos != -1 and clear_pos != -1
assert success_pos < clear_pos, "queue cleanup should run only after delete success"
def test_active_session_not_restored_as_draft(self):
"""When agent is active (INFLIGHT), queue restore must NOT run."""
# The restore block must be inside the else branch (idle path), not the INFLIGHT branch
inflight_pos = sess_src.find("if(INFLIGHT[sid]){")
restore_pos = sess_src.find("_readPersistedSessionQueue(sid)")
else_pos = sess_src.find("}else{", inflight_pos)
assert restore_pos > else_pos, \
"Queue restore must be inside the else (idle) branch, not the INFLIGHT branch"