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&&renderTailStart0&&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(aIdxa-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_cycle_aware_per_virtual_window_burst(): assert "function _messageVirtualMeasurementCycleKeyFor(windowMetrics)" in UI_JS assert "function _scheduleMessageVirtualMeasurementRefresh(windowMetrics)" in UI_JS assert "_scheduleMessageVirtualMeasurementRefresh(virtualWindow);" in UI_JS assert "_markMessageVirtualMeasurementsSettled(virtualWindow);" in UI_JS # Cycle-aware burst tracking: unseen keys proceed, repeated keys terminate, # and the seen-key memory lives in _messageVirtualMeasurementSeenKeys. assert "_messageVirtualMeasurementSeenKeys.includes(cycleKey)" in UI_JS assert "_messageVirtualMeasurementSeenKeys.push(cycleKey)" in UI_JS assert "function _resetMessageVirtualMeasurementBurst()" in UI_JS def test_measurement_burst_not_reset_by_cycle_key_change(): """#6654/#6717: the cycle-key branch of _scheduleMessageVirtualMeasurementRefresh must record the new key but never reset the burst state. Resetting on key change lets WebKit's A->B->A->B metric oscillation renew the chain forever, keeping the rAF/measure loop alive. The ONLY resets allowed inside the scheduler are: the repeated-key termination (oscillation) and the fresh-burst start (when no burst is active), guarded by if(!_messageVirtualMeasurementBurstActive).""" 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 key_branch_start = body.index("if(_messageVirtualMeasurementCycleKey!==cycleKey){") key_branch_end = body.index("}", key_branch_start) key_branch = body[key_branch_start:key_branch_end] assert "_messageVirtualMeasurementSeenKeys" not in key_branch, ( "cycle-key change must not reset the burst's seen-key memory (issue #6654)" ) assert "_messageVirtualMeasurementBurstActive=false" not in key_branch, ( "cycle-key change must not end the burst (issue #6654)" ) # The fresh-burst reset is allowed only at the start of a new external cycle. assert "if(!_messageVirtualMeasurementBurstActive){" in body guard_pos = body.index("if(!_messageVirtualMeasurementBurstActive){") seen_reset_pos = body.index("_messageVirtualMeasurementSeenKeys=[];") assert guard_pos < seen_reset_pos, ( "the seen-key reset must be guarded by the burst-active check " "(per-burst lifecycle, issue #6717)" ) # A NON-consecutive repeat (A -> B -> A) is oscillation and terminates the # burst. A consecutive repeat (same window needing a second re-measure, e.g. # a row shrinking 1500 -> 1400 -> 100 px) proceeds; unseen keys proceed. assert "if(cycleKey !== lastKey && _messageVirtualMeasurementSeenKeys.includes(cycleKey)){" in body assert "_messageVirtualMeasurementSeenKeys.push(cycleKey);" in body # The internal-measurement origin travels WITH the scheduled render request # (threaded through both rAF layers), never as a global consumable flag that # a coalesced external render could steal (#6717 re-gate MUST-FIX 2). assert "origin:'internal'" in body assert "_messageVirtualMeasurementRenderPending" not in UI_JS # Absolute per-burst ceiling: even an all-distinct key sequence terminates, # bounding the seen-key memory (#6717 re-gate MUST-FIX 1). assert "MESSAGE_VIRTUAL_MEASUREMENT_MAX_RERENDERS" in UI_JS