Function core

python/tests/test_transport.py:104–117

()

@pytest.mark.asyncio
async def test_stream_collects_events_and_rejects_failed_done() -> None:
    transport, process = _transport()
    stream = transport.call_stream(
        pb.Envelope(
            ask_machine_req=pb.AskMachineRequest(machine_id="m1", prompt="hello")
        )
    )
    collected = asyncio.create_task(stream.collect())
    request = await _request(process)
    _feed(
        process,
        pb.Envelope(
            id=request.id,
            kind=pb.Envelope.KIND_EVENT,
            ask_event=pb.AskEvent(event_type=pb.AGENT_EVENT_TYPE_TOKEN, payload_json='{"delta":"hi"}'),
        ),
    )
    _feed(