diff --git a/tests/test_b19_headless_reconcile_fixes.py b/tests/test_b19_headless_reconcile_fixes.py index 4f925cd..36c56c1 100644 --- a/tests/test_b19_headless_reconcile_fixes.py +++ b/tests/test_b19_headless_reconcile_fixes.py @@ -88,7 +88,12 @@ def test_bug4_send_keys_safe_gating_order(): sks_idx = content.find("send_keys_safe() {") assert sks_idx != -1 - sks_body = content[sks_idx:sks_idx + 2500] + # The whole function body (up to the closing brace at column 0), not a fixed + # number of characters: lines added earlier in the function must not push + # a later call out of view. + sks_end = content.find("\n}\n", sks_idx) + assert sks_end != -1 + sks_body = content[sks_idx:sks_end] quiescent_idx = sks_body.find("_pane_quiescent") dialog_idx = sks_body.find("_pane_dialog_open") prompt_idx = sks_body.find("agent prompt")