Add more sending benchmarks (#1053)

esphome · Feb 2, 2025 · ab15d7b · ab15d7b
1 parent 9e03aa0
commit ab15d7b
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/tests/benchmarks/test_requests.py b/tests/benchmarks/test_requests.py
@@ -9,7 +9,7 @@
 from aioesphomeapi.client import APIClient
 
 
-def test_sending_request(
+def test_sending_light_command_request_with_bool(
     benchmark: BenchmarkFixture,
     api_client: tuple[
         APIClient, APIConnection, asyncio.Transport, APIPlaintextFrameHelper
@@ -22,3 +22,18 @@ def test_sending_request(
     @benchmark
     def send_request():
         client.light_command(1, True)
+
+
+def test_sending_empty_light_command_request(
+    benchmark: BenchmarkFixture,
+    api_client: tuple[
+        APIClient, APIConnection, asyncio.Transport, APIPlaintextFrameHelper
+    ],
+) -> None:
+    client, connection, transport, protocol = api_client
+
+    connection._frame_helper._writelines = lambda lines: None
+
+    @benchmark
+    def send_request():
+        client.light_command(1)