ed/manual_slop
Private
Public Access
0
0
Fork 0
Code Activity

test(infra): poll-for-event race fixes + watchdog timeout bump + spec update

Browse Source
This commit is contained in:
ed
2026-06-10 15:14:35 -04:00
parent 563e609505
commit 2c924fe6df
4 changed files with 55 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_reset_session_clears_mma_and_rag.py
+45 -8
View File
@@ -26,10 +26,27 @@ def test_reset_session_clears_mma_tier_usage(live_gui) -> None:
'tier_usage': {'Tier 1': {'model': 'polluted'}},
'tickets': []
})
time.sleep(0.5)
# Poll until the polluted entry is visible. Without this, the reset
# can fire BEFORE the push_event task is processed (async via io_pool
# + GUI render loop), and the test would falsely pass even if the
# reset didn't actually clear anything.
for _ in range(40):
state = client.get_gui_state()
mma = state.get('mma_state', {})
tier1 = mma.get('tier_usage', {}).get('Tier 1', {})
if tier1.get('model') == 'polluted':
break
time.sleep(0.25)
# Trigger the reset
client.reset_session()
time.sleep(0.5)
# Poll until the polluted entry is gone
for _ in range(40):
state = client.get_gui_state()
mma = state.get('mma_state', {})
tier1 = mma.get('tier_usage', {}).get('Tier 1', {})
if tier1.get('model') != 'polluted':
break
time.sleep(0.25)
# Verify the polluted entry is gone
state = client.get_gui_state()
mma = state.get('mma_state', {})
@@ -50,10 +67,21 @@ def test_reset_session_clears_mma_status(live_gui) -> None:
'tier_usage': {},
'tickets': []
})
time.sleep(0.5)
# Poll for the polluted status to be visible BEFORE the reset
for _ in range(40):
state = client.get_gui_state()
if state.get('mma_status') == 'running':
break
time.sleep(0.25)
client.reset_session()
time.sleep(0.5)
state = client.get_gui_state()
# Poll for the reset to have taken effect. Without this, the
# mma_state_update task can fire AFTER the reset, setting status
# back to 'running' (race condition surfaced in batched live_gui).
for _ in range(40):
state = client.get_gui_state()
if state.get('mma_status') == 'idle':
break
time.sleep(0.25)
assert state.get('mma_status') == 'idle', (
f"mma_status not reset: {state.get('mma_status')!r}"
)
@@ -70,10 +98,19 @@ def test_reset_session_clears_active_tier(live_gui) -> None:
'tier_usage': {},
'tickets': []
})
time.sleep(0.5)
# Poll for the polluted active_tier to be visible BEFORE the reset
for _ in range(40):
state = client.get_gui_state()
if state.get('active_tier') == 'Tier 2 (Tech Lead)':
break
time.sleep(0.25)
client.reset_session()
time.sleep(0.5)
state = client.get_gui_state()
# Poll for the reset to have taken effect
for _ in range(40):
state = client.get_gui_state()
if state.get('active_tier') is None:
break
time.sleep(0.25)
assert state.get('active_tier') is None, (
f"active_tier not reset: {state.get('active_tier')!r}"
)