From 36b916779774b76e799fe895d28357403fd10486 Mon Sep 17 00:00:00 2001 From: Santhi Prakash Date: Thu, 6 Aug 2026 03:57:40 +0000 Subject: [PATCH] test(buzz-agent): wait for steer response in steer_rejected_on_empty_prompt Signed-off-by: Santhi Prakash --- crates/buzz-agent/tests/fake_llm.rs | 15 +++------------ 1 file changed, 3 insertions(+), 12 deletions(-) diff --git a/crates/buzz-agent/tests/fake_llm.rs b/crates/buzz-agent/tests/fake_llm.rs index 4253ef329c..7bbf8cb849 100644 --- a/crates/buzz-agent/tests/fake_llm.rs +++ b/crates/buzz-agent/tests/fake_llm.rs @@ -1428,7 +1428,7 @@ async fn steer_rejected_on_empty_prompt() { .await; let mut h = Harness::spawn(&url).await; let sid = init_session(&mut h).await; - let p_id = h + let _p_id = h .send( "session/prompt", json!({"sessionId": sid, "prompt": [{"type":"text","text":"go"}]}), @@ -1441,17 +1441,8 @@ async fn steer_rejected_on_empty_prompt() { json!({"sessionId": sid, "expectedRunId": run_id, "prompt": []}), ) .await; - let mut saw_reject = false; - for _ in 0..40 { - let v = h.recv().await; - if v["id"] == json!(s_id) { - assert_eq!(v["error"]["code"], -32602, "empty prompt must be rejected"); - saw_reject = true; - } else if v["id"] == json!(p_id) { - break; - } - } - assert!(saw_reject, "empty steer prompt was not rejected"); + let v = h.recv_until(|v| v["id"] == json!(s_id)).await; + assert_eq!(v["error"]["code"], -32602, "empty prompt must be rejected"); h.shutdown().await; }