Function _reply_for

tests/server/test_agent_server_e2e.py:606–612

(rid, req)

            assert init["subtype"] == "init"

            async def _reply_for(rid, req):
                await handle.send_to_agent({"type": "control_request", "request_id": rid, "request": req})
                for _ in range(12):
                    msg = await asyncio.wait_for(gen.__anext__(), timeout=5)
                    if msg.get("type") == "control_response" and msg["response"].get("request_id") == rid:
                        return msg["response"]["response"]
                raise AssertionError(f"no reply for {rid}")

            ok = await _reply_for("s1", {"subtype": "set_output_style", "style": "concise"})
            assert ok["ok"] is True and ok["style"] == "concise"