Files
hermes-webui/tests/test_issue734_message_windowing.py
T
webtecnica 4df32ea332 fix(perf): don't reset virtual measurement retry budget on window-key oscillation (#6654)
_scheduleMessageVirtualMeasurementRefresh() reset _messageVirtualMeasurementRetryCount
every time _messageVirtualMeasurementCycleKey changed. When WebKit alternates
between two measured window-metric sets (A -> B -> A -> B) after
getBoundingClientRect(), each key transition renewed the two-render budget and
the requestAnimationFrame/geometry-measurement loop never terminated
(WebContent ~30% CPU for minutes after session recovery).

The MESSAGE_VIRTUAL_MEASUREMENT_MAX_RERENDERS budget now caps the whole
convergence burst globally; the cycle-key branch only records the new key.
The budget still resets at the correct lifecycle boundaries:
_markMessageVirtualMeasurementsSettled() and _clearMessageVirtualHeightCache().

Adds a deterministic node-based regression test driving the real ui.js
function with alternating keys ['A','B','A','B'] (before: 4 scheduled
re-renders; after: 2), plus a stable-key guard and a structural test.
2026-08-30 21:29:24 -03:00

80 lines
4.6 KiB
Python

from pathlib import Path
REPO = Path(__file__).resolve().parents[1]
UI_JS = (REPO / "static" / "ui.js").read_text(encoding="utf-8")
SESSIONS_JS = (REPO / "static" / "sessions.js").read_text(encoding="utf-8")
CSS = (REPO / "static" / "style.css").read_text(encoding="utf-8")
def test_message_virtualization_switches_render_messages_to_scroll_driven_window():
assert "function _messageVirtualWindow(opts)" in UI_JS
assert "function _messageVirtualKeepTailCount()" in UI_JS
assert "const virtualWindow=_currentMessageVirtualWindow(visWithIdx,_messageVirtualKeepTailCount())" in UI_JS
assert "const renderHeadVisWithIdx=visWithIdx.slice(windowStart, windowEnd)" in UI_JS
assert "const renderTailStart=virtualWindow.virtualized?Math.max(windowEnd, virtualWindow.tailStart):windowEnd" in UI_JS
assert "const renderTailVisWithIdx=virtualWindow.virtualized&&renderTailStart<visWithIdx.length" in UI_JS
assert "const renderVisWithIdx=renderHeadVisWithIdx.concat(renderTailVisWithIdx)" in UI_JS
assert "if(virtualWindow.virtualized&&virtualWindow.bottomPad>0&&vi===headRenderCount)" in UI_JS
def test_load_earlier_only_pages_server_history_and_preserves_scroll():
assert "function _wireMessageWindowLoadEarlierButton()" in UI_JS
assert "if(typeof _loadOlderMessages==='function') _loadOlderMessages();" in UI_JS
assert "if(hasServerOlder){" in UI_JS
assert "if(virtualWindow.virtualized&&virtualWindow.topPad>0)" in UI_JS
assert "_messageRenderWindowSize=_currentMessageRenderWindowSize()+Math.max(addedRenderable, MESSAGE_RENDER_WINDOW_DEFAULT);" in SESSIONS_JS
assert "renderMessages({ preserveScroll: true });" in SESSIONS_JS
assert "_scheduleMessageVirtualizedRender();" in UI_JS
def test_windowed_render_keeps_streaming_and_tool_activity_anchored_to_rendered_messages():
assert "_scrollAfterMessageRender(preserveScroll, scrollSnapshot);" in UI_JS
assert "const assistantIdxs=[...assistantSegments.keys()].sort((a,b)=>a-b);" in UI_JS
assert "if(aIdx<assistantIdxs[0]) continue;" in UI_JS
assert "const renderedAssistantIdxs=[...assistantSegments.keys()].sort((a,b)=>a-b);" in UI_JS
assert "const seg=assistantSegments.get(mi);" in UI_JS
def test_window_state_participates_in_cache_and_cached_button_is_rewired():
assert "cached.renderWindowKey===renderWindowKey" in UI_JS
assert "cached.signature===renderSignature" in UI_JS
assert "_sessionHtmlCache.set(sid,{html:_html,msgCount,renderWindowKey,signature:renderSignature})" in UI_JS
assert "_messageVirtualWindowKey=renderWindowKey" in UI_JS
assert "function _wireMessageWindowLoadEarlierButton()" in UI_JS
assert "_wireMessageWindowLoadEarlierButton();" in UI_JS
assert UI_JS.count("_wireMessageWindowLoadEarlierButton();") >= 2
def test_virtualization_affordances_have_styling_hooks():
assert "message-window-load-earlier" in UI_JS
assert ".message-window-load-earlier" in CSS
assert ".message-virtual-spacer" in CSS
assert "border-radius:999px" in CSS
def test_measurement_rerenders_are_bounded_per_virtual_window_cycle():
assert "const MESSAGE_VIRTUAL_MEASUREMENT_MAX_RERENDERS=2;" in UI_JS
assert "function _messageVirtualMeasurementCycleKeyFor(windowMetrics)" in UI_JS
assert "function _scheduleMessageVirtualMeasurementRefresh(windowMetrics)" in UI_JS
assert "if(_messageVirtualMeasurementRetryCount>=MESSAGE_VIRTUAL_MEASUREMENT_MAX_RERENDERS) return;" in UI_JS
assert "_scheduleMessageVirtualMeasurementRefresh(virtualWindow);" in UI_JS
assert "_markMessageVirtualMeasurementsSettled(virtualWindow);" in UI_JS
def test_measurement_retry_budget_not_reset_by_cycle_key_change():
"""#6654: the cycle-key branch of _scheduleMessageVirtualMeasurementRefresh
must record the new key but never reset _messageVirtualMeasurementRetryCount.
Resetting on key change lets WebKit's A->B->A->B metric oscillation renew
the two-render budget forever, keeping the rAF/measure loop alive."""
idx = UI_JS.index("function _scheduleMessageVirtualMeasurementRefresh(windowMetrics)")
end = UI_JS.index("function _markMessageVirtualMeasurementsSettled", idx)
body = UI_JS[idx:end]
# The key-change branch must only assign the key.
assert "if(_messageVirtualMeasurementCycleKey!==cycleKey){" in body
assert "_messageVirtualMeasurementRetryCount=0;" not in body, (
"cycle-key change must not reset the retry budget (issue #6654)"
)
# The budget guard and increment must still be present.
assert "if(_messageVirtualMeasurementRetryCount>=MESSAGE_VIRTUAL_MEASUREMENT_MAX_RERENDERS) return;" in body
assert "_messageVirtualMeasurementRetryCount++;" in body