async def _stream() -> WSStream:
stream = WSStream(AsyncMock(), Config(), AsyncMock(), False, None, None,
AsyncMock(), 1)
stream.context.spawn_app.return_value = AsyncMock() # type: ignore
stream.app_put = AsyncMock()
stream.config._log = AsyncMock(spec=Logger)
return stream
async def _stream() -> WSStream:
stream = WSStream(Config(), False, None, None, CoroutineMock(),
CoroutineMock(), 1)
stream.spawn_app.return_value = CoroutineMock()
stream.app_put = CoroutineMock()
stream.config._log = AsyncMock(spec=Logger)
return stream
async def test_handle_connection(stream: WSStream) -> None:
await stream.handle(
Request(
stream_id=1,
http_version="2.0",
headers=[(b"sec-websocket-version", b"13")],
raw_path=b"/?a=b",
method="GET",
)
)
await stream.app_send({"type": "websocket.accept"})
stream.app_put = CoroutineMock()
await stream.handle(Data(stream_id=1, data=b"\x81\x85&`\x13\x0eN\x05\x7fbI"))
stream.app_put.assert_called()
assert stream.app_put.call_args_list == [
call({"type": "websocket.receive", "bytes": None, "text": "hello"})
]
async def test_pings(stream: WSStream, event_loop: asyncio.AbstractEventLoop) -> None:
stream.config.websocket_ping_interval = 0.1
await stream.handle(
Request(
stream_id=1,
http_version="2",
headers=[(b"sec-websocket-version", b"13")],
raw_path=b"/?a=b",
method="GET",
)
)
async with TaskGroup(event_loop) as task_group:
stream.context = Context(task_group)
await stream.app_send({"type": "websocket.accept"})
stream.app_put = AsyncMock()
await asyncio.sleep(0.15)
assert stream.send.call_args_list == [
call(Response(stream_id=1, headers=[], status_code=200)),
call(Data(stream_id=1, data=b"\x89\x00")),
call(Data(stream_id=1, data=b"\x89\x00")),
]
await stream.handle(StreamClosed(stream_id=1))