Class _FakeStream

tests/lib/tools/test_session_runner.py:131–170

Stand-in for the AsyncStream returned by ``events.stream()``. Yields scripted events in order; once exhausted, blocks forever (the real stream stays open until a network event closes it). If ``raise_after`` is set, raises ``raise_with`` after producing that many events — used to exe

class _FakeStream:
    """Stand-in for the AsyncStream returned by ``events.stream()``.

    Yields scripted events in order; once exhausted, blocks forever (the real
    stream stays open until a network event closes it). If ``raise_after`` is
    set, raises ``raise_with`` after producing that many events — used to
    exercise the reconnect-with-backoff path.
    """

    def __init__(
        self,
        events: list[_StubEvent],
        *,
        raise_after: int | None = None,
        raise_with: BaseException | None = None,
    ) -> None:
        self._events = events
        self._raise_after = raise_after
        self._raise_with = raise_with

    async def __aenter__(self) -> _FakeStream:
        return self

    async def __aexit__(self, *exc: object) -> None:
        return None

    def __aiter__(self) -> Any:
        return self._gen()

    async def _gen(self) -> Any:
        for i, ev in enumerate(self._events):
            yield ev
            # Yield control so the dispatch task can pick up the event we just
            # produced before we run on to the next one.
            await asyncio.sleep(0)
            if self._raise_after is not None and i + 1 == self._raise_after:
                assert self._raise_with is not None
                raise self._raise_with
        # Keep the connection "open" until cancelled.
        await asyncio.Event().wait()


class FakeAsyncEvents: