reduce code

aio-libs · Nov 3, 2024 · db92393 · db92393
1 parent 319f13e
commit db92393
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/tests/test_benchmarks_http_websocket.py b/tests/test_benchmarks_http_websocket.py
@@ -64,9 +64,7 @@ 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())
+    benchmark(loop.run_until_complete(_send_one_hundred_websocket_text_messages()))
 
 
 def test_send_one_hundred_large_websocket_text_messages(
@@ -80,9 +78,7 @@ 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())
+    benchmark(loop.run_until_complete(_send_one_hundred_websocket_text_messages()))
 
 
 def test_send_one_hundred_websocket_text_messages_with_mask(
@@ -96,6 +92,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())
+    benchmark(loop.run_until_complete(_send_one_hundred_websocket_text_messages()))