[PR #9635/23d2966b backport][3.10] Add benchmark for sending large WebSocket messages (#9638)

Co-authored-by: J. Nick Koston <[email protected]>

Author: patchback[bot]

tests/test_benchmarks_http_websocket.py

@@ -1,12 +1,14 @@
 """codspeed benchmarks for http websocket."""
 
 import asyncio
+from typing import Union
 
 from pytest_codspeed import BenchmarkFixture  # type: ignore[import-untyped]
 
 from aiohttp import DataQueue
 from aiohttp.base_protocol import BaseProtocol
 from aiohttp.http_websocket import (
+    MSG_SIZE,
     WebSocketReader,
     WebSocketWriter,
     WSMessage,
@@ -41,7 +43,7 @@ def is_closing(self) -> bool:
         """Swallow is_closing."""
         return False
 
-    def write(self, data: bytes) -> None:
+    def write(self, data: Union[bytes, bytearray, memoryview]) -> None:
         """Swallow writes."""
 
 
@@ -67,6 +69,22 @@ def _run() -> None:
         loop.run_until_complete(_send_one_hundred_websocket_text_messages())
 
 
+def test_send_one_hundred_large_websocket_text_messages(
+    loop: asyncio.AbstractEventLoop, benchmark: BenchmarkFixture
+) -> None:
+    """Benchmark sending 100 WebSocket text messages."""
+    writer = WebSocketWriter(MockProtocol(loop=loop), MockTransport())
+    raw_message = b"x" * MSG_SIZE * 4
+
+    async def _send_one_hundred_websocket_text_messages() -> None:
+        for _ in range(100):
+            await writer._send_frame(raw_message, WSMsgType.TEXT)
+
+    @benchmark
+    def _run() -> None:
+        loop.run_until_complete(_send_one_hundred_websocket_text_messages())
+
+
 def test_send_one_hundred_websocket_text_messages_with_mask(
     loop: asyncio.AbstractEventLoop, benchmark: BenchmarkFixture
 ) -> None: